Knowledge Distillation with Deep Supervision

Shiya Luo,Defang Chen,Can Wang
DOI: https://doi.org/10.48550/arXiv.2202.07846
2023-05-25
Abstract:Knowledge distillation aims to enhance the performance of a lightweight student model by exploiting the knowledge from a pre-trained cumbersome teacher model. However, in the traditional knowledge distillation, teacher predictions are only used to provide the supervisory signal for the last layer of the student model, which may result in those shallow student layers lacking accurate training guidance in the layer-by-layer back propagation and thus hinders effective knowledge transfer. To address this issue, we propose Deeply-Supervised Knowledge Distillation (DSKD), which fully utilizes class predictions and feature maps of the teacher model to supervise the training of shallow student layers. A loss-based weight allocation strategy is developed in DSKD to adaptively balance the learning process of each shallow layer, so as to further improve the student performance. Extensive experiments on CIFAR-100 and TinyImageNet with various teacher-student models show significantly performance, confirming the effectiveness of our proposed method. Code is available at: $\href{<a class="link-external link-https" href="https://github.com/luoshiya/DSKD" rel="external noopener nofollow">this https URL</a>}{<a class="link-external link-https" href="https://github.com/luoshiya/DSKD" rel="external noopener nofollow">this https URL</a>}$
Machine Learning,Computer Vision and Pattern Recognition
What problem does this paper attempt to address?