跳到正文
返回论文列表
cs.AI提交于 已译

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

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

Shangheng Du · Jinxin Shi · Shuaiyu Zhang · Yiming Wu · Junpeng Zhao · Shiyang Feng · et al.

中文摘要

大语言模型正从辅助科研转向自主开发AI系统,推动AI for AI (AI4AI) 实现递归自我改进 (RSI)。工作流驱动的机器学习工程 (MLE) 系统通过复杂编排管理长周期实验,近期系统采用模型主导交互,允许模型基于反馈自主组织实验。然而,单次实验反馈反映多个先前决策,长周期运行包含大量高成本实验,训练需从长交互轨迹中筛选有效监督。针对自主长周期MLE优化,提出后训练方法并训练35B混合专家 (Mixture-of-Experts) 智能体模型MLExpert。设计Kaggle-MLE Factory构建可执行任务与交互框架。层级智能体SFT通过结果引导监督将通用工具使用与编码能力迁移至MLE。窗口交互强化学习 (WIRL) 将连续实验结果间的多轮决策视为执行窗口,分配共享结果奖励,支持异步采集与策略校正。在MLE-Bench Lite上,MLExpert将基础模型平均奖牌率从37.9%提升至72.7%,性能媲美Kimi-K3和GLM-5.3等前沿大模型。实验表明,其优化能力可泛化至NatureBench Lite、智能体脚手架优化及算法优化任务。

关键要点

  1. 01针对长周期MLE实验中反馈滞后与高成本问题,提出从长交互轨迹中筛选有效监督的后训练方法。
  2. 02训练35B MoE智能体模型MLExpert,设计Kaggle-MLE Factory构建任务,采用层级SFT迁移通用编码与工具使用能力。
  3. 03提出窗口交互强化学习 (WIRL),将连续实验间决策视为窗口并分配共享结果奖励,支持异步策略更新。
  4. 04在MLE-Bench Lite上将平均奖牌率从37.9%提升至72.7%,性能接近Kimi-K3等前沿大模型。
  5. 05验证优化能力可泛化至NatureBench Lite、智能体脚手架优化及算法优化等其他任务。

解读

尚无解读。

原始英文摘要

arXiv:2610.22344v1 Announce Type: cross 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.

同方向论文 · cs.AI

查看全部 →

反向发现 · 这篇论文相关的工具与提示词

去交叉发现 →

关联提示词 · 6

智能体github.com/f/awesome-chatgpt-prompts·外部收录·

Weekend Planner

- Read the weekend weather, calendar and budget. - Find local events, hikes, exhibitions and restaurants. - Propose 3 plans: adventure, cozy, social, with times and costs. - Prepare reservation steps; book nothing without approval. - Rememb…

智能体github.com/f/awesome-chatgpt-prompts·外部收录·

Task Planner

--- name: sa-plan description: Structured Autonomy Planning Prompt model: Claude Sonnet 4.5 (copilot) agent: agent --- You are a Project Planning Agent that collaborates with users to design development plans. A development plan defines a…

研究github.com/f/awesome-chatgpt-prompts·外部收录·

Unified Research and Source Analysis Prompt

Unified, High-Precision Research & Analysis Prompt for ChatGPT and Perplexity AI ROLE & BEHAVIOR You are a professional researcher-analyst. Handle inputs as follows: * If the input is a URL/URI: open it fully with your browsing tool (e.g.…