基于前缀重放的多轮在线策略蒸馏

Hugging Face Daily Papers 论文

摘要

本文提出 ReOPD,一种用于 LLM 智能体的在线策略蒸馏方法,该方法重用预先收集的教师轨迹作为重放前缀,无需新的环境交互即可提高效率和准确性。

我们研究针对智能体任务的在线策略蒸馏(OPD),其中 LLM 智能体与环境进行多轮交互,学生模型在这些多轮交互历史中模仿教师模型。完全在线 OPD 成本高昂,因为每次更新都需要学生通过环境进行全新的 rollout,并在访问过的历史状态下查询教师。我们提出基于重放前缀的在线策略蒸馏(ReOPD),这是一种脱离环境的替代方法,它重用预先收集的教师轨迹作为重放前缀:学生在选定的步骤上行动,而教师提供密集的逐步骤监督,无需执行新的环境交互。我们证明多轮 OPD 引入了一个前缀陷阱:使历史更符合学生在线策略可以提高与学生的相关性,但可能在某些历史状态下查询教师时,其目标不可靠。这造成了学生占据分布与教师可靠分布之间的双向分布偏移。ReOPD 通过将多轮 OPD 视为一个可靠性感知的前缀分布设计来解决这个问题,并采用一个简单的逐步衰减采样调度来实现,该调度强调早期、低偏移的前缀。在使用 Python 进行数学推理以及搜索环境中的多个教师和学生模型规模的实验中,ReOPD 保持或提高了 OPD 级别的准确率,在学生训练期间使用零工具调用,且每次 rollout 比 OPD 快至少 4 倍。因此,ReOPD 将昂贵的智能体-环境交互转化为可重用的离线资源,实现了跨工具、任务和环境的可扩展蒸馏。
查看原文
查看缓存全文

缓存时间: 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

相似文章

传递接力棒:轨迹中继在策略蒸馏

arXiv cs.CL

提出中继在策略蒸馏(Relay-OPD),通过在检测到的交接触发点让教师短暂接管以纠正推理轨迹,解决了在策略蒸馏中的前缀失败问题。在数学推理基准上实现一致改进,并将训练轨迹长度减少超过50%。

ShortOPD:通过短到长在策略蒸馏恢复剪枝后的大语言模型

arXiv cs.LG

ShortOPD提出了一种短到长的在策略蒸馏方案,通过将训练集中在有效前缀上,恢复用于自由形式生成的剪枝后大语言模型,相较于未恢复模型实现了高达9倍的改进,并以四分之一的训练时间达到了长视界蒸馏的效果。

OPMD:超越输出监督的在线策略机制蒸馏

Hugging Face Daily Papers

本文提出了在线策略机制蒸馏(OPMD),在在线策略蒸馏过程中迁移与学生模型架构兼容的教师中间计算,在多种师生组合中均超越了仅基于输出监督的蒸馏基线。