跳到正文
arXiv cs.CL· Taiming Lu, Zhuang Liu·· 13 小时前AI 评分34

强教师并非必需?论 LLM 预训练中的知识蒸馏

Strong Teacher Not Needed? On Distillation in LLM Pretraining

AI 导读

一项 NeurIPS 2026 研究挑战了知识蒸馏必须“强教师带弱学生”的假设:在 LLM 预训练中,只要合理混合语言建模损失与蒸馏损失,即使规模小、训练不足的教师也能提升更大学生的表现。而通过增加参数量或训练 token 继续强化教师,蒸馏收益可能饱和甚至反转。研究还发现,蒸馏对分布外和下游性能等泛化能力的提升比对域内拟合更明显。

正文 · AI 翻译

查看 PDF HTML(实验性)

摘要:知识蒸馏通常假设存在一种强到弱的关系,即更强的教师能培养出更好的学生。在本工作中,我们考察了大型语言模型预训练中关于蒸馏的这一假设。通过改变架构规模和训练 token 预算,我们构建了强到弱、同级别和弱到强的师生关系,并研究了每种关系下蒸馏的有效性。我们发现教师不必很强:只要适当混合语言建模损失和知识蒸馏损失,即使规模小且训练不足的教师也能提升更大学生的表现。同时,更强的教师并不总是更好:通过更多参数或更多训练 token 进一步强化教师,可能会使蒸馏收益饱和甚至逆转。我们进一步观察到,蒸馏比域内拟合更容易提升泛化能力(分布外性能和下游性能)。总之,这些结果挑战了蒸馏预训练总是需要强教师的普遍看法。
评论: NeurIPS 2026
主题: 机器学习(cs.LG);计算与语言(cs.CL)
引用为: arXiv:2605.23857 [cs.LG]
  (或本版本使用 arXiv:2605.23857v2 [cs.LG])
  https://doi.org/10.48550/arXiv.2605.23857

由 DataCite 颁发的 arXiv DOI

提交历史

来自:TaiMing Lu [查看邮箱]
[v1] 2026年5月22日星期五 17:16:35 UTC(5,513 KB)
[v2] 2026年10月26日星期一 22:40:35 UTC(5,037 KB)

来源:arXiv cs.CL · arxiv.org