Amazon paper uses a language model to predict AI training experiment resultsMachine translation

New Amazon paper shows that an LLM can act as a research world model, predicting whether a training change will help before anyone spends GPU time on it. AI research agents can propose experiments far faster than teams can afford to run them. Choosing what gets GPU time means guessing outcomes in advance. They used an LLM as a research world model that predicts an experiment's gain before it runs. They tested it on 2,653 real experiment records from 9 setups, from pretraining to inference. Past records from the same setup raised average ranking correlation with actual results from 0.506 to 0.774 across 5 setups. On the OLMo3-100M setup, adding records at low reasoning effort scored 0.892, while max effort without records reached 0.648. If you run research agents, log every experiment, including failures, and feed those records to whatever model picks the next run.