用共享模型同时生成策略与价值,实现低内存高效强化学习。
SAPO: Single-Rollout Autoregressive Policy Optimization for Agentic Reinforcement Learning

- 共享自回归主干网络,政策与价值函数协同生成
- 在ALFWorld和WebShop上比PPO提升15.1%,比GRPO提升12.1%
- 无需独立评论家模型,训练速度提升33.2%
代理强化学习已成为大语言模型后训练的关键阶段。现有无评论家、组相对方法通过多轮次采样估计策略优势,避免了传统近端策略优化(PPO)的高内存开销,并在长时交互任务中表现优异。然而近期研究揭示三大局限:(1) 缺乏显式值函数泛化与有效时间信用分配;(2) 在复杂长时任务中易出现优势坍缩;(3) 需在采样预算与策略性能间进行昂贵权衡。本文提出单轮次自回归策略优化(SAPO),一种低内存、低计算开销的框架,其中策略与值函数共享单一自回归主干。SAPO利用大语言模型的自回归结构,在不同因果边界上分别生成策略与值预测,共享参数的同时独立优化PPO目标与辅助的在线SARSA目标。为鲁棒估计每一步贡献,引入结合λ收益与批归一化的轨迹级广义优势估计器。在ALFWorld与WebShop上使用Qwen2.5-1.5B/7B的实验表明,SAPO训练稳定,相比PPO平均提升+15.1个百分点,相比GRPO提升+12.1个百分点,同时消除独立评论家模型的内存开销,并将每轮迭代运行时间减少33.2%。
原文摘要 · Abstract (English)
Agentic reinforcement learning (RL) has become a critical stage in the post-training of large language models. Existing critic-free, group-relative methods estimate policy advantages from multiple rollouts, avoiding the substantial memory overhead of conventional proximal policy optimization (PPO) and achieving strong performance on long-horizon interactive tasks. Despite their success, recent studies revealed three limitations: (1) Lack explicit value generalization and effective temporal credit assignment; (2) Suffer from potential advantage collapse in long-horizon complex tasks; (3) Require a costly trade-off between sampling budget and policy performance. In this work, we propose Single-rollout Autoregressive Policy Optimization (SAPO), a low-memory and compute-efficient framework in which the policy and value functions share a single autoregressive backbone. SAPO exploits the autoregressive structure of LLMs to produce policy and value predictions at distinct causal boundaries with shared parameters, while independently optimizing the PPO objectives and auxiliary on-policy SARSA objectives. To robustly estimate the contribution of each turn, we further introduce a trajectory-level generalized advantage estimator that combines lambda-returns with batch normalization. Experiments across ALFWorld and WebShop with Qwen2.5-1.5B/7B show that SAPO trains stably and outperforms PPO and GRPO by mean +15.1 and +12.1 percentage points, respectively, while eliminating the memory cost of a separate critic model and reducing per-iteration runtime by 33.2% over PPO. Project page: https://github.com/dy-liang/SAPO.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。