Can LLMs predict the outcomes of machine learning experiments?
Organizations: Autolab AI · Institute of Foundation Models · MIT
Abstract
Recent advances in AI make Recursive Self-Improvement (RSI) increasingly plausible. Large Language Models (LLMs) can already propose, select, and run Machine Learning (ML) experiments. However, the space of possible design choices in ML is vast. An LLM that proposes an experiment needs a reliable intuition of whether it is worth running given the timing constraints of long training runs. In this work, we measure such intuition in current frontier LLMs by evaluating how accurately they predict the impact of self-generated ML experiment ideas before these are put into training runs. To this end, we construct a dataset of 288 LLM-proposed ideas on eight MLE-bench tasks. We then prompt five frontier LLMs (GPT-6 Astra, Claude Opus 5.5, Claude Fable 5.1, Kimi K3, and GLM-5.3) to predict the impact of each idea and compare their predictions with the actual outcomes after training. We report four main findings. First, LLM predictions are substantially better than random chance but far from accurate: even the best model, Claude Opus 5.5, explains only half of the variance in outcomes (R^2 = 0.49). Second, we notice that all tested models systematically underestimate the magnitude of the changes. Third, prediction quality varies widely across tasks. Finally, the choice of the coding harness that actually implements the ideas has little effect on outcomes, and we find no evidence of self-preference. Higher reasoning settings and additional context lead to modest gains. Overall, our results show that current LLMs have useful but limited intuition about which ML ideas will work. We release our dataset to support further study.
Figures & tables
Appendix figures & tables2 assets
Supplementary material from the paper’s appendix.
Appendix
| Change in score ( ) | ||||
|---|---|---|---|---|
| Task | Idea | Real | Opus 5.5 | Kimi K3 |
| Spooky author identification Fable 5.1, asked to worsen | Remove English stop words from the word-level features (word TF-IDF block and the naive Bayes count features) | |||
| Toxic comment classification Kimi K3, asked to improve | Tune per-label A/B rank-blend weights on a 10% holdout instead of fixed 0.5/0.5 | |||
| Random acts of pizza Fable 5.1, asked to improve | Add a giver_username_if_known indicator column | |||
| Tabular playground Kimi K3, asked to improve | Refit LightGBM on 100% of the training rows at the early-stopped iteration before predicting the test set | |||
| Leaf classification Fable 5.1, asked to worsen | Label smoothing (eps=0.1) in the two-branch NN cross-entropy loss | |||
LLM Evaluation
Large language model evaluation measures what models can do, how reliably they do it, and whether benchmark scores predict real-world performance.
See all 2,408 papers