让模型分阶段学习老师输出,提升小模型对低概率类别的识别能力。
Progressive Class-level Distillation
- 分阶段进行知识蒸馏,按优先级逐步转移不同类别知识。
- 在多个阶段内分别对不同类别组进行双向渐进式对齐,提升对齐精度。
- 特别适合需要精准识别冷门类别的任务,如细粒度分类与目标检测。
知识蒸馏中,logit蒸馏(LD)通过精确的教师-学生输出对齐,将强教师网络的类别级知识传递给小型学生模型。然而,传统方法常因高置信度类别主导而忽略低概率但具有判别性的类别,导致知识传递不充分。为此,本文提出一种简单而有效的LD方法——渐进式类别级蒸馏(PCD)。不同于现有全类别联合蒸馏的方法,PCD采用分阶段策略:先根据教师与学生输出差异进行排序,确定蒸馏优先级,再将整个蒸馏过程划分为多个阶段;每个阶段内实施双向渐进式蒸馏,包含从细到粗的逐步学习和反向粗到精的优化,从而在不同类别组中实现充分的输出对齐。在公开基准数据集上的扩展实验表明,该方法在分类和检测任务上均优于当前最先进方法。
原文摘要 · Abstract (English)
In knowledge distillation (KD), logit distillation (LD) aims to transfer class-level knowledge from a more powerful teacher network to a small student model via accurate teacher-student alignment at the logits level. Since high-confidence object classes usually dominate the distillation process, low-probability classes which also contain discriminating information are downplayed in conventional methods, leading to insufficient knowledge transfer. To address this issue, we propose a simple yet effective LD method termed Progressive Class-level Distillation (PCD). In contrast to existing methods which perform all-class ensemble distillation, our PCD approach performs stage-wise distillation for step-by-step knowledge transfer. More specifically, we perform ranking on teacher-student logits difference for identifying distillation priority from scratch, and subsequently divide the entire LD process into multiple stages. Next, bidirectional stage-wise distillation incorporating fine-to-coarse progressive learning and reverse coarse-to-fine refinement is conducted, allowing comprehensive knowledge transfer via sufficient logits alignment within separate class groups in different distillation stages. Extension experiments on public benchmarking datasets demonstrate the superiority of our method compared to state-of-the-arts for both classification and detection tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。