让教师动态优化解释,提升小模型学习效果
UNDO: Understanding Distillation as Optimization
- 通过迭代识别学生错误,让教师针对性改进解释
- 在数学和常识推理任务上性能提升最高达20%
- 优化后的教师数据可通用不同小模型,适用性广
知识蒸馏已成为将大语言模型知识压缩到更高效小模型的有效策略。然而,传统的一次性蒸馏方法因教师生成的推理过程与学生实际学习需求不匹配,常导致效果不佳。本文提出UNDO:将知识蒸馏重构为优化过程的框架,通过迭代识别学生错误,并引导教师相应优化解释。每轮都精准针对学生的学习缺陷,促使教师提供更具针对性的增强型推理说明。在多个具有挑战性的数学与常识推理任务上的实证评估表明,该迭代蒸馏方法显著优于标准单步蒸馏,性能提升最高达20%。此外,经迭代优化的教师数据即使应用于不同学生模型仍保持有效性,凸显方法的广泛适用性。本工作从根本上将知识蒸馏重新理解为教师-学生间的动态交互过程,实现教师动态优化以提升蒸馏效果。
原文摘要 · Abstract (English)
Knowledge distillation has emerged as an effective strategy for compressing large language models' (LLMs) knowledge into smaller, more efficient student models. However, standard one-shot distillation methods often produce suboptimal results due to a mismatch between teacher-generated rationales and the student's specific learning requirements. In this paper, we introduce the UNDO: UNderstanding Distillation as Optimization framework, designed to bridge this gap by iteratively identifying the student's errors and prompting the teacher to refine its explanations accordingly. Each iteration directly targets the student's learning deficiencies, motivating the teacher to provide tailored and enhanced rationales that specifically address these weaknesses. Empirical evaluations on various challenging mathematical and commonsense reasoning tasks demonstrate that our iterative distillation method, UNDO, significantly outperforms standard one-step distillation methods, achieving performance gains of up to 20%. Additionally, we show that teacher-generated data refined through our iterative process remains effective even when applied to different student models, underscoring the broad applicability of our approach. Our work fundamentally reframes knowledge distillation as an iterative teacher-student interaction, effectively leveraging dynamic refinement by the teacher for better knowledge distillation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。