MLE-World: a learned world model predicts ML experiment outcomes, halving compute while raising scores

sherryyangML · x · 2026-10-23

Researchers from UW, NYU and others introduce MLE-World, a learned world model (fine-tuned Qwen3.5-4B) that predicts execution feedback — terminal output, exception class, validation metric — before ML research agents actually run experiments, with a confidence gate deciding when real execution is needed. On 17 MLE-bench Lite tasks it lifts mean leaderboard percentile from 45.4 to 54.1 (+19%) over baseline while cutting real execution time from 9.94 to 4.77 hours per trial. Paper, model, data and code are open.

Related event: MLE-World model predicts ML experiment outcomes, halving compute and lifting rankings(4 posts)→

Original post →

More from coding & agent

coding & agent channel →