Language Models as AI Research World Models
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 and can be reused across environments. Across five autoregressive pretraining environments, adding source-environment records without target-environment records reduces macro-average best-of-3 regret (Regret@3) by 52.2% when selecting three of sixteen candidates. These benefits extend to multi-round Autoresearch under fixed selection budgets. 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.
Then back it, or bet against it.
Related papers
Open the market on this paper to see 7 more related papers.