← 返回信息流
技术博客arXiv cs.AI·1 天前

RODS:基于奖励驱动的在线数据合成优化多轮工具使用智能体

原标题:RODS: Reward-Driven Online Data Synthesis for Multi-Turn Tool-Use Agents

速览

针对多轮工具使用强化学习中静态数据集信息样本快速枯竭的瓶颈,研究提出RODS(Reward-driven Online Data Synthesis)方法。该方法利用进度奖励方差作为零成本的边界检测器,持续识别能力边界样本并合成新的多轮变体,构建与策略共同演进的动态重放缓冲区。实验表明,RODS仅需约20倍更少的轨迹即可达到与17K样本离线流水线相当的性能,显著优于固定数据RL和环境增强方法。

AI 深度解读

RODS:基于奖励驱动的在线数据合成,破解多轮工具使用智能体的数据瓶颈

背景

在大型语言模型(LLM)的智能体(Agent)应用中,多轮工具使用(Multi-turn Tool-Use) 是一个极具挑战性但也至关重要的场景。智能体需要通过与外部 API、数据库或软件工具进行多次交互,才能完成复杂的任务(如编写代码并执行、查询多个数据源并汇总等)。

目前,提升这类智能体能力的主流方法是强化学习(Reinforcement Learning, RL)。然而,现有的 RL 训练流程面临一个严重的瓶颈:静态数据集的迅速枯竭

在传统的离线 RL 训练管线中,模型依赖于预先收集好的静态数据集。随着训练的深入,模型会逐渐掌握数据集中简单的样本,导致这些样本提供的梯度信号(Gradient Signal)变得微弱甚至为零。真正能推动模型进步的是那些处于“能力边界”的样本——即那些模型有时能成功、有时会失败的中等难度样本。但在静态数据集中,这类高信息量的样本是有限的,一旦模型掌握了它们,训练效率就会急剧下降。

此外,现有的数据增强方法往往成本高昂,或者无法保证新生成数据的结构复杂性(如 API 调用拓扑和依赖深度)与原始难题相匹配。

核心内容

为了解决上述问题,研究团队提出了 RODS(Reward-Driven Online Data Synthesis,基于奖励驱动的在线数据合成) 框架。RODS 的核心思想是关闭强化学习训练与数据生成之间的闭环,利用训练过程中的实时反馈来动态生成高质量数据。

1. 理论洞察:梯度信号与 Popoviciu 上界

研究团队观察到,在采用 GRPO(Group Relative Policy Optimization,组相对策略优化) 等算法时,梯度信号主要集中在那些具有最高 rollout 奖励方差的任务上。

这可以用统计学中的 Popoviciu 上界定理 来解释:对于有界随机变量,其方差在取值概率分布接近 0.5(即成功与失败概率大致平衡)时达到最大。这意味着:

  • 太简单的任务(几乎总是成功)和太复杂的任务(几乎总是失败)提供的梯度信号都很小。
  • 处于智能体能力边界附近的样本(成功率和失败率各半),对策略梯度的贡献不成比例地大。

随着训练的进行,智能体的能力边界不断外移,原本处于边界的样本变成了简单样本,而静态数据集中新的、同等难度的样本却不再存在。这就是静态数据集“枯竭”的根本原因。

2. RODS 的工作机制

RODS 通过以下三个步骤解决这一枯竭问题:

  • 零成本的边界检测器: RODS 利用训练过程中已经计算出的 rollout 奖励方差作为实际且零成本的边界检测器。它不需要额外的推理步骤,只需监控当前训练批次中奖励的方差,即可识别出哪些样本正处于“能力边界”上。

  • 技能对齐的重采样管线(Skill-Aligned Resampling Pipeline): 一旦识别出边界样本,RODS 会合成新的多轮变体。关键在于,这些新样本必须保持与原始边界样本相同的结构复杂性(例如,API 调用的拓扑结构和依赖深度)。通过技能对齐重采样,确保新生成的数据既具有挑战性,又在智能体可学习的范围内。

  • 动态回放缓冲区(Dynamic Replay Buffer): RODS 维护一个动态的回放缓冲区,该缓冲区与策略共同进化。它不断用新合成的边界样本替换那些已经变得太简单或太难的旧样本,确保持续有足够的高信息量样本供模型学习。

3. 实验结果

在受控实验设置下,RODS 展示了显著的效率优势:

  • 数据效率极高:仅从 400 个人工种子数据(Human Seeds)开始,并维持约 800 个活跃训练样本的池子,RODS 就达到了与使用 17,000 个样本的离线管线相当的性能。
  • 轨迹节省:所需的训练轨迹数量减少了约 20 倍
  • 性能超越:在固定数据 RL 和环境增强(Environment Augmentation)基线之上,RODS 均实现了性能提升。

关键要点

  • 痛点明确:多轮工具使用 RL 训练的主要瓶颈是静态数据集中高信息量样本(即处于能力边界的样本)的快速耗尽。
  • 理论支撑:基于 Popoviciu 上界,GRPO 等算法的梯度信号自然集中于奖励方差最大的样本,即成功与失败概率平衡的“边界样本”。
  • 创新方法:RODS 利用现有的 rollout 奖励方差作为边界检测器,无需额外推理成本。
  • 数据合成策略:通过技能对齐重采样,生成结构复杂性(API 拓扑、依赖深度)匹配的新多轮样本,而非随机生成。
  • 动态管理:引入动态回放缓冲区,使训练数据池随模型能力进化,确保持续提供高梯度信号。
  • 显著成效:用 400 个种子样本和 ~800 个活跃样本,实现了与 17K 样本离线管线相当的性能,轨迹消耗减少 20 倍。

意义与影响

RODS 的提出对智能体训练领域具有深远意义:

  1. 大幅降低数据获取成本:传统方法依赖大规模人工标注或昂贵的离线数据收集。RODS 证明了仅需少量人工种子数据,即可通过在线合成实现高性能训练,极大地降低了构建复杂工具使用智能体的门槛。
  2. 提升训练效率:通过精准定位并持续提供“最佳难度”样本,RODS 避免了模型在简单样本上的无效训练,显著提升了每一步训练的价值。
  3. 推动在线学习范式:RODS 展示了如何将 RL 训练与数据生成紧密耦合,为其他需要复杂交互环境的智能体任务(如机器人控制、自动化代码生成)提供了新的数据合成思路。
  4. 解决长尾分布问题:在工具使用中,许多复杂场景是长尾分布的。RODS 的动态边界检测机制能够自动发现并聚焦于这些难例,有助于提升模型在复杂、罕见场景下的鲁棒性。

总之,RODS 不仅是一个新的算法框架,更是一种高效利用计算资源、以数据驱动方式突破智能体能力边界的新范式。

查看原文 →arxiv.org