DRIFT: 解耦轨迹展开与重要性加权微调以实现高效多轮优化
摘要
本文提出DRIFT框架,该框架结合离线轨迹与重要性加权监督微调,高效实现与强化学习相当的多轮交互学习性能。
查看缓存全文
缓存时间: 2026/06/01 07:18
论文页面 - DRIFT:解耦轨迹展开与重要性加权微调实现高效多轮优化
来源:https://huggingface.co/papers/2605.31455 发布于 5月29日
·
由https://huggingface.co/mujianijan 提交
mj (https://huggingface.co/mujianijan)于 6月1日
摘要
DRIFT 是一个框架,它结合离线轨迹与重要性加权监督微调,实现了与强化学习相当的多轮交互学习效率与性能。
大型语言模型越来越多地部署于多轮交互场景,其中用户或环境可以迭代地提供轻量级反馈。然而,在实践中优化此类行为面临一个尖锐的两难困境:在线强化学习 (https://huggingface.co/papers?q=online%20reinforcement%20learning) 能够有效处理多轮动态 (https://huggingface.co/papers?q=multi-turn%20dynamics),但代价高昂,因为每次更新都需要生成完整的修正轨迹;而离线监督微调 (https://huggingface.co/papers?q=offline%20supervised%20fine-tuning) (SFT) 虽然高效,却面临分布偏移和行为崩溃 (https://huggingface.co/papers?q=behavioral%20collapse) 的问题。为此,我们创新性地提出了 DRIFT(解耦轨迹展开 (https://huggingface.co/papers?q=Rollouts) 与重要性加权微调 (https://huggingface.co/papers?q=Importance-Weighted%20Fine-Tuning)),该框架将 KL 正则化 RL 目标 (https://huggingface.co/papers?q=KL-regularized%20RL%20objective) 等价于重要性加权监督学习 (https://huggingface.co/papers?q=importance-weighted%20supervised%20learning) 的理论洞见付诸实践。DRIFT 通过从固定参考策略 (https://huggingface.co/papers?q=reference%20policy) 采样离线交互轨迹、推导基于回报的重要性权重 (https://huggingface.co/papers?q=importance%20weights),并在所得数据集上通过加权 SFT 优化策略,从而将轨迹展开与优化解耦。实验表明,DRIFT 在多轮强化学习基准上达到了匹配或更优的性能,同时保持了标准监督微调的训练效率与简洁性。代码已开源至 https://github.com/2020-qqtcg/DRIFT。
查看 arXiv 页面 (https://arxiv.org/abs/2605.31455)查看 PDF (https://arxiv.org/pdf/2605.31455)GitHub0 (https://github.com/2020-qqtcg/DRIFT)添加到收藏 (https://huggingface.co/login?next=%2Fpapers%2F2605.31455)
在你的智能体中获取此论文:
hf papers read 2605.31455
未安装最新 CLI?curl -LsSf https://hf.co/cli/install.sh | bash
引用此论文的模型0
暂无模型链接此论文
请在模型 README.md 中引用 arxiv.org/abs/2605.31455 以从此页面链接。
引用此论文的数据集0
暂无数据集链接此论文
请在数据集 README.md 中引用 arxiv.org/abs/2605.31455 以从此页面链接。
引用此论文的 Spaces0
暂无 Space 链接此论文
请在 Space README.md 中引用 arxiv.org/abs/2605.31455 以从此页面链接。
包含此论文的收藏0
暂无收藏包含此论文
请将此论文添加至一个收藏 (https://huggingface.co/new-collection) 以从此页面链接。
相似文章
CoDrift:离线强化学习的组合漂移框架
CoDrift是一种用于离线强化学习的组合框架,它将行为目标和价值目标结合成一个统一的策略场,在OGBench和D4RL等基准测试中达到了最先进的性能。
漂移约束优化:微调指令模型时只有方向重要
本文提出了一种用于微调指令模型的漂移约束优化方法,着重于更新方向以提升目标任务性能,同时保留现有能力。实验表明,层选择性探测可以在多种语言中增强科学推理和多语言翻译能力。
用于高效多轮智能体微调的依赖感知轨迹优化
本文提出了一种依赖感知轨迹优化方法,用于高效多轮智能体微调,以提高准确度并降低推理成本。
Drift Q-Learning
提出了DriftQL,它结合了基于漂移的行为正则化器与评论家驱动的策略改进,用于离线强化学习,在D4RL和OGBench上优于扩散和流方法,同时保持简单性和效率。
RTPO:用于稳定智能体强化学习训练的反序回合策略优化
论文提出了反序回合策略优化(RTPO)来稳定多轮智能体强化学习训练,通过将展开组织为反序树并按反序执行回合级更新,展示了相较于基线的改进。