Skip to content

论文arXiv 2610.2234413 天前

MLExpert:迈向用于自主机器学习工程的通用智能体模型 ​

原文链接:MLExpert: Towards a Generalist Agentic Model for Autonomous Machine Learning Engineering(arXiv 2610.22344 · 发布日期:2026-10-15)

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

针对长程自主机器学习工程优化,提出后训练方法训练MLExpert,并在MLE-Bench Lite等任务上验证。

核心要点 ​

  • 长程MLE运行包含许多高成本实验,训练需从长交互轨迹中选择有效监督并学习中间结果。
  • 方法构建Kaggle-MLE Factory,并结合分层agentic SFT与Windowed Interaction RL进行训练。
  • 在MLE-Bench Lite上,MLExpert将基础模型平均奖牌率从37.9%提升至72.7%。
  • 进一步实验显示,其优化能力可迁移到NatureBench Lite、agent scaffold优化和算法优化。

验证范围与局限:验证范围限于摘要所述MLE-Bench Lite、NatureBench Lite、agent scaffold优化和算法优化实验。

英文摘要 (Abstract) ​

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.

作者:Shangheng Du, Jinxin Shi, Shuaiyu Zhang, Yiming Wu, Junpeng Zhao, Shiyang Feng, Xin Li, Liang He, Xiangchao Yan, Bo Zhang

来源

本页事实对应的公开页面,以原文为准。

  1. arXiv 2610.22344:MLExpert: Towards a Generalist Agentic Model for Autonomous Machine Learning Engineering访问 2026-10-21