arXiv:2512.04625cs.LGcs.CV2025-12中稿 · IEEE TNNLS被引 3

从预测分布视角重思考解耦知识蒸馏,提升模型性能。

Rethinking Decoupled Knowledge Distillation: A Predictive Distribution Perspective

  • 提出广义解耦蒸馏损失,更灵活地分离教师模型输出 logits。
  • 发现顶部logit划分能增强非顶部logit间的关联性,提升知识提取。
  • 设计高效分区策略,适配教师模型多模态输出分布,适合视觉任务研究者。

在知识蒸馏的发展历程中,方法重心曾从基于logit的范式转向基于特征的范式。然而,随着解耦知识蒸馏(DKD)的出现,这一趋势被重新审视,其通过先进的解耦与加权策略再次强调了logit知识的重要性。尽管如此,其内在机制仍有待深入探索。为此,本文从预测分布的角度重新思考DKD。首先提出一种改进的广义解耦知识蒸馏(GDKD)损失,提供更灵活的logit解耦方式;随后重点分析教师模型预测分布对GDKD损失梯度的影响,揭示两个常被忽略的关键洞察:(1) 以最高logit进行划分可显著改善非最高logit之间的关联性;(2) 强化非最高logit的蒸馏损失有助于提升它们之间的知识提取。基于上述发现,进一步提出一种简化的GDKD算法,采用高效分区策略应对教师模型预测分布的多模态特性。在CIFAR-100、ImageNet、Tiny-ImageNet、CUB-200-2011和Cityscapes等多个基准上的实验表明,GDKD在性能上优于原始DKD及其他主流知识蒸馏方法。

原文摘要 · Abstract (English)

In the history of knowledge distillation, the focus has once shifted over time from logit-based to feature-based approaches. However, this transition has been revisited with the advent of Decoupled Knowledge Distillation (DKD), which re-emphasizes the importance of logit knowledge through advanced decoupling and weighting strategies. While DKD marks a significant advancement, its underlying mechanisms merit deeper exploration. As a response, we rethink DKD from a predictive distribution perspective. First, we introduce an enhanced version, the Generalized Decoupled Knowledge Distillation (GDKD) loss, which offers a more versatile method for decoupling logits. Then we pay particular attention to the teacher model's predictive distribution and its impact on the gradients of GDKD loss, uncovering two critical insights often overlooked: (1) the partitioning by the top logit considerably improves the interrelationship of non-top logits, and (2) amplifying the focus on the distillation loss of non-top logits enhances the knowledge extraction among them. Utilizing these insights, we further propose a streamlined GDKD algorithm with an efficient partition strategy to handle the multimodality of teacher models' predictive distribution. Our comprehensive experiments conducted on a variety of benchmarks, including CIFAR-100, ImageNet, Tiny-ImageNet, CUB-200-2011, and Cityscapes, demonstrate GDKD's superior performance over both the original DKD and other leading knowledge distillation methods. The code is available at https://github.com/ZaberKo/GDKD.

知识蒸馏模型压缩深度学习预测分布

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。