Internalizing the Future: A Unified Agentic Training Paradigm for World Model Planning
On the map: World models → World models for language agents
also in 1 more
Abstract (arXiv)
Large language model (LLM) agents have demonstrated strong capability in sequential decision-making, yet they remains fundamentally reactive in long-horizon tasks. Unlike humans who employ "what-if" reasoning to evaluate potential plans before commitment, standard agents lack an internal world model to simulate future outcomes. Therefore, we propose to internalize future-aware planning by training a single autoregressive model to verbalize both a prospective state rollout and a plan-conditioned success estimate-a textual analogue of the Q-value. Crucially, we identify a format-capability gap: simply fine-tuning agents on look-ahead traces during post-training leads to superficial mimicry of foresight without genuine predictive grounding. To bridge this gap, we introduce a three-stage training paradigm: (i) World Model Agentic Mid-Training (WM-AMT) to inject latent predictive capabilities into the policy; (ii) Format-Eliciting SFT (FE-SFT) to structure this injected capability; and (iii) Foresight-Conditioned Reinforcement Learning (FC-RL) to refine the calibration and utility of the generated simulations. Evaluated on search and mathematical reasoning tasks, our approach consistently outperforms other training baselines. Our results demonstrate that effective internal world modeling in LLM agents requires a capability-first training pipeline to achieve grounded and calibrated foresight.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Placed in World models for language agents, one of 11 directions in World models. Builds on 39 of its 48 references that are in the map. First posted on arXiv 25 Jun 2026.
Builds on
39 of 48 references.
Closest ancestors · 6
Same direction, newest first
- 2025From Word to World: Can Large Language Models be Implicit Text-based World Models?21 Dec
- 2025Why Do LLM Agents Fail in Exploring New Environments? A World-Modeling Perspective16 Oct
- 2025WebEvolver: Enhancing Web Agent Self-Improvement with Coevolving World Model23 Apr
- 2024Is Your LLM Secretly a World Model of the Internet? Model-Based Planning for Web Agents10 Nov
- 2024Web Agents with World Models: Learning and Leveraging Environment Dynamics in Web Navigation17 Oct
- 2023Reasoning with Language Model is Planning with World Model24 May
Other references · 33
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,302
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2020Dense Passage Retrieval for Open-Domain Question Answeringcited by 3,779
- 2018HotpotQA: A Dataset for Diverse, Explainable Multi-hop Question Answeringcited by 3,686
- 2017TriviaQA: A Large Scale Distantly Supervised Challenge Dataset for Reading Comprehensioncited by 2,895
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,678
- 2023A Survey on Large Language Model based Autonomous Agentscited by 2,203
- 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 1,980
- 2019Dream to Control: Learning Behaviors by Latent Imaginationcited by 1,598
- 2025Search-R1: Training LLMs to Reason and Leverage Search Engines with Reinforcement Learningcited by 1,371
- 2020Constructing A Multi-hop QA Dataset for Comprehensive Evaluation of Reasoning Stepscited by 1,259
21 more
- 2022Text Embeddings by Weakly-Supervised Contrastive Pre-trainingcited by 1,216
- 2023Mind2Web: Towards a Generalist Agent for the Webcited by 1,210
- 2023Mastering Diverse Domains through World Modelscited by 1,193
- 2022When Not to Trust Language Models: Investigating Effectiveness of Parametric and Non-Parametric Memoriescited by 1,103
- 2022Measuring and Narrowing the Compositionality Gap in Language Modelscited by 1,054
- 2020Mastering Atari with Discrete World Modelscited by 990
- 2021MuSiQue: Multihop Questions via Single-hop Question Compositioncited by 893
- 2018Recurrent World Models Facilitate Policy Evolutioncited by 761
- 2025Group-in-Group Policy Optimization for LLM Agent Trainingcited by 442
- 2023Cognitive Architectures for Language Agentscited by 422
- 2025ReTool: Reinforcement Learning for Strategic Tool Use in LLMscited by 377
- 2023Android in the Wild: A Large-Scale Dataset for Android Device Controlcited by 314
- 2024DigiRL: Training In-The-Wild Device-Control Agents with Autonomous Reinforcement Learningcited by 210
- 2024OpenCoder: The Open Cookbook for Top-Tier Code Large Language Modelscited by 127
- 2025Explorer: Scaling Exploration-driven Web Trajectory Synthesis for Multimodal Web Agentscited by 68
- 2025Self-Challenging Language Model Agentscited by 56
- 2025AgentSynth: Scalable Task Generation for Generalist Computer-Use Agentscited by 41
- 2025Scaling Agents via Continual Pre-trainingcited by 39
- 2025Mid-Training of Large Language Models: A Surveycited by 18
- 2025LLMs as Scalable, General-Purpose Simulators For Evolving Digital Agent Trainingcited by 17
- 2025Learning to Reason as Action Abstractions with Scalable Mid-Training RLcited by 3
Built on by
Citing papers per year: 2026: 2.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2026EVOKE: Eliciting World Knowledge in Agents for Transferable Decision-Making
- 2026Current Agents Fail to Leverage World Model as Tool for Foresight
- 2026Self-Evolving World Models for LLM Agent Planning
- 2026PriorZero: Bridging Language Priors and World Models for Decision Making
- 2025Beyond Pipelines: A Survey of the Paradigm Shift toward Model-Native Agentic AI
- 2026FutureWorld: A Live Reinforcement Learning Environment for Predictive Agents with Real-World Outcome Rewards
- 2026MemWM: Memory-Augmented Text-Based World Model
- 2026Beyond Next-Observation Prediction: Agent-Authored World Modeling for Sequential Decision Making