English

A Closer Look at Knowledge Distillation with Features, Logits, and Gradients

Machine Learning 2022-03-22 v1 Artificial Intelligence Computer Vision and Pattern Recognition

Abstract

Knowledge distillation (KD) is a substantial strategy for transferring learned knowledge from one neural network model to another. A vast number of methods have been developed for this strategy. While most method designs a more efficient way to facilitate knowledge transfer, less attention has been put on comparing the effect of knowledge sources such as features, logits, and gradients. This work provides a new perspective to motivate a set of knowledge distillation strategies by approximating the classical KL-divergence criteria with different knowledge sources, making a systematic comparison possible in model compression and incremental learning. Our analysis indicates that logits are generally a more efficient knowledge source and suggests that having sufficient feature dimensions is crucial for the model design, providing a practical guideline for effective KD-based transfer learning.

Keywords

Cite

@article{arxiv.2203.10163,
  title  = {A Closer Look at Knowledge Distillation with Features, Logits, and Gradients},
  author = {Yen-Chang Hsu and James Smith and Yilin Shen and Zsolt Kira and Hongxia Jin},
  journal= {arXiv preprint arXiv:2203.10163},
  year   = {2022}
}
R2 v1 2026-06-24T10:18:50.106Z