长程智能体强化学习的采样成本优化路线

长程语言智能体做强化学习有个结构性矛盾:策略只从轨迹末端的标量奖励里学东西,中间每句命令、每次观测、每个状态转移都白白流走。想补监督,常规两条路——构造价值目标估计每步成功概率,或构造观测目标拟合环境下一步返回什么——前者信号区分度低,后者又长又噪还只能往前看一步。更底层的约束是经济学的:rollout 成本随步数线性甚至超线性上涨,预算有限只能压采样量,而末端奖励又需要足够多成功样本才能组内对比。省采样就是省算力,这是所有长程训练的公共瓶颈。

CoPE 的答案:只加一类目标函数

论文 Co-Evolving Acting and Forecasting(arXiv:2610.20597)的做法很克制:不动架构、不加模型、不做独立预训练,引入第三类监督——多步动作目标。让同一个策略主干在训练时同时学"现在该做什么"和"往后几步大致怎么走",推理时仍一步一个动作,零前瞻开销。

核心机制四步。第一,只在成功轨迹上构造预测目标——失败轨迹里没有值得模仿的后续序列。第二,跳过中间观测直接预测动作块:给定历史,让主干直接生成往后 K 步的动作序列,中间观测不参与。这是它与观测目标类方法(如 ECHO)的分水岭:动作短、离散、决策相关;观测长、含噪、混着环境随机性。第三,时间集成——同一个动作被执行时和被前 K 步历史预测出时各监督一次,多条历史指向同一标签,等价于从不同视角喂给共享主干。第四,部署零改动。

# 伪代码:CoPE 的混合目标
loss_pg = grpo_loss(actions, group_rewards)      # 末端奖励
for t in success_trajectories:                    # 仅成功轨迹
    for k in range(1, K+1):
        target = actions[t : t+K]                 # 动作块,跳观测
        logp = policy(h[t-k:t], target)           # 同一主干同一前向
        loss_forecast += normalize(-logp) * valid_mask(target)
loss = loss_pg + lambda_cope * loss_forecast

两个稳定化细节别漏:不合法的预测动作不产生梯度;预测损失要做归一化,避免多步目标的量级盖过策略梯度。

数据与适用边界

论文自报(无第三方复现):样本效率上相对 GRPO 最高约 2 倍,更早摸到基准分数;具体点位上 ALFWorld 成功率 95.5% 对基线 90.5%;应用型基准 AppWorld 普通集从 41.7% 提到 55.4%。训练开销方面,预测通路每步多约 11%,但 rollout 更短让端到端每步时间反而少 9%。注意 τ²-bench 提升有限且有子项回退(航空项下降),收益并非在所有任务形态上都稳定。

放在路线图里看:观测目标类(ECHO)复用前向但混入环境噪声;动作目标类(CoPE)绕开了预测环境本身的难度,代价是依赖成功轨迹、冷启动阶段无信号可用;系统层方案(动态上下文预算、轨迹去重)不碰监督信号,只是把同样的样本喂得更快——两者互补不冲突。

落地清单

适合的场景有三个特征:可重置、可复现、有明确成功判据——网页操作、家务规划、应用编排、终端工具调用都算。不适合:无法判定成功轨迹的任务、环境不可重置、以及瓶颈在外部工具 API 账单而非 GPU 的负载——那里 CoPE 帮不上忙。

接入现有 GRPO 管线按五条走:先加一路预测损失,K 从 2 到 4 起步别贪大;只对成功轨迹开预测梯度;预测损失量级失控先调归一化系数而不是关通路;主指标用"达到同一分数所需 rollout 数"而非最终分——样本效率才是卖点;最后认真看子项回退,判断收益来自方法还是任务分布。验收时构造一组已知成功轨迹回放,确认预测损失确实收敛、推理行为无回归,再扩大训练规模。

0

评论0

请先
显示验证码
没有账号?注册  忘记密码?