arXiv:2506.18244cs.LG2025-06被引 3

用双路径提示调优,缩小师生模型能力差距

Dual-Forward Path Teacher Knowledge Distillation: Bridging the Capacity Gap Between Teacher and Student

  • 在教师模型中引入可调提示路径,动态适配学生能力
  • 相比传统方法,学生模型性能提升显著,达最新水平
  • 适合需要高效压缩大模型的部署场景

知识蒸馏通过预训练教师指导学生网络提升性能,但通常存在师生模型能力差距过大问题,限制蒸馏效果。现有方法或丢失精确知识表示,或无法动态调整传输知识,难以有效缓解能力差距。本文提出双前向路径教师知识蒸馏(DFPT-KD),将提示学习思想引入教师模型,构建额外的提示调优路径,在冻结原教师的前提下优化该路径,使传输知识更匹配学生表达能力。大量实验表明,DFPT-KD使学生性能优于传统蒸馏。进一步对整个提示路径进行微调,形成改进版DFPT-KD+,实验显示其优于DFPT-KD并达到当前最优精度。

原文摘要 · Abstract (English)

Knowledge distillation (KD) provides an effective way to improve the performance of a student network under the guidance of pre-trained teachers. However, this approach usually brings in a large capacity gap between teacher and student networks, limiting the distillation gains. Previous methods addressing this problem either discard accurate knowledge representation or fail to dynamically adjust the transferred knowledge, which is less effective in addressing the capacity gap problem and hinders students from achieving comparable performance with the pre-trained teacher. In this work, we extend the ideology of prompt-based learning to address the capacity gap problem, and propose Dual-Forward Path Teacher Knowledge Distillation (DFPT-KD), which replaces the pre-trained teacher with a novel dual-forward path teacher to supervise the learning of student. The key to DFPT-KD is prompt-based tuning, i.e., establishing an additional prompt-based forward path within the pre-trained teacher and optimizing it with the pre-trained teacher frozen to make the transferred knowledge compatible with the representation ability of the student. Extensive experiments demonstrate that DFPT-KD leads to trained students performing better than the vanilla KD. To make the transferred knowledge better compatible with the representation abilities of the student, we further fine-tune the whole prompt-based forward path, yielding a novel distillation approach dubbed DFPT-KD+. By extensive experiments, it is shown that DFPT-KD+ improves upon DFPT-KD and achieves state-of-the-art accuracy performance.

知识蒸馏提示学习模型压缩

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