Progress-conditioned Group Policy Optimization for Long-Horizon Agentic Tasks
The paper proposes Progress-conditioned Group Policy Optimization (ProGPO) to address sampling imbalance in group-based policy optimization for long-horizon agentic tasks. ProGPO assigns higher relative advantages to trajectories or steps that visit more new states when all samples in a group receive zero outcome reward, breaking the self-reinforcing credit trap. Experiments on ALFWorld and WebShop benchmarks with Qwen2.5-1.5/7B-Instruct show consistent improvements over group-based baselines, e
Analysis
TL;DR
- The paper proposes Progress-conditioned Group Policy Optimization (ProGPO) to address sampling imbalance in group-based policy optimization for long-horizon agentic tasks.
- ProGPO assigns higher relative advantages to trajectories or steps that visit more new states when all samples in a group receive zero outcome reward, breaking the self-reinforcing credit trap.
- Experiments on ALFWorld and WebShop benchmarks with Qwen2.5-1.5/7B-Instruct show consistent improvements over group-based baselines, especially on hard tasks.
Why It Matters
This work addresses a critical challenge in training LLM agents for complex, long-horizon tasks where sparse rewards often lead to ineffective learning due to sampling bias. By introducing progress-based advantage assignment during failure-dominated rollouts, ProGPO provides a practical solution to escape local optima and improve sample efficiency—directly impacting how researchers design reward structures and optimization strategies for autonomous agents in real-world environments.
Technical Details
- Group-based policy optimization compares trajectories within a group using outcome rewards but suffers from sampling imbalance: repeated low-effect actions dominate high-probability regions while useful state-changing actions are under-sampled.
- When all samples in a group yield zero reward (all-failed groups), ProGPO switches to first-visit observation coverage as a proxy for progress, assigning higher advantages to steps that explore new states.
- This mechanism breaks the "credit trap" where failure-dominated sampling prevents correction of suboptimal policies, enabling continued exploration even without positive outcome signals.
- Evaluated on two challenging benchmarks—ALFWorld (interactive 3D environment) and WebShop (e-commerce task)—using Qwen2.5-1.5 and Qwen2.5-7B Instruct models, demonstrating consistent gains particularly on harder task variants.
Industry Insight
As AI agents increasingly tackle multi-step, real-world problems with sparse feedback, methods like ProGPO offer a scalable way to maintain exploration pressure without dense reward engineering. Practitioners should consider integrating progress-aware advantage functions into existing group-based RL frameworks when dealing with long-horizon tasks, especially in domains where success is rare but incremental state changes indicate meaningful progress. This approach could significantly reduce training time and improve final performance in production-grade agentic systems.
Disclaimer: The above content is generated by AI and is for reference only.