基于前缀重放的多轮在线策略蒸馏
摘要
本文提出 ReOPD,一种用于 LLM 智能体的在线策略蒸馏方法,该方法重用预先收集的教师轨迹作为重放前缀,无需新的环境交互即可提高效率和准确性。
查看缓存全文
缓存时间: 2026/07/24 13:01
论文页面 - 多轮同策略蒸馏与前缀重放
来源:https://huggingface.co/papers/2607.04763
摘要
我们研究了用于智能体任务中的同策略蒸馏(On-Policy Distillation, OPD),其中LLM智能体与环境进行多轮交互,学生模型在这些多轮交互历史中模仿教师模型。完全在线的OPD成本高昂,因为每次更新都需要学生模型通过环境进行全新的 rollout,并在访问过的历史状态中查询教师模型。我们提出重放前缀同策略蒸馏(Replayed-Prefix On-Policy Distillation, ReOPD),这是一种脱离环境的替代方案,它重用预先收集的教师轨迹作为重放前缀:学生模型在选定的步骤执行动作,同时教师模型提供密集的逐步监督,而无需执行新的环境交互。我们发现多轮OPD引入了一个前缀陷阱:使历史更符合学生策略可以提高与学生模型的相关性,但可能会在教师目标不可靠的历史上查询教师模型。这导致了学生占据分布与教师可靠性之间的双向分布偏移。ReOPD通过将多轮OPD视为可靠性感知的前缀分布设计来解决这一问题,并采用简单的逐步衰减采样调度,重点关注早期、偏移较小的前缀。在使用Python和搜索环境进行数学推理的多个教师和学生模型规模上,ReOPD保持或提高了OPD级别的准确性,在学生训练过程中零工具调用,且每次rollout至少比OPD快4倍。因此,ReOPD将昂贵的智能体-环境交互转化为可重用的离线资源,实现了跨工具、任务和环境的可扩展蒸馏。
查看arXiv页面 (https://arxiv.org/abs/2607.04763)查看PDF (https://arxiv.org/pdf/2607.04763)项目页面 (https://baohaoliao.github.io/ReOPD/)GitHub4 (https://github.com/BaohaoLiao/ReOPD)添加到收藏 (https://huggingface.co/login?next=%2Fpapers%2F2607.04763)
在你的智能体中获取这篇论文:
hf papers read 2607\.04763
没有最新的CLI?curl \-LsSf https://hf\.co/cli/install\.sh \| bash
引用该论文的模型0
没有模型链接到该论文
请在模型README.md中引用arxiv.org/abs/2607.04763,以便从本页链接。
引用该论文的数据集0
没有数据集链接到该论文
请在数据集README.md中引用arxiv.org/abs/2607.04763,以便从本页链接。
引用该论文的Space0
没有Space链接到该论文
请在Space README.md中引用arxiv.org/abs/2607.04763,以便从本页链接。
包含该论文的收藏集1
相似文章
传递接力棒:轨迹中继在策略蒸馏
提出中继在策略蒸馏(Relay-OPD),通过在检测到的交接触发点让教师短暂接管以纠正推理轨迹,解决了在策略蒸馏中的前缀失败问题。在数学推理基准上实现一致改进,并将训练轨迹长度减少超过50%。
TurnOPD: 使在线策略蒸馏对回合感知以实现高效长程智能体训练
TurnOPD 引入了回合级别的预算机制用于长程智能体的在线策略蒸馏,通过自适应 rollout 深度和渐进式回合归一化损失预算解决了传统 OPD 的低效问题,在相同训练预算下实现了更高的准确性。
ShortOPD:通过短到长在策略蒸馏恢复剪枝后的大语言模型
ShortOPD提出了一种短到长的在策略蒸馏方案,通过将训练集中在有效前缀上,恢复用于自由形式生成的剪枝后大语言模型,相较于未恢复模型实现了高达9倍的改进,并以四分之一的训练时间达到了长视界蒸馏的效果。
OPMD:超越输出监督的在线策略机制蒸馏
本文提出了在线策略机制蒸馏(OPMD),在在线策略蒸馏过程中迁移与学生模型架构兼容的教师中间计算,在多种师生组合中均超越了仅基于输出监督的蒸馏基线。
过滤后重加权:重新思考在线策略蒸馏中的优化粒度
介绍FiRe-OPD,一种用于大语言模型在线策略蒸馏的方法,它过滤低质量轨迹并应用软重加权来强调信息丰富的令牌,在强到弱、单教师和多教师设置中实现了改进的性能。