Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Knowledge Distillation from A Stronger Teacher

About

Unlike existing knowledge distillation methods focus on the baseline settings, where the teacher models and training strategies are not that strong and competing as state-of-the-art approaches, this paper presents a method dubbed DIST to distill better from a stronger teacher. We empirically find that the discrepancy of predictions between the student and a stronger teacher may tend to be fairly severer. As a result, the exact match of predictions in KL divergence would disturb the training and make existing methods perform poorly. In this paper, we show that simply preserving the relations between the predictions of teacher and student would suffice, and propose a correlation-based loss to capture the intrinsic inter-class relations from the teacher explicitly. Besides, considering that different instances have different semantic similarities to each class, we also extend this relational match to the intra-class level. Our method is simple yet practical, and extensive experiments demonstrate that it adapts well to various architectures, model sizes and training strategies, and can achieve state-of-the-art performance consistently on image classification, object detection, and semantic segmentation tasks. Code is available at: https://github.com/hunto/DIST_KD .

Tao Huang, Shan You, Fei Wang, Chen Qian, Chang Xu• 2022

Related benchmarks

TaskDatasetResultRank
Image ClassificationCIFAR-100 (test)
Accuracy78.86
3518
Image ClassificationImageNet-1K 1.0 (val)
Top-1 Accuracy75.13
1952
Image ClassificationImageNet-1K
Top-1 Acc71.76
1239
Video Object SegmentationDAVIS 2017 (val)
J mean68.5
1193
Image ClassificationCIFAR-100 (val)--
776
Object DetectionCOCO (val)
mAP41.8
633
Video Object SegmentationDAVIS 2016 (val)--
564
Video Object SegmentationYouTube-VOS 2018 (val)--
493
Image ClassificationTinyImageNet (test)--
440
Image ClassificationImageNet (val)
Top-1 Accuracy73.24
354
Showing 10 of 17 rows

Other info

Code

Follow for update