3SPO让大模型智能体每步都能高效学习,解决长任务中奖励稀疏难题。
3SPO: State-Score-Supervised Policy Optimization for LLM Agents

- 每步计算状态得分,动态监督策略优化,无需估值函数
- 在ALFWorld和WebShop上比GRPO提升22.6%和15.6分
- 探索效率提升2.4倍,收敛速度加快1.8倍,适合长程决策场景
通过强化学习训练大语言模型作为自主智能体,已在长周期任务中实现超人表现。但现有强化学习算法仅在完整轨迹收集后进行策略优化,粒度粗,在多轮交互中面临奖励稀疏、延迟严重及步骤信用分配困难的问题。本文提出一种新算法——状态-得分-监督策略优化(3SPO),在每一步执行策略优化,并基于历史成功率动态计算状态得分,实现细粒度的信用分配与自适应回放。该方法无需价值函数估计或额外辅助模型。理论上,在每状态老虎机抽象下,所提得分监督机制可实现对数级分配遗憾,并提供动作识别、得分区分性与过滤稳定性的样本复杂度保证。在ALFWorld与WebShop上使用Qwen2.5-1.5B/7B-Instruct模型的实验表明,3SPO持续优于GRPO,在ALFWorld上提升22.6%,在WebShop上提升15.6分,同时资源消耗相当,状态探索量提升2.4倍,收敛速度加快1.8倍。代码已开源。
原文摘要 · Abstract (English)
Training large language models (LLMs) as autonomous agents via reinforcement learning (RL) has enabled frontier models to achieve superhuman performance in long-horizon tasks. However, existing RL algorithms operate at the trajectory level, performing policy optimization only after collecting complete episode rollouts. This coarse-grained approach faces fundamental challenges in multi-turn agent settings where rewards are sparse, delayed, and credit assignment across individual steps is critical. In this work, we propose \textbf{State-Score-Supervised Policy Optimization (3SPO)}, a novel RL algorithm that performs post-step policy optimization with dynamic state score supervision. At each step, 3SPO computes the state score based on historical success rates, supervising step-wise credit assignment, adaptive rollout and post-step policy optimization without requiring value function estimation or additional auxiliary models. Theoretically, under a per-state bandit abstraction, we show that the proposed score-supervised allocation mechanism achieves logarithmic allocation regret and provide sample-complexity guarantees for action identification, score distinguishability, and filtering stability. Experiments on ALFWorld and WebShop with Qwen2.5-1.5B/7B-Instruct show that 3SPO consistently outperforms GRPO by $+22.6\%$ on ALFWorld and $+15.6$ points on WebShop, while using comparable resources to achieve $2.4\times$ more state exploration and $1.8\times$ faster convergence. Code is available at https://github.com/genalyu/3SPO.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。