最弱环节:用最坏情况约束RL蒸馏LLM推理
原文标题:The Weakest Link: Distilling LLM Reasoning with Worst-Case Constrained Reinforcement Learning
重要度 3/5 · AI 综合评估报道影响力与新鲜度,4–5 分为高价值资讯。
论文指出LLM推理蒸馏存在张力:纯任务奖励优化导致奖励黑客,软散度正则又稀释性能并允许学生用其他步骤的高教师一致性补偿严重逻辑违规。作者将推理蒸馏建模为约束强化学习,在轨迹每个前缀上对教师对数似然施加最坏情况约束。实验在数学推理与代码生成上扩展了准确率-保真度帕累托前沿,取得最高严格推理成功率。
- 问题
- LLM推理蒸馏中纯任务奖励导致奖励黑客,软散度正则又稀释性能并容忍局部逻辑违规。
- 方法
- 把推理蒸馏建模为约束强化学习,对轨迹每个前缀的教师对数似然施加最坏情况约束,导出无增广约束MDP与低方差策略梯度。
- 结果
- 在数学推理与代码生成上扩展准确率-保真度帕累托前沿,取得最高严格推理成功率。
- 影响
- 为小模型蒸馏提供兼顾任务性能与推理链正确性的训练目标,降低部署模型靠错误中间步骤蒙对答案的风险。
深读 · 结论先行AI 生成展开看要点与影响 ↓
把推理蒸馏建模为约束RL,对轨迹每个前缀施加教师似然最坏情况约束,扩展准确率-保真度帕累托前沿
- •张力:纯任务奖励(如GRPO)导致奖励黑客,学生用错误中间逻辑得正确答案
- •软散度正则(如KL蒸馏)稀释性能,且允许学生用其他步骤的高教师一致性补偿某步严重逻辑违规
- •核心论点:思维链的有效性取决于最弱环节,平均化与推理本质不符
- •方案:在轨迹每个前缀上对教师对数似然施加最坏情况约束,最大化任务奖励
- •推导出无增强约束MDP,奖励变换保留硬约束语义,策略梯度可分解为单步与长期项,惩罚极限下几乎必然满足约束
为什么重要:为「答案对但过程错」的蒸馏失效提供了原理性修正,对需要高保真推理链的小模型部署场景尤为关键。
以上深读由 AI 依据标题/摘要/正文节选生成,仅供参考;事实以原文为准。
AI 处理过程展开/收起
相关推荐
OpenAI 宣布数学、计算机科学、物理学三项新发现
AI 摘要OpenAI 宣布在调和分析、复杂性理论和量子场论三个方向取得新发现,并采用与学术界合作的新方法。公司邀请外部专家组织团队理解研究结果,参与者无需签署保密协议、可自由讨论并发表结论,且为自愿无报酬。OpenAI 表示不会为内部模型产生的发现寻求学术署名。
数学家称需数年消化 OpenAI 最新数学成果
AI 摘要OpenAI 本周突然发布近 400 项 AI 生成的数学成果,分布在 700 多份手稿中,覆盖组合数学、几何、数论等多个领域。多位数学家表示「震惊」「难以置信」,认为仅理解这些成果就需数年,且验证程度参差不齐。
OpenAI 公布数学难题突破成果
AI 摘要OpenAI 发布由未公开前沿模型生成的 722 篇数学手稿,覆盖 372 个成果族,声称解决了数百个长期未解问题,并公开 Lean 形式化证明与算力估算。此举延续了 AI 冲击数学界的势头,也引发研究伦理与学术规范的争议。独立顾问组织 AGMAI 呼吁实验室通过正规学术渠道发布成果,而非将其当作营销工具。
OpenAI 再抛数学突破成果引争议
AI 摘要OpenAI 公布 722 篇手稿,称未发布的前沿模型解决了数百个数学开放问题,平均每个结果消耗约三小时 ChatGPT Pro 级算力。成果虽令人震撼,但数学界对发布方式、研究伦理与学术规范提出质疑。AGMAI 建议实验室及时通过正规渠道披露模型、提示词与算力成本,避免把成果当作营销手段。