Language Models as AI Research World Models
Not on the map. Why?
Abstract (arXiv)
AI research agents automate the cycle of proposing, implementing, and evaluating experiments, opening a path toward recursive self-improvement. Yet their ability to propose experiments outpaces their capacity to execute them in real environments, making outcome prediction a key capability for sustained self-improvement under limited experimental budgets. We investigate language models as Research World Models (RWMs), which predict the outcomes of candidate interventions across research environments. Our evaluation draws on over 2,600 experimental records from nine research environments spanning pretraining, post-training, and inference, representing more than 171,000 H100 GPU-hours of experimentation. Research knowledge acquired from real experimental experience improves RWM predictions of unseen interventions within the same environment (Spearman +0.27), and can be reused across environments. For example, using only pretraining experience from OLMo3, Marin, and Nanochat, an RWM reduces selection regret in the Qwen3 environment by 78% compared with zero-experience setting. These benefits extend to multi-round Autoresearch under a fixed selection budget: RWMs with in-env and cross-env research knowledge increase the best gain achieved by 15.8% and 11.6%, respectively. Ablations across 13 language models used as RWMs show that adding research knowledge can improve intervention ranking more than changing models or increasing reasoning effort alone. These findings support language models as RWMs and motivate accumulating experimental data for future RWM training.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 33 of its 48 references that are in the map. First posted on arXiv 8 Oct 2026.
Builds on
33 of 48 references.
References · 33
- 2022ReAct: Synergizing Reasoning and Acting in Language Modelscited by 8,355
- 2021Evaluating Large Language Models Trained on Codecited by 8,053
- 2020Scaling Laws for Neural Language Modelscited by 6,411
- 2023Reflexion: Language Agents with Verbal Reinforcement Learningcited by 4,064
- 2022Training Compute-Optimal Large Language Modelscited by 2,927
- 2023Voyager: An Open-Ended Embodied Agent with Large Language Modelscited by 2,081
- 2019Dream to Control: Learning Behaviors by Latent Imaginationcited by 1,598
- 2018World Modelscited by 1,444
- 2018Learning Latent Dynamics for Planning from Pixelscited by 1,221
- 2023Mastering Diverse Domains through World Modelscited by 1,193
- 2024The AI Scientist: Towards Fully Automated Open-Ended Scientific Discoverycited by 1,002
- 2021An Explanation of In-context Learning as Implicit Bayesian Inferencecited by 915
21 more
- 2023ExpeL: LLM Agents Are Experiential Learnerscited by 814
- 2019When to Trust Your Model: Model-Based Policy Optimizationcited by 753
- 2018Deep Reinforcement Learning in a Handful of Trials using Probabilistic Dynamics Modelscited by 748
- 2022What Can Transformers Learn In-Context? A Case Study of Simple Function Classescited by 700
- 2024MLE-bench: Evaluating Machine Learning Agents on Machine Learning Engineeringcited by 429
- 2025Agent Laboratory: Using LLM Agents as Research Assistantscited by 405
- 2023DoReMi: Optimizing Data Mixtures Speeds Up Language Model Pretrainingcited by 385
- 2025The AI Scientist-v2: Workshop-Level Automated Scientific Discovery via Agentic Tree Searchcited by 375
- 2023MLAgentBench: Evaluating Language Agents on Machine Learning Experimentationcited by 339
- 2024ResearchAgent: Iterative Research Idea Generation over Scientific Literature with Large Language Modelscited by 200
- 2022In-context Reinforcement Learning with Algorithm Distillationcited by 186
- 2025Encoder-Decoder Diffusion Language Models for Efficient Training and Inferencecited by 21
- 2025Predicting Empirical AI Research Outcomes with Language Modelscited by 14
- 2026AIRA_2: Overcoming Bottlenecks in AI Research Agentscited by 10
- 2017Accelerating Neural Architecture Search using Performance Predictioncited by 9
- 2026DUEL: Exact Likelihood for Masked Diffusion via Deterministic Unmaskingcited by 8
- 2026AI Research Preference Modelscited by 7
- 2026AutoLLMResearch: Training Research Agents for Automating LLM Experiment Configuration - Learning from Cheap, Optimizing Expensivecited by 4
- 2026SciPredict: Can LLMs Predict the Outcomes of Scientific Experiments in Natural Sciences?cited by 3
- 2026AI-Supervisor: Autonomous AI Research Supervision via a Persistent Research World Modelcited by 2
- 2026Scaling Automatic Research Agents via World Modelscited by 1
Built on by
Not cited yet.