Language models predict experiment results to guide AI research choices
Language Models as AI Research World Models
Computation and Language
Summary
AI research involves running experiments, but doing lots of them can be very expensive and slow. The authors studied how language models can act like 'world models' that predict what will happen if a particular experiment is done. These predictions help choose better experiments without actually running all of them, saving time and effort. They showed that using data from previous experiments makes these predictions more accurate, both within and across different research areas.
What this means in practice
- •For ai development teams: Use prediction models to select effective AI training or tuning experiments and reduce costly trial runs.
- •For machine learning infrastructure teams: Improve resource allocation by forecasting experiment outcomes to prioritize GPU time on promising configurations.
Authors
Zijun Wang, Zewen Liu, Minhua Lin, Zhaotian Weng, Zhan Shi, Bing He, Yisi Sang, Dakuo Wang, Benoit Dumoulin, Wei Jin, Yuyin Zhou, Cihang Xie, Hanqing Lu
Abstract
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.