Toward Autonomous Long-Horizon Engineering for ML Research
Abstract
Agentic systems increasingly automate pieces of AI research. Yet turning underspecified research objectives into runnable, experimentally validated ML systems remains a central bottleneck. We study this operational setting as long-horizon ML research engineering: converting a research specification into a runnable ML system through repeated implementation, experimentation, and refinement. The central challenge is to sustain cumulative project progress across heterogeneous stages under delayed, confounded feedback. We introduce AiScientist, a multi-agent system built around thin control over thick state: a lightweight hierarchical research team coordinates through a File-as-Bus workspace that preserves decision-relevant artifacts across roles and invocations. On PaperBench, AiScientist improves over the strongest matched baselines by 9.92 and 11.15 points with Gemini-3-Flash and GLM-5, respectively. On MLE-Bench Lite, it reaches 81.82 Any Medal% under both backbones, improving over the strongest matched baselines by 4.55 and 16.67 points, and exceeding a Codex/GPT-5.5 xhigh frontier harness reference by 13.64 Any Medal points. Ablations and process analyses show that durable project state is central to later-round refinement: removing File-as-Bus lowers PaperBench score by 6.41 points and MLE-Bench Lite Any Medal% by 31.82 points. These results suggest that long-horizon AI research is not only a problem of stronger local reasoning, but a systems problem of maintaining cumulative, inspectable project progress.
Figures & tables
| Task Name | GPT-5.5 | Gemini-3-Flash | GLM-5 | ||||||
| Codex | BasicAgent | IterAgent | AiScientist | BasicAgent | IterAgent | AiScientist | |||
| adaptive-pruning | 33.42 | 24.53 | 3.05 | 27.25 | +2.72 | 30.82 | 11.93 | 33.26 | +2.44 |
| all-in-one | 52.93 | 20.86 | 45.13 | 46.29 | +1.16 | 33.78 | 44.43 | 49.47 | +5.04 |
| bam | 56.65 | 48.46 | 45.04 | 56.59 | +8.13 | 51.45 | 47.91 | 61.11 | +9.66 |
| bbox | 17.30 | 15.43 | 8.30 | 33.79 | +18.36 | 23.55 | 19.28 | 30.02 | +6.47 |
| bridging-data-gaps | 14.16 | 12.59 | 12.44 | 23.09 | +10.50 | 9.80 | 12.50 | 26.46 | +13.96 |
| Agent | Model | Valid Submission | Above Median | Bronze | Silver | Gold | Any Medal |
| Official MLE-Bench Leaderboard Results | |||||||
| InternAgent | Deepseek-R1 | ||||||
| ML-Master | Deepseek-R1 | ||||||
| AIRA-dojo | o3 | ||||||
| ML-Master 2.0 | Deepseek-V3.2-Spe | ||||||
| R&D-Agent | GPT-5 | ||||||
Appendix figures & tables3 assets
Supplementary material from the paper’s appendix.
Appendix
| Artifact | Purpose | Update Mode | Writer | Readers |
| paper_analysis/summary.md | Concise implementation-oriented summary of the paper, including task goal, target result, and key assumptions. | Versioned artifact | Paper Comprehension | Orchestrator, Prioritization, Implementation, Experimentation |
| paper_analysis/structured.md | Structured extraction of datasets, metrics, methods, baselines, training details, and resource requirements. | Versioned artifact | Paper Comprehension | Orchestrator, Prioritization, Implementation, Experimentation |
| paper_analysis/algorithm.md | Algorithm-level reconstruction of the proposed method and implementation-relevant details. | Versioned artifact | Paper Comprehension | Implementation, Experimentation, Orchestrator |
| paper_analysis/baseline.md | Baseline methods, expected comparisons, and result targets that should be reproduced or approximated. | Versioned artifact | Paper Comprehension | Prioritization, Implementation, Experimentation, Orchestrator |
| agent/prioritized_task.md | Current ordered task plan, including dependencies, expected impact, feasibility, and next actions. | Versioned artifact | Prioritization | Orchestrator, Implementation, Experimentation |
| submission/ | Runnable repository under construction, including source code, configuration, scripts, and generated project files. | Mutable runnable artifact | Implementation | Implementation, Experimentation, Orchestrator |
Long-Horizon LLM Agents
Long-horizon large language model agents plan, use tools, and preserve task state across extended, multi-step interactions.
See all 446 papers