Does Learning to Predict the World Help Agents Act? Auditing World-Model Post-Training
cs.CL, cs.AI
Submitted: 2026-09-27
Updated: 2026-09-27
Comments: 24 pages, 6 figures. Preprint
Code: https://github.com/KosmoCHE/WM-PostTraining-Audit
License: http://creativecommons.org/licenses/by/4.0/
The gist: Predicting how an environment will change before acting is a natural route to better decision making for agents.
Terminology
Abstract
Predicting how an environment will change before acting is a natural route to better decision making for agents. Recent post-training methods therefore require agents to predict the next observation and turn that prediction into a reward or a direct supervision signal, which is called world model. Existing next-observation training methods help the agent to learn the environmental content. However, they additionally involve an optimization process, which may introduce several effects other than learning to predict the world. Consequently, where the performance gain comes from during the training process remains an open question. We answer this research question through replacing true next-observation targets with in-distribution mismatched observations during the training process. Across two interactive text environments, mismatched targets lower prediction accuracy by 15.3-61.6% relative to ground-truth targets, yet retain substantial task gains over the base model. Compared with the base model, trained models consider more candidate actions and exhibit less looping. We also introduce a setting that replaces prediction-based rewards with independent random signals. This training expands task coverage (pass@64) even when the reward carries no environment information. We also generalize this finding to VisualWebArena, where random-reward training raises pass@64 by 14.3% relative to the base model, without observation-matching rewards or an external multimodal teacher for reward construction.
Sources
- PatchWorld: Gradient-Free Optimization of Executable World Models for Agent Environments
- Qwen2.5-VL Technical Report
- Beyond Next-Observation Prediction: Agent-Authored World Modeling for Sequential Decision Making
- Evaluating Large Language Models Trained on Code
- The Llama 3 Herd of Models
- Better World Models Can Lead to Better Post-Training Performance
- Dream to Control: Learning Behaviors by Latent Imagination
- Privileged, but Biased: How PI-Conditioned Teachers Break Self-Distillation
- Privileged Solutions or Context-Induced Teacher Behavior? Dissecting On-Policy Self-Distillation
- Form, Not Content? A Preregistered, Placebo-Controlled Evaluation of Learned Error-Conditioned Self-Repair Through Prompts and Weights in Frozen Small Code Models
- Why Does Self-Distillation (Sometimes) Degrade the Reasoning Capability of LLMs?
- TAPO: Transition-Aware Policy Optimization for LLM Agents
- CoMAP: Co-Evolving World Models and Agent Policies for LLM Agents
- Policy and World Modeling Co-Training for Language Agents
- Describe-Then-Act: Proactive Agent Steering via Distilled Language-Action World Models
- Qwen2.5 Technical Report
- DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Models
- ECHO: Terminal Agents Learn World Models for Free
- Thinking by Doing: Building Efficient World Model Reasoning in LLMs via Multi-turn Interaction
- SWE-World: Building Software Engineering Agents in Docker-Free Environments
Related papers
- Exploring Solution Divergence and Its Effect on Large Language Model Problem Solving
- Ishigaki-IDS-Bench: A Benchmark for Generating Information Delivery Specification from BIM Information Requirements
- Subliminal Steering: Stronger Encoding of Hidden Signals
- MedStruct-S: A Benchmark for Key Discovery, Key-Conditioned QA and Semi-Structured Extraction from OCR Clinical Reports
- The End of Transformers? On Challenging Attention and the Rise of Sub-Quadratic Architectures
- Untangling the Mechanisms of Misleading Context in Medical Question Answering