提出分层奖励机制,让大模型智能体在多步任务中精准获得每一步的反馈。
Group-in-Group Policy Optimization for LLM Agent Training

- 两级奖励结构:先按完整轨迹分组,再通过重复状态回溯构建步骤级分组。
- 相比现有方法,多步任务成功率提升超12%(ALFWorld)和超9%(WebShop)。
- 无需额外模型或采样,内存与计算开销几乎不变,适合实际部署。
近期基于群体的强化学习推动了大语言模型在单轮任务(如数学推理)上的进展,但在多轮智能体训练中仍存在扩展性瓶颈。由于智能体-环境交互持续多步且奖励稀疏延迟,对各步骤的信用分配极为困难。本文提出分组内分组策略优化(GiGPO),一种新型强化学习算法,在保持群体强化学习优点(免评论器、低内存、稳定收敛)的同时实现细粒度信用分配。GiGPO采用两层相对优势估计:(i)在回合级,基于完整轨迹组计算宏观相对优势;(ii)在步骤级,引入锚点状态分组机制,通过跨轨迹识别重复环境状态,将同状态下的动作归为一组,实现微观相对优势估计。该层次化结构无需依赖辅助模型或额外采样,即可同时捕捉全局轨迹质量与局部步骤有效性。我们在复杂基准测试(ALFWorld、WebShop)及搜索增强问答任务上评估了使用Qwen2.5-1.5B/3B/7B-Instruct的性能。结果表明,GiGPO实现了细粒度每步信用信号,在ALFWorld上性能提升超过12%,WebShop上超过9%,并在问答任务上取得42.1%(3B)和47.2%(7B)的准确率,同时维持相同显存开销、相同的LLM回放方式,并几乎无额外时间成本。
原文摘要 · Abstract (English)
Recent advances in group-based reinforcement learning (RL) have driven frontier large language models (LLMs) in single-turn tasks like mathematical reasoning. However, their scalability to multi-turn LLM agent training remains limited. Unlike static tasks, agent-environment interactions unfold over many steps and often yield sparse or delayed rewards, making credit assignment across individual steps significantly more challenging. In this work, we propose Group-in-Group Policy Optimization (GiGPO), a novel RL algorithm that achieves fine-grained credit assignment for LLM agents while preserving the appealing properties of group-based RL: critic-free, low memory, and stable convergence. GiGPO introduces a two-level structure for estimating relative advantage: (i) At the episode-level, GiGPO computes macro relative advantages based on groups of complete trajectories; (ii) At the step-level, GiGPO introduces an anchor state grouping mechanism that retroactively constructs step-level groups by identifying repeated environment states across trajectories. Actions stemming from the same state are grouped together, enabling micro relative advantage estimation. This hierarchical structure effectively captures both global trajectory quality and local step effectiveness without relying on auxiliary models or additional rollouts. We evaluate GiGPO on challenging agent benchmarks, including ALFWorld and WebShop, as well as tool-integrated reasoning on search-augmented QA tasks, using Qwen2.5-1.5B/3B/7B-Instruct. Crucially, GiGPO delivers fine-grained per-step credit signals, achieves performance gains of > 12% on ALFWorld and > 9% on WebShop over GRPO, and obtains superior performance on QA tasks (42.1% on 3B and 47.2% on 7B): all while maintaining the same GPU memory overhead, identical LLM rollout, and incurring little to no additional time cost.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。