Skip to main navigation Skip to search Skip to main content

Teacher–student complementary sample contrastive distillation

Zhiqiang Bao, Zhenhua Huang, Jianping Gou, Lan Du, Kang Liu, Jingtao Zhou, Yunwen Chen

Research output: Contribution to journalArticleResearchpeer-review

Abstract

Knowledge distillation (KD) is a widely adopted model compression technique for improving the performance of compact student models, by utilizing the “dark knowledge” of a large teacher model. However, previous studies have not adequately investigated the effectiveness of supervision from the teacher model, and overconfident predictions in the student model may degrade its performance. In this work, we propose a novel framework, Teacher–Student Complementary Sample Contrastive Distillation (TSCSCD), that alleviate these challenges. TSCSCD consists of three key components: Contrastive Sample Hardness (CSH), Supervision Signal Correction (SSC), and Student Self-Learning (SSL). Specifically, CSH evaluates the teacher's supervision for each sample by comparing the predictions of two compact models, one distilled from the teacher and the other trained from scratch. SSC corrects weak supervision according to CSH, while SSL employs integrated learning among multi-classifiers to regularize overconfident predictions. Extensive experiments on four real-world datasets demonstrate that TSCSCD outperforms recent state-of-the-art knowledge distillation techniques.

Original languageEnglish
Pages (from-to)176-189
Number of pages14
JournalNeural Networks
Volume170
DOIs
Publication statusPublished - Feb 2024

Keywords

  • Deep learning
  • Knowledge distillation
  • Model regularization
  • Sample hardness
  • Transfer learning

Cite this