English

Knowledge distillation for optimization of quantized deep neural networks

Machine Learning 2019-10-24 v3 Machine Learning

Abstract

Knowledge distillation (KD) is a very popular method for model size reduction. Recently, the technique is exploited for quantized deep neural networks (QDNNs) training as a way to restore the performance sacrificed by word-length reduction. KD, however, employs additional hyper-parameters, such as temperature, coefficient, and the size of teacher network for QDNN training. We analyze the effect of these hyper-parameters for QDNN optimization with KD. We find that these hyper-parameters are inter-related, and also introduce a simple and effective technique that reduces \textit{coefficient} during training. With KD employing the proposed hyper-parameters, we achieve the test accuracy of 92.7% and 67.0% on Resnet20 with 2-bit ternary weights for CIFAR-10 and CIFAR-100 data sets, respectively.

Keywords

Cite

@article{arxiv.1909.01688,
  title  = {Knowledge distillation for optimization of quantized deep neural networks},
  author = {Sungho Shin and Yoonho Boo and Wonyong Sung},
  journal= {arXiv preprint arXiv:1909.01688},
  year   = {2019}
}
R2 v1 2026-06-23T11:05:05.907Z