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.
This is an automated message from the Librarian Bot. I found the following papers similar to this paper.
The following papers were recommended by the Semantic Scholar API
- MLToolBench: Learning Tool-Augmented Agents for Machine Learning Development (2026)
- FrogNano: Training a 4B Coding Agent via Online Task Synthesis (2026)
- ReTerm: Recovery Training for Self-Improving Terminal Agents (2026)
- Harness Learning Enables Generalizable Test-Time Adaptation (2026)
- Assembling Insights for Agentic Machine Learning Engineering Systems (2026)
- WEFT: Scaling Tool-Use Post-Training for General-Purpose Agents (2026)
- AutoCompact: Learning When to Compact Context in Long-Horizon Coding Agents (2026)
Please give a thumbs up to this comment if you found it helpful!
If you want recommendations for any Paper on Hugging Face checkout this Space
You can directly ask Librarian Bot for paper recommendations by tagging it in a comment: @librarian-bot recommend
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