PlanPO: Group Planning-Aware Policy Optimization for Multi-Turn Agentic LLMs
Authors: Dayang Liang, Liyuan He, Xuan Feng, Shuxin Li, Bo An, Yunlong Liu
Organizations: Department of Automation, Xiamen University, Xiamen, China · School of Artificial Intelligence, Shanghai Jiaotong University, Shanghai, China · College of Cyberspace Security, Jinan University, Guangzhou, China · College of Computing and Data Science, Nanyang Technological University, Singapore
Group-relative policy optimization has emerged as a key paradigm for training agentic large language models (LLMs) on multi-turn interactive tasks. However, most existing variants fail to distinguish advantages among successful trajectories even when these trajectories differ substantially in their interaction efficiency. For instance, circuitous successes are often assigned the identical outcome reward, causing advantage collapse and severe performance bottlenecks. To this end, we propose Group Planning-aware Policy Optimization (PlanPO), a simple yet effective RL method for learning generalizable planning abilities beyond task-specific high-quality behavior patterns. Specifically, PlanPO introduces coarse-to-fine advantage signals, which capture the relative differences in trajectory-level lengths and turn-level response lengths conditioned on successful trajectories sampled for the same task. Within the group-relative optimization structure, this enables agents to actively learn generalizable and deliberate behaviors spanning interaction planning and textual generation from high-quality rollouts, without degenerating into vanilla length minimization. Experimentally, PlanPO improves over GRPO by 27.2% on average across the challenging multi-turn benchmarks ALFWorld, WebShop, and SciWorld, outperforming recent powerful baselines while incurring negligible additional training cost.
Figures & tables
Figure 1: Left: Successful rollouts are not equally informative. Rollouts that reach the same task goal, can differ substantially in length, directness, and reasoning quality, while the most optimal strategies may reveal more sophisticated and generalizable capabilities. Right: Averaged normalization performance comparison across ALFWorld, WebShop, and SciWorld environments using the Qwen2.5-1.5b model.
Figure 2: Overview of PlanPO. Left : Rollouts sampled for the same task can reach the same outcome through interaction paths and textual responses of markedly different quality. Middle : PlanPO converts the outcome into success-conditioned trajectory- and turn-level scores using trajectory and response lengths, respectively. Right : The two scores are normalized separately within the same-task group and combined into a coarse-to-fine advantage for policy optimization.
Type
Method
ALFWorld
WebShop
Pick
Look
Clean
Heat
Cool
Pick2
All
Score
Succ.
Closed-Source Model
Prompting
GPT-4o
75.3
60.8
31.2
56.7
21.6
49.8
48.0
31.8
23.7
Prompting
Gemini-2.5-Pro
92.8
63.3
62.1
69.0
26.6
58.7
60.3
42.5
35.9
Qwen2.5-1.5B-Instruct
Prompting
ReAct
17.4
20.5
15.7
6.2
7.7
2.0
12.8
40.1
11.3
Table 1: Evaluation results on ALFWorld and WebShop. For each RL training method, we report the mean and variance over three random seeds. The ALFWorld contains six categories: Pick & Place (Pick), Examine in Light (Look), Clean & Place (Clean), Heat & Place (Heat), Cool & Place (Cool), and Pick Two & Place (Pick2). Most entries in this table are reported by Feng et al. ( Feng et al. 2026 ) . Notably, the baseline GiGPO w/ std denotes Fnorm=std , and GiGPO w/o std denotes Fnorm=1 .
Type
Model
Measure
Test-Cond.
Find
Chem-Mix
Lifespan
Overall
Proprietary
GPT-4o
15.09
6.02
38.64
20.00
73.33
21.00
Qwen-Max
9.43
0.00
34.09
20.00
40.00
13.50
Gemini-2.5-Flash
11.32
0.00
54.55
0.00
80.00
21.00
OpenAI o4-mini
20.75
14.46
47.73
0.00
100.00
29.50
OpenAI o3
47.17
25.30
56.82
40.00
66.67
41.50
Gemini-2.5-Pro
9.43
0.00
29.55
0.00
46.67
12.50
Table 2: Evaluation results on SciWorld over three random seeds. The task Test-Cond. means test conductivity and Chem-Mix means chemistry mix. Baseline entries are reported by Zhang et al. ( Zhang et al. 2026b ) .
Figure 3: Left two panels: Ablation study on the weight schedule α(k) with αinit=2αfinal in PlanPO. Third panel: Component ablation. We compared the ablation results of PlanPO with default decay , trajectory-level advantage AE only , turn-level advantage AS only , and PlanPO with constant α=0.1 . Right panel: Runtime breakdown of each PlanPO iteration on four NVIDIA A40 GPUs. Compared to the GRPO, the additional computation introduced by PlanPO lies in the reward and advantage stages, which is negligible. Conversely, PlanPO can reduce overall task runtime by 12.5%
Type
Method
In-Success
Out-Success
Prompting
GPT-4o
48.0
46.0
Prompting
Gemini2.5
60.3
50.5
RL Training
PPO
54.4 ±3.1
50.9 ±7.6
RL Training
RLOO
69.7 ±2.5
68.7 ±10.7
RL Training
GRPO
72.8 ±3.6
70.1 ±2.5
RL Training
GiGPO
86.7 ±1.7
82.4 ±2.0
Table 3: Generalization evaluation with three seeds in ALFWorld using Qwen2.5-1.5B-Instruct. Some entries are reported by He et al. ( He et al. 2026 ) . The In-Success reports the success rate on the in-distribution tasks, while the Out-Success reports the success rate on out-of-distribution tasks.
Figure 4: Length-normalized Comparison Analysis with different reward shaping settings in ALFWorld.
Figure 5: Mean Length Comparison in ALFWorld tasks.
Group-based policy optimization has been increasingly used to train large language model (LLM) agents from sparse outcome rewards by comparing trajectories or steps within a group. However, on difficult long-horizon agentic tasks, we found this comparison suffers from a sampling imbalance: repeated or low-effect actions dominate the high-probability region of the policy, leaving useful state-changing actions under-sampled. This imbalance produces many all-failed rollout groups, where the relative advantages within such a group provide no direction for updating the policy. Consequently, the sampling imbalance and the all-failed groups collapse into a self-perpetuating credit trap: failure-dominated sampling yields no learning signal, allowing repeated low-effect actions to persist. To break this trap, we propose Coverage-conditioned Group Policy Optimization (CovGPO), which uses first-visit observation coverage only when all samples in a group receive zero outcome reward. Specifically, within such groups, CovGPO assigns higher relative advantages to state-exploring actions without corrupting the original reward signal, as reaching new observations is a prerequisite for task success. Experiments on two challenging agentic benchmarks, ALFWorld and WebShop, show that CovGPO consistently improves over group-based baselines, with particularly large gains on hard tasks.
Kaibing Yang, Guangfeng Cai, Shengtian Yang +6
Southeast University · Kuaishou Technology · Meituan
Group-based Reinforcement Learning (RL) has significantly enhanced Large Language Models (LLMs) in agentic scenarios. To achieve finer-grained policy updates, recent agentic RL frameworks have shifted from trajectory-level to step-level training. However, long-horizon agentic RL suffers from severe reward sparsity and delay, as feedback is often deferred for dozens of interaction steps. While existing step-level frameworks refine training granularity, their credit assignment remains coarse-grained and still treats agent exploration as isolated, linear trajectories. This oversimplified perspective ignores the inherent graph structure of state transitions, leading to high-variance state-value estimation and myopic, localized credit assignment. To overcome these critical bottlenecks, we propose Group-Graph Policy Optimization (G2PO), a novel group-based RL algorithm tailored for multi-turn agentic tasks. G2PO explicitly transforms linear interaction trajectories into a global state-transition graph. By aggregating identical observations across different trajectories, we introduce group-aggregation state-value estimation that reduces sampling variance and trajectory-dependent bias. Furthermore, we redefine agent actions as transitions between state nodes and propose an edge-centric advantage estimation strategy. By globally standardizing Temporal Difference (TD) errors across the entire graph, G2PO explicitly identifies and prioritizes critical transitions that drive absolute task progress. Extensive experiments on representative long-horizon benchmarks-WebShop, ALFWorld, and AppWorld-demonstrate that G2PO substantially outperforms state-of-the-art prompt-based and RL baselines, achieving remarkable success rate improvements of up to 22.2% over GRPO.
Reinforcement learning for agentic large language models (LLMs) typically relies on a sparse, trajectory-level outcome reward, making it difficult to evaluate the contribution of individual tool-calls within multi-turn interactions. Existing approaches to such process credit assignment either depend on separate external process reward models that introduce additional consumption, or tree-based structural rollout that merely redistributes the outcome signal while constraining trajectory diversity. A promising alternative leverages the per-turn change in the policy's predicted probability of the ground-truth, termed Information Gain (IG), as an intrinsic process signal without an external evaluator. However, prior work on leveraging IG signals within the RL training loop faces three systematic challenges: normalizing across turns that face heterogeneous positional contexts can distort the relative standing of individual turns, accumulating a variable number of terms causes advantage magnitudes to drift with trajectory depth, and a fixed clipping range governs policy updates identically for turns with vastly different IG signals. In this paper, we propose A2TGPO (Agentic Turn-Group Policy Optimization with Adaptive Turn-level Clipping), which retains IG as the intrinsic signal but re-designs how it is normalized, accumulated, and consumed: (i) turn-group normalization: normalizes IG within each (prompt, turn-index) group so that each turn is compared only against peers at the same interaction depth; (ii) variance-rescaled discounted accumulation: divides cumulative normalized IG by square root of accumulated terms to keep advantage magnitudes comparable across turn positions; and (iii) adaptive turn-level clipping: modulates each turn's clipping range based on its normalized IG, widening the update region for informative turns and narrowing it for uninformative ones.
Dingwei Chen, Zefang Zong, Zhipeng Ma +5
Tencent Inc · The Chinese University of Hong Kong · Shenzhen MSU-BIT University