中间层知识蒸馏能提升小样本场景下的模型性能
Too much of a good thing -- when knowledge distillation promotes overfitting, and how to avoid it

- 用对齐结构的模块化设计实现跨层知识迁移
- 在细粒度数据集上,加一个中间监督点就能显著缩小性能差距
- 通过注意力分析指导蒸馏策略,适合资源受限的高效建模
卷积神经网络规模扩大导致模型庞大且训练成本高。知识蒸馏(KD)通过将大模型(教师)的知识迁移到小模型(学生)来缓解这一问题,同时减少对训练数据的需求。传统KD仅作用于网络末尾输出层,而对中间层应用的关注较少。本文提出一种基于同构模块的学生结构,实现教师与学生对应模块间的知识蒸馏。在十一组数据集上的实验表明:在经典数据集上,仅蒸馏最后一层已足够甚至最优;而在细粒度、样本稀少的场景下,中间层监督可显著提升性能,即使增加一个蒸馏点也能大幅缩小差距。进一步通过注意力图、中心核对齐(CKA)和Grad-CAM等可解释性分析,探索了不同粒度的监督配置,并研究了教师与学生微调策略的影响。结果表明,合理引导的中间块级蒸馏是构建紧凑、数据高效且不失准确性的模型的关键。
原文摘要 · Abstract (English)
The growing size of Convolutional Neural Networks has led to increasingly large and costly models. Knowledge Distillation (KD) addresses this by transferring knowledge from a large network (teacher) to a small one (student), also reducing the training data required. KD is traditionally applied only at the network's final output. However, its behaviour when applied at intermediate network layers has received little attention. This raises the question of whether intermediate block-wise KD, which provides supervision throughout the network, could offer an advantage under specific conditions, such as few instances per class, which is common in fine-grained datasets. This work proposes a student design based on simple, homogeneous blocks mirroring those of the teacher, distilling knowledge between corresponding blocks. Across eleven datasets, we show that on classic datasets, distilling only the last block is sufficient -- and often best--, whereas fine-grained, data-scarce settings benefit substantially from intermediate supervision, with even a single additional distillation point narrowing the gap considerably. We further study how this supervision should be guided, exploring configurations of varying granularity and informed by an explainability analysis based on attention maps, Centered Kernel Alignment, and Grad-CAM, alongside the impact of teacher and student fine-tuning strategies. This work shows that intermediate block-wise distillation, guided appropriately, is key to building compact data-efficient models without sacrificing accuracy.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。