通过师生协作生成高质量训练数据,解决知识蒸馏中的分布差距问题。
Speculative Knowledge Distillation: Bridging the Teacher-Student Gap Through Interleaved Sampling
- 学生提词,教师替换低质量词,动态生成适配推理分布的训练样本。
- 在翻译、摘要、数学和指令跟随任务中均超越现有蒸馏方法。
- 适合追求小模型高性能且关注推理一致性的研究者与工程师。
近期知识蒸馏(KD)进展使小型学生模型可逼近大型教师模型性能。然而,监督蒸馏与同策略蒸馏等主流方法在实际场景中受师生间知识鸿沟影响:监督蒸馏存在训练数据分布与学生最终输出分布不匹配问题;同策略蒸馏虽用学生生成样本训练,但可能引入教师不熟悉、质量低的样本,导致反馈失准。为此,本文提出推测性知识蒸馏(Speculative Knowledge Distillation, SKD),通过师生协同实时生成高质量训练数据,并对齐学生推理时的分布。在SKD中,学生提出候选词,教师根据自身分布替换排名低的词,实现自适应知识迁移。我们在多种文本生成任务(包括翻译、摘要、数学推理和指令遵循)上评估,结果表明SKD在不同领域、数据规模及初始化策略下均持续优于现有方法。
原文摘要 · Abstract (English)
Recent advances in knowledge distillation (KD) have enabled smaller student models to approach the performance of larger teacher models. However, popular methods such as supervised KD and on-policy KD, are adversely impacted by the knowledge gaps between teacher-student in practical scenarios. Supervised KD suffers from a distribution mismatch between training with a static dataset and inference over final student-generated outputs. Conversely, on-policy KD, which uses student-generated samples for training, can suffer from low-quality training examples with which teacher models are not familiar, resulting in inaccurate teacher feedback. To address these limitations, we introduce Speculative Knowledge Distillation (SKD), a novel approach that leverages cooperation between student and teacher models to generate high-quality training data on-the-fly while aligning with the student's inference-time distribution. In SKD, the student proposes tokens, and the teacher replaces poorly ranked ones based on its own distribution, transferring high-quality knowledge adaptively. We evaluate SKD on various text generation tasks, including translation, summarization, math, and instruction following, and show that SKD consistently outperforms existing KD methods across different domains, data sizes, and model initialization strategies.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。