Papers
arxiv:2610.22344

MLExpert: Towards a Generalist Agentic Model for Autonomous Machine Learning Engineering

Published on Oct 15
· Submitted by
Xiangchao Yan
on Oct 16
Authors:
,
,
,
,
,
,
,
,

Abstract

A post-training framework using hierarchical supervised fine-tuning and windowed interaction reinforcement learning trains a 35B mixture-of-experts agent to autonomously optimize long-horizon machine learning engineering workflows.

Large language models (LLMs) are expanding from assisting scientific research and engineering to autonomously developing and improving AI systems, advancing AI for AI (AI4AI) as a pathway toward recursive self-improvement (RSI). Workflow-driven machine learning engineering (MLE) systems provide a practical path toward this pursuit through complex orchestration that manages long-horizon experimentation, while recent MLE systems adopt model-directed interaction, allowing models to autonomously organize experiments and adapt optimization strategies based on execution feedback. However, feedback from each experiment reflects several preceding decisions, and a long MLE run contains many such costly experiments. Training must therefore select useful supervision from long interaction trajectories and learn from intermediate outcomes. We propose a post-training method for autonomous long-horizon MLE optimization and apply it to train MLExpert, a 35B Mixture-of-Experts agentic model. We design the Kaggle-MLE Factory to construct executable tasks and an interaction harness for model-directed experimentation. Hierarchical agentic SFT transfers general tool-use and coding capabilities to MLE through outcome-guided supervision. Windowed Interaction RL (WIRL) treats the multi-turn decisions between successive experiment results as an execution window and assigns it a shared outcome-based reward. Asynchronous collection and policy correction support policy updates before the trajectory ends. On MLE-Bench Lite, MLExpert improves the average medal rate of the base model from 37.9% to 72.7%, achieving performance comparable to larger frontier models such as Kimi-K3 and GLM-5.3. Further experiments show that MLExpert improves performance on NatureBench Lite, agent scaffold optimization, and algorithm optimization, suggesting that the learned optimization capability generalizes to other tasks.

Community

Paper submitter

MLExpert is a 35B Mixture-of-Experts agentic model trained to autonomously plan, execute, and refine machine-learning experiments through sustained tool interaction. Rather than relying on hand-crafted workflows, it learns to guide optimization directly from execution feedback. Its training combines the Kaggle-MLE Factory for executable task construction, hierarchical agentic supervised fine-tuning, and Windowed Interaction Reinforcement Learning (WIRL), which turns intermediate experiment outcomes into learning signals without waiting for an entire run to finish. On MLE-Bench Lite, MLExpert raises the base model’s average medal rate from 37.9% to 72.7%, reaching performance comparable to larger frontier models. Further gains on scientific tasks, agent scaffold optimization, and algorithm optimization suggest that its learned optimization capabilities extend beyond machine-learning engineering.

Sign up or log in to comment

Get this paper in your agent:

hf papers read 2610.22344
Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash

Models citing this paper 1

Datasets citing this paper 0

No dataset linking this paper

Cite arxiv.org/abs/2610.22344 in a dataset README.md to link it from this page.

Spaces citing this paper 1

Collections including this paper 0

No Collection including this paper

Add this paper to a collection to link it from this page.