arXiv:2409.18785cs.CV2024-09被引 3

让教师知识更贴合学生,提升模型压缩效果

Student-Oriented Teacher Knowledge Refinement for Knowledge Distillation

  • 动态优化教师知识以匹配学生能力
  • 通过关键区域检测聚焦有效知识传递
  • 可无缝接入多种现有蒸馏方法

知识蒸馏广泛用于将大型教师网络的知识迁移到小型学生网络。传统方法多采用教师主导范式,将教师的复杂知识强加给学生,但因模型容量和结构差异,学生难以理解,导致性能不佳。本文提出学生导向的知识蒸馏(SoKD),通过训练中可学习的特征增强策略,动态优化教师知识以适应学生需求。同时引入独特区域检测模块(DAM),识别师生间共同关注区域,集中在此类关键区域进行知识传递,避免无关信息干扰。该方法作为即插即用模块,可与多种蒸馏方法兼容。大量实验验证了其有效性与通用性。

原文摘要 · Abstract (English)

Knowledge distillation has become widely recognized for its ability to transfer knowledge from a large teacher network to a compact and more streamlined student network. Traditional knowledge distillation methods primarily follow a teacher-oriented paradigm that imposes the task of learning the teacher's complex knowledge onto the student network. However, significant disparities in model capacity and architectural design hinder the student's comprehension of the complex knowledge imparted by the teacher, resulting in sub-optimal performance. This paper introduces a novel perspective emphasizing student-oriented and refining the teacher's knowledge to better align with the student's needs, thereby improving knowledge transfer effectiveness. Specifically, we present the Student-Oriented Knowledge Distillation (SoKD), which incorporates a learnable feature augmentation strategy during training to refine the teacher's knowledge of the student dynamically. Furthermore, we deploy the Distinctive Area Detection Module (DAM) to identify areas of mutual interest between the teacher and student, concentrating knowledge transfer within these critical areas to avoid transferring irrelevant information. This customized module ensures a more focused and effective knowledge distillation process. Our approach, functioning as a plug-in, could be integrated with various knowledge distillation methods. Extensive experimental results demonstrate the efficacy and generalizability of our method.

知识蒸馏模型压缩学生导向动态优化

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