Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Hierarchy-of-Groups Policy Optimization for Long-Horizon Agentic Tasks

About

Group-based reinforcement learning (RL), such as GRPO, has advanced the capabilities of large language models on long-horizon agentic tasks. To enable more fine-grained policy updates, recent research has increasingly shifted toward stepwise group-based policy optimization, which treats each step in a rollout trajectory independently while using a memory module to retain historical context. However, we find a key issue in estimating stepwise relative advantages, namely context inconsistency, where steps within the same group may differ in their historical contexts. Empirically, we reveal that this issue can lead to severely biased advantage estimation, thereby degrading policy optimization significantly. To address the issue, in this paper, we propose Hierarchy-of-Groups Policy Optimization (HGPO) for long-horizon agentic tasks. Specifically, within a group of rollout trajectories, HGPO assigns each step to multiple hierarchical groups according to the consistency of historical contexts. Then, for each step, HGPO computes distinct advantages within each group and aggregates them with an adaptive weighting scheme. In this way, HGPO can achieve a favorable bias-variance trade-off in stepwise advantage estimation, without extra models or rollouts. Evaluations on two challenging agentic tasks, ALFWorld and WebShop with Qwen2.5-1.5B-Instruct and Qwen2.5-7B-Instruct, show that HGPO significantly outperforms existing agentic RL methods under the same computational constraints. Code is available at https://github.com/langfengQ/verl-agent/tree/master/recipe/hgpo.

Shuo He, Lang Feng, Qi Wei, Xin Cheng, Lei Feng, Bo An• 2026

Related benchmarks

TaskDatasetResultRank
Information RetrievalBEIR
SciFact0.764
120
Online ShoppingWebShop (test)
Score88.4
59
Macro-average Exact MatchShellOps-Pro
Macro-average Exact Match Accuracy22.8
36
Multi-turn Agent InteractionALFWorld (test)
Success Rate (Pick)92.3
31
Reasoning-based RetrievalBRIGHT 1.0 (test)
NDCG@10 (Bio.)61.77
16
File EditingShellOps
Exact Match11.9
9
Agentic task solvingShellOps
Pass@30.219
9
Agentic task solvingAB-OS
Pass@362.8
9
Agentic task solvingAB-DB
Pass@337.8
9
Agentic task solvingEHRCon
Pass@371.1
9
Showing 10 of 29 rows

Other info

Follow for update