跳到主要内容
AI News HubLIVE
站内改写1 分钟阅读

基于同伴成功与失败的多轮次在线策略蒸馏

文章摘要

提出MOPD框架,利用学生的多个生成轨迹(包括成功和失败的)构建更有效的教师信号,优于标准在线策略蒸馏方法。实验表明,混合成功-失败上下文能更好地对齐教师分数与验证器奖励,带来更忠实、实例自适应的监督。

来源arXiv Machine Learning作者: Weichen Yu, Xiaomin Li, Yizhou Zhao, Xiaoze Liu, Ruowang Zhang, Haixin Wang, Yinyi Luo, Chen Henry Wu, Gaurav Mittal, Matt Fredrikson, Yu Hu
基于同伴成功与失败的多轮次在线策略蒸馏
报告错误

纠错通道尚未开通,可先复制下方文章信息留存。

查看更正说明
直接读正文

大型语言模型在后期训练中常使用稀疏的验证器奖励,这种奖励仅能指示采样轨迹是否成功,却无法对推理过程中的成功或失败位置提供具体指导。在线策略蒸馏(On-policy Distillation, OPD)通过训练学生模型自身生成的轨迹来提供更密集的令牌级监督,但现有方法通常独立处理每个轨迹,忽略了同一提示下其他尝试中蕴含的丰富信息。

为了克服这一局限,研究团队提出了多轮次在线策略蒸馏(Multi-Rollout On-Policy Distillation, MOPD),这是一种基于同伴条件的蒸馏框架。MOPD的核心创新在于,它利用学生模型在同一提示下生成的多个轨迹(包括成功和失败的尝试)来构建信息更为丰富的教师信号。具体来说,成功的同伴轨迹为正面的推理模式提供证据,而失败的同伴轨迹则提供结构化的负面证据,帮助模型学会避免常见的错误模式。

研究探索了两种同伴上下文构建方式:正面同伴模仿(positive peer imitation)和对比成功-失败条件(contrastive success-failure conditioning)。在竞争编程、数学推理、科学问答和工具使用等多个基准测试上,MOPD均表现出优于标准在线策略基线的性能。进一步的教师信号分析显示,混合成功-失败上下文能更好地使教师评分与验证器奖励对齐,这表明性能提升源于更忠实、实例自适应的监督信号。这些结果证明,有效的在线策略蒸馏应充分利用学生的多轮次试错行为,而非将各轨迹视为孤立样本。

该工作由Weichen Yu等人完成,论文共23页,涵盖机器学习与人工智能领域的研究。实验覆盖了多个具有挑战性的推理任务,充分验证了MOPD框架的通用性和有效性。

展开要点与分析

文章情报

工程师进阶

要点

  • MOPD通过考虑同一提示下的多个尝试(成功和失败)来增强蒸馏。
  • 研究了两种同伴上下文构建:正面模仿和对比成功-失败条件。
  • 在编程、数学推理、科学问答和工具使用基准上均优于标准基线。
  • 混合成功-失败上下文能产生更忠实、实例自适应的教师信号。

要点与分析由自动化流程生成,可能有误,请结合原始来源核实。