Revisit Knowledge Distillation: a Teacher-free Framework

By Li Yuan, Francis E. H. Tay, Guilin Li, Tao Wang, Jiashi Feng. 2019

98 citations recorded by the source corpus.

Knowledge Distillation (KD) aims to distill the knowledge of a cumbersome teacher model into a lightweight student model. Its success is generally attributed to the privileged information on similarities among categories provided by the teacher model, and in this sense, only strong teacher models are deployed to teach weaker students in practice. In this work, we challenge this common belief by following experimental observations: 1) beyond the acknowledgment that the teacher can improve the student, the student can also enhance the teacher significantly by reversing the KD procedure; 2) a poorly-trained teacher with much lower accuracy than the student can still improve the latter significantly. To explain these observations, we provide a theoretical analysis of the relationships between KD and label smoothing regularization. We prove that 1) KD is a type of learned label smoothing regularization and 2) label smoothing regularization provides a virtual teacher model for KD. From these results, we argue that the success of KD is not fully due to the similarity information between categories, but also to the regularization of soft targets, which is equally or even more important. Based on these analyses, we further propose a novel Teacher-free Knowledge Distillation (Tf-KD) framework, where a student model learns from itself or manually-designed regularization distribution. The Tf-KD achieves comparable performance with normal KD from a superior teacher, which is well applied when teacher model is unavailable. Meanwhile, Tf-KD is generic and can be directly deployed for training deep neural networks. Without any extra computation cost, Tf-KD achieves up to 0.65\% improvement on ImageNet over well-established baseline models, which is superior to label smoothing regularization. The codes are in: \url{this https URL}

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition, Efficiency & Compression

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Dacheng Tao
  2. Jiashi Feng
  3. Shuicheng Yan
  4. Zhiqiang Shen
  5. Jianping Gou
  6. Li Yuan
  7. Mostafa Dehghani
  8. S. Maybank

In Computer Vision and Pattern Recognition

  1. Dacheng Tao
  2. Jiashi Feng
  3. Luc Van Gool
  4. Shuicheng Yan
  5. Trevor Darrell
  6. Ming-Hsuan Yang
  7. Xiaogang Wang
  8. Yi Yang

In Efficiency & Compression

  1. Dacheng Tao
  2. Jianping Gou
  3. Li Yuan
  4. Shuicheng Yan
  5. S. Maybank
  6. Kwang-Ting Cheng
  7. Zhiqiang Shen
  8. Zhilu Zhang

Citations in the corpus