English

Revisiting Intermediate-Layer Matching in Knowledge Distillation: Layer-Selection Strategy Doesn't Matter (Much)

Machine Learning 2025-12-11 v2 Artificial Intelligence Computation and Language

Abstract

Knowledge distillation (KD) is a popular method of transferring knowledge from a large "teacher" model to a small "student" model. Previous work has explored various layer-selection strategies (e.g., forward matching and in-order random matching) for intermediate-layer matching in KD, where a student layer is forced to resemble a certain teacher layer. In this work, we revisit such layer-selection strategies and observe an intriguing phenomenon that layer-selection strategy does not matter (much) in intermediate-layer matching -- even seemingly nonsensical matching strategies such as reverse matching still result in surprisingly good student performance. We provide an interpretation for this phenomenon by examining the angles between teacher layers viewed from the student's perspective. Our work sheds light on KD practice, as layer-selection strategies may not be the main focus of KD system design, and vanilla forward matching works well in most setups.

Keywords

Cite

@article{arxiv.2502.04499,
  title  = {Revisiting Intermediate-Layer Matching in Knowledge Distillation: Layer-Selection Strategy Doesn't Matter (Much)},
  author = {Zony Yu and Yuqiao Wen and Lili Mou},
  journal= {arXiv preprint arXiv:2502.04499},
  year   = {2025}
}

Comments

Accepted at IJCNLP-AACL 2025

R2 v1 2026-06-28T21:35:29.119Z