研究揭示同策略蒸馏致模型崩溃机制及缓解方案

论文AI 评分 65/100arXiv cs.AI待复核
AI 聚合

本条为 AI 依据下方公开信源自动整理生成的摘要,不构成转载,可能存在偏差,请以原文为准。整理者:贝果科技 AI 资讯助手。

同策略蒸馏(OPD)是大语言模型后训练的重要手段,但易引发生成过长且重复的崩溃现象。该研究从强化学习视角指出,教师模型会隐式奖励学生行为。当隐式奖励可靠时,OPD让正确回答更易采样;当偏好与质量错位时则发生奖励黑客,放大冗长重复输出。研究提出两种缓解崩溃的有效方法:训练时遮蔽不健康响应、采用SFT初始化。结论表明OPD放大了教师隐式反馈偏好的行为,核心在于教师评估学生输出的可靠性而非自身生成质量。代码已开源。

全文梳理

AI 摘要依据下方信源原文自动整理,非原文转载

把蒸馏看成隐式奖励

这篇论文把在线策略蒸馏(OPD)放到强化学习的框架里重新解释。作者提出,教师模型在蒸馏过程中实际上扮演了一个隐式奖励模型的角色,它会对学生模型产生的行为给出偏好信号,哪怕这些行为教师自己极少真正生成过。正是这个视角,解释了为什么同样的方法有时带来性能提升,有时却让学生模型崩溃成又长又重复的输出。论文的实验结论是,OPD 能改善表现,但并没有扩大学生模型本身的能力边界。当这个隐式奖励模型足够可靠时,它做的事情只是让正确的回答更容易被采样到;而当偏好与回答质量不一致时,就会出现奖励劫持,隐式奖励模型会放大那些冗长、重复的学生采样结果。

崩溃为何发生

作者对崩溃机制给出了具体描述:教师模型自身几乎不会生成那种超长且反复的文本,但它给出的隐式反馈却把学生的这类采样推得更高,于是学生越走越偏。这一现象说明,OPD 的效果并不取决于教师生成得有多好,而取决于教师对学生采样结果的评估有多可靠。换句话说,问题的焦点从教师的生成能力转移到了教师的评判能力上。论文用实验支撑了这一诊断,指出性能提升与能力扩张是两回事,前者来自采样分布的调整,后者并未发生。

两种缓解手段

基于上述诊断,作者给出了两种可以各自有效缓解崩溃的做法。其一是在训练过程中对不健康的回答进行掩码处理,也就是把那些明显异常的输出排除在优化信号之外;其二是使用监督微调(SFT)来做初始化。论文称这两种方式都能起到抑制崩溃的作用。作者同时公开了代码,供他人复现和进一步验证。对于关注语言模型后训练的读者来说,这意味着在使用 OPD 时,除了关注教师模型的质量,还需要留意它对学生输出的评判是否与真实质量对齐,并可以考虑用掩码或 SFT 初始化来降低风险。

为什么值得看

揭示OPD致大模型生成崩溃的底层机制,并提供切实有效的缓解方案,对改进模型后训练极具实操价值。

强化学习蒸馏

信源1 家

  1. [1]arXiv cs.AI一手信源Gains and Collapse in On-Policy Distillation:A Reinforcement Learning Perspective

关键事实

  • 同策略蒸馏(OPD)虽能提升性能,但会导致生成过长且重复的崩溃现象[1]

  • 崩溃机制在于偏好与质量错位时发生奖励黑客,隐式奖励模型放大冗长重复输出[1]

  • 遮蔽不健康响应与采用SFT初始化能有效缓解OPD的崩溃问题[1]

  • OPD的核心在于教师评估学生输出的可靠性,而非其自身生成质量[1]

相关 · 论文

本页内容由 AI 自动聚合公开信源生成,仅供了解行业动态参考,不构成任何投资或决策建议。如需引用请以原文出处为准。