arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.00573cs.CV

Multi-level Knowledge Distillation via Knowledge Alignment and Correlation

Clemson University(克莱姆森大学);University at Buffalo(布法罗大学)

Fei Ding, Yin Yang, Hongxin Hu, Venkat Krovi, Feng Luo

英文摘要:

Knowledge distillation (KD) has become an important technique for model compression and knowledge transfer. In this work, we first perform a comprehensive analysis of the knowledge transferred by different KD methods. We demonstrate that traditional KD methods, which minimize the KL divergence of softmax outputs between networks, are related to the knowledge alignment of an individual sample only. Meanwhile, recent contrastive learning-based KD methods mainly transfer relational knowledge between different samples, namely, knowledge correlation. While it is important to transfer the full knowledge from teacher to student, we introduce the Multi-level Knowledge Distillation (MLKD) by effectively considering both knowledge alignment and correlation. MLKD is task-agnostic and model-agnostic, and can easily transfer knowledge from supervised or self-supervised pretrained teachers. We show that MLKD can improve the reliability and transferability of learned representations. Experiments demonstrate that MLKD outperforms other state-of-the-art methods on a large number of experimental settings including different (a) pretraining strategies (b) network architectures (c) datasets (d) tasks.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑