分阶段融合多种知识蒸馏方法,防止遗忘并提升学生模型性能。
Integrating Knowledge Distillation Methods: A Sequential Multi-Stage Framework
- 按阶段顺序训练,每阶段用不同蒸馏方法,保留前一阶段知识。
- 在多个模型和组合上均超越基线,显著提升学生模型准确率。
- 自适应加权机制根据教师置信度动态调整损失,适合实际部署。
知识蒸馏(KD)将大型教师模型的知识迁移到紧凑的学生模型中,便于在资源受限设备上高效部署。尽管基于响应、特征和关系的多种KD方法能捕捉教师知识的不同方面,但整合多种方法常因实现复杂、组合僵化和灾难性遗忘而受限。本文提出SMSKD(分阶段多级知识蒸馏)框架,通过逐阶段训练学生模型,每个阶段使用特定蒸馏方法,并利用前一阶段的冻结参考模型锚定已学知识以缓解遗忘。此外,引入基于教师真实类别概率(TCP)的自适应加权机制,动态调整样本级别的参考损失,平衡知识保留与整合。该设计支持任意方法组合与阶段数,计算开销极低。大量实验表明,SMSKD在多种教师-学生架构及方法组合下持续提升学生模型准确率,优于现有基线。消融实验确认阶段式蒸馏与参考模型监督是性能提升主因,TCP自适应加权提供补充优势。整体上,SMSKD是一种实用且高效的异构KD方法集成方案。
原文摘要 · Abstract (English)
Knowledge distillation (KD) transfers knowledge from large teacher models to compact student models, enabling efficient deployment on resource constrained devices. While diverse KD methods, including response based, feature based, and relation based approaches, capture different aspects of teacher knowledge, integrating multiple methods or knowledge sources is promising but often hampered by complex implementation, inflexible combinations, and catastrophic forgetting, which limits practical effectiveness. This work proposes SMSKD (Sequential Multi Stage Knowledge Distillation), a flexible framework that sequentially integrates heterogeneous KD methods. At each stage, the student is trained with a specific distillation method, while a frozen reference model from the previous stage anchors learned knowledge to mitigate forgetting. In addition, we introduce an adaptive weighting mechanism based on the teacher true class probability (TCP) that dynamically adjusts the reference loss per sample to balance knowledge retention and integration. By design, SMSKD supports arbitrary method combinations and stage counts with negligible computational overhead. Extensive experiments show that SMSKD consistently improves student accuracy across diverse teacher student architectures and method combinations, outperforming existing baselines. Ablation studies confirm that stage wise distillation and reference model supervision are primary contributors to performance gains, with TCP based adaptive weighting providing complementary benefits. Overall, SMSKD is a practical and resource efficient solution for integrating heterogeneous KD methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。