New Policy Optimization Improves Multi-Turn LLM Agents

Hao Dou· July 21, 2026 View original

Summary

This paper introduces Contextual Information-Gain Policy Optimization (CIGPO) to stabilize and enhance the training of multi-turn evidence-reading LLM agents, addressing the issue of reward-variance collapse in outcome-only reinforcement learning. CIGPO uses per-turn information-gain rewards to prevent optimization deadlocks and significantly boosts performance on tasks like HotpotQA.

Training multi-turn evidence-reading agents using outcome-only reinforcement learning often leads to instability because intermediate steps receive insufficient direct feedback. A common issue, observed with GRPO on HotpotQA, is a "zero-advantage lock-in" where all sampled trajectories receive the same minimum penalty, causing relative advantages to vanish and optimization to stall. To overcome this, researchers propose Contextual Information-Gain Policy Optimization (CIGPO). This method injects variance into the reward signal by assigning per-turn rewards to intermediate evidence-reading steps. The per-turn signal is derived from the marginal increase in a frozen reference model's log-likelihood of the ground-truth answer, effectively preserving the reward variation needed for effective policy optimization. CIGPO, combined with separate normalization of information-gain and F1 rewards and an information-gain weight curriculum, achieved significant performance gains on HotpotQA. It reached a standard F1 score of 0.518, a substantial improvement over the best GRPO checkpoint and completely avoiding the optimization deadlock. This highlights reward-variance collapse as a critical failure mode and demonstrates the effectiveness of turn-level information-gain rewards.

Why it matters

For professionals developing advanced LLM agents, particularly those requiring multi-turn reasoning and evidence synthesis, CIGPO offers a robust training methodology to achieve higher performance and stability, making these agents more reliable for complex tasks.

How to implement this in your domain

  1. 1Adopt CIGPO or similar variance-injection strategies when training multi-turn LLM agents with reinforcement learning.
  2. 2Design per-turn reward mechanisms that provide granular feedback, such as information gain, to intermediate steps.
  3. 3Implement separate normalization and curriculum learning for different reward components to optimize training.
  4. 4Diagnose reward-variance collapse as a potential failure mode in outcome-only RL for sequential tasks.

Who benefits

AI DevelopmentCustomer ServiceLegalTechHealthcare (for information retrieval)Research & Development

Key takeaways

  • Outcome-only RL can lead to reward-variance collapse and optimization deadlocks in multi-turn LLM agents.
  • CIGPO introduces per-turn information-gain rewards to stabilize training.
  • This method significantly improves performance and avoids optimization failures.
  • Granular, context-aware rewards are crucial for complex sequential AI tasks.

Original post by Hao Dou

"arXiv:2607.16244v1 Announce Type: new Abstract: Training multi-turn evidence-reading agents with outcome-only reinforcement learning is unstable because intermediate turns receive little direct credit. In HotpotQA experiments with Qwen2.5-3B-Instruct, GRPO initially improves (sta…"

View on X

Originally posted by Hao Dou on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses