arrow
Return

Multi-target Knowledge Distillation via Student Self-reflection

delete2023-04-25
delete24
delete
OA
AI
J
Jianping Gou
X
Xiangshuo Xiong
B
Baosheng Yu *
L
Lan Du
Y
Yibing Zhan
D
Dacheng Tao
DOI:10.1007/s11263-023-01792-zdelete
deleteOriginal
deleteShare
deleteSave
Abstract

Abstract

En 中文
Knowledge distillation is a simple yet effective technique for deep model compression, which aims to transfer the knowledge learned by a large teacher model to a small student model. To mimic how the teacher teaches the student, existing knowledge distillation methods mainly adapt an unidirectional knowledge transfer, where the knowledge extracted from different intermedicate layers of the teacher model is used to guide the student model. However, it turns out that the students can learn more effectively through multi-stage learning with a self-reflection in the real-world education scenario, which is nevertheless ignored by current knowledge distillation methods. Inspired by this, we devise a new knowledge distillation framework entitled multi-target knowledge distillation via student self-reflection or MTKD-SSR, which can not only enhance the teacher's ability in unfolding the knowledge to be distilled, but also improve the student's capacity of digesting the knowledge. Specifically, the proposed framework consists of three target knowledge distillation mechanisms: a stage-wise channel distillation (SCD), a stage-wise response distillation (SRD), and a cross-stage review distillation (CRD), where SCD and SRD transfer feature-based knowledge (i.e., channel features) and response-based knowledge (i.e., logits) at different stages, respectively; and CRD encourages the student model to conduct self-reflective learning after each stage by a self-distillation of the response-based knowledge. Experimental results on five popular visual recognition datasets, CIFAR-100, Market-1501, CUB200-2011, ImageNet, and Pascal VOC, demonstrate that the proposed framework significantly outperforms recent state-of-the-art knowledge distillation methods.
Keywords:
Knowledge distillation
Self-reflection learning
Model compression
Deep learning

Journal

International Journal of Computer Vision cover
International Journal of Computer Vision
IF:
9.3
Papers:
4.0K
Citations:
2.8W

Organization

S
southwest university - china
Scholars:
0
Papers: 361
Citations: 0
J
jiangsu university
Scholars:
3.0K
Papers: 725
Citations: 0
U
University of Sydney
Scholars:
1.1K
Papers: 510
Citations: 0
M
monash university
Scholars:
1.7K
Papers: 709
Citations: 0
researcher View more organizations
Cited Papers

Cited Papers

errShare
errSave
errShare
errSave
Knowledge Distillation: A Survey
err2021-03-22
err1.5K
PREAI
errGou, Jianping; Yu, Baosheng; Maybank, Stephen J.; Tao, Dacheng
errShare
errSave
Collaborative Knowledge Distillation via Multiknowledge Transfer
err2024-05-01
err11
PREAI
errGou, Jianping; Sun, Liyuan; Yu, Baosheng; Du, Lan; Ramamohanarao, Kotagiri; Tao, Dacheng
errShare
errSave
A role for anion transport in the regulation of release from chromaffin granules and exocytosis from cells
err2004-02-19
err0
PREAI
errHarvey B. Pollard; Christopher J. Pazoles; Carl E. Creutz; Avner Ramu; Charles A. Strott; Probhati Ray; Edward M. Brown; G. D. Aurbach; Karen M. Tack‐Goldman; N. Raphael Shulman
errShare
errSave
researcher View more