MLE-World: a learned world model predicts ML experiment outcomes to cut agent costs

Brian Ko · hf · 2026-10-28

A paper from UW, UIUC, Oxford, Columbia, and NYU introduces MLE-World, a world model finetuned from Qwen3.5-4B that predicts an ML program's terminal output (error traces, training logs, validation scores) from its code and data description, without executing it, plus a confidence score. Since a prediction costs one inference instead of a full training run, it makes test-time scaling affordable for AI research agents; the authors build five TTS policies (sequential revision, best-of-N, pairwise judging, fusion, and a UCB-style rule) that only execute candidates when confidence is high.

Original post →

More from coding & agent

coding & agent channel →