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

Internalizing the Future: A Unified Agentic Training Paradigm for World Model Planning

About

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.

Xuan Zhang, Zhijian Zhou, Lingfeng Qiao, Yulei Qin, Ke Li, Xing Sun, Xiaoyu Tan, Chao Qu, Yuan Qi• 2026

Related benchmarks

TaskDatasetResultRank
Question AnsweringBamboogle--
227
Question AnsweringPopQA
Accuracy42.8
158
Question AnsweringHotpotQA
Accuracy59
50
Question AnsweringMuSiQue
Accuracy (MuSiQue QA)34.2
33
Question Answering2WikiMultihopQA
Accuracy46.2
13
Mathematical ReasoningAIME 2025
Mean Score @3026.8
8
Mathematical ReasoningAIME Average
Mean@3029.5
8
Mathematical ReasoningAIME 2024
Mean @3034.6
8
Mathematical ReasoningAIME 2026
Mean Score @3027.1
8
Showing 9 of 9 rows

Other info

Follow for update