MLExpert: Towards a Generalist Agentic Model for Autonomous Machine Learning Engineering
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
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.
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
Spaces citing this paper 1
Collections including this paper 0
No Collection including this paper