Co-Evolving Acting and Forecasting for Language Agents
cs.AI, cs.LG
Submitted: 2026-10-14
Updated: 2026-10-15
Code: https://github.com/ulab-uiuc/CoPE
License: http://creativecommons.org/licenses/by/4.0/
The gist: Training language agents is bottlenecked by the high cost of long-horizon rollouts, yet standard agentic reinforcement learning (RL) learns only from a terminal scalar reward, leaving the supervision
Abstract
Training language agents is bottlenecked by the high cost of long-horizon rollouts, yet standard agentic reinforcement learning (RL) learns only from a terminal scalar reward, leaving the supervision in intermediate steps largely unused. Prior work exploits this supervision through value targets that estimate per-step success probabilities or observation targets that fit environment response. Both are hard to construct well: (1) value targets come from the same single scalar signal for each step in the trajectory, and (2) observation targets, such as web pages and terminal logs, are long and noisy and reach only one step ahead. We propose, which adds a third kind of multi-step action targets. On successful on-policy trajectories, trains the policy backbone to generate the action sequences that follows each history. Each action is also supervised when forecast from the preceding histories, forming a temporal ensemble that the shared backbone absorbs during training, and this enables acting and forecasting to co-evolve. The policy still acts one step at a time at inference, with no overhead. Across five benchmarks (ALFWorld, ScienceWorld, WebShop, τ squared-bench, and AppWorld), (1) achieves stronger aggregate performance than GRPO and RL baselines with predictive auxiliary targets across simulated environments, while improving overall performance on both realistic applications; (2) delivers up to 1.9 times and 2.0 times higher sample efficiency than ECHO and GRPO, respectively, on ScienceWorld; and (3) achieves these gains with a single policy backbone, without additional models, separate pretraining, or inference overhead. The codebase is available at https://github.com/ulab-uiuc/CoPE.
Sources
- $\tau^2$-Bench: Evaluating Conversational Agents in a Dual-Control Environment
- Scaling Agent Learning via Experience Synthesis
- Process Reward Models for LLM Agents: Practical Framework and Directions
- Gemini 2.5: Pushing the Frontier with Advanced Reasoning, Multimodality, Long Context, and Next Generation Agentic Capabilities
- Mind2Web: Towards a Generalist Agent for the Web
- DynaWeb: Model-Based Reinforcement Learning of Web Agents
- Group-in-Group Policy Optimization for LLM Agent Training
- Hindsight Credit Assignment
- MetaGPT: Meta Programming for A Multi-Agent Collaborative Framework
- Sample-Efficient Online Learning in LM Agents via Hindsight Trajectory Rewriting
- InfoPO: Information-Driven Policy Optimization for User-Centric Agents
- Why Does Action Chunking Improve Behavioral Cloning Performance in Robotic Control?
- Truly Self-Improving Agents Require Intrinsic Metacognitive Learning
- Understanding R1-Zero-Like Training: A Critical Perspective
- The AI Scientist: Towards Fully Automated Open-Ended Scientific Discovery
- Policy and World Modeling Co-Training for Language Agents
- Agent Lightning: Train ANY AI Agents with Reinforcement Learning
- Qwen2.5 Technical Report
- Success Conditioning as Policy Improvement: The Optimization Problem Solved by Imitating Success
- Vibe Coding vs. Agentic Coding: Fundamentals and Practical Implications of Agentic AI
Related papers
- MAVEN-T: Reinforced Heterogeneous Distillation for Real-Time Multi-Agent Trajectory Prediction
- Model Discovery Agent: LLM-assisted Bayesian experiment design for data-efficient discovery of mechanistic world models
- Agents' Last Exam
- The Clinician's Veto: Navigating Trust, Liability, and Uncertainty in Autonomous AI Prescribing
- MindHelper: Closed-Loop Embodied Mental-State Reasoning for Precision Intervention
- Incumbent Advantage: Brand Bias and Cognitive Manipulation Dynamics in LLM Recommendation Systems