arXiv:2604.14455cs.AI2026-04被引 9

AI Agent 自动构建模型,从任务描述到可部署模型全流程无需人工干预。

AIBuildAI: An AI Agent for Automatically Building AI Models

  • 分层代理架构:管理器协调设计、编码、调优三类专用AI代理
  • 在MLE-Bench上达成63.1%奖牌率,媲美资深工程师
  • 首次实现从任务描述到可部署模型的全链路自动化

AI模型支撑现代智能系统,在科学、医疗、金融与技术领域推动进展。然而,构建高性能模型仍依赖专家反复设计架构、特征工程、实现训练流程并经实验优化,耗时费力。现有AutoML方法仅覆盖超参数优化与模型选择等局部环节,难以覆盖完整开发周期。为此,我们提出AIBuildAI,一个可自动从任务描述和训练数据构建AI模型的AI代理。其采用分层代理架构,由管理器协调设计师(建模策略)、编码员(实现与调试)、调优师(训练与性能优化)三个专业子代理。每个子代理均为基于大语言模型的智能体,具备多步推理与工具使用能力,实现端到端自动化,超越现有AutoML能力边界。我们在涵盖视觉、文本、时间序列与表格数据的现实世界Kaggle风格任务基准MLE-Bench上评估,AIBuildAI以63.1%奖牌率排名第一,优于所有基线方法,达到资深AI工程师水平。结果表明,分层代理系统可自动化完成从任务定义到可部署模型的全过程,为低介入式广泛可用的AI开发提供新路径。

原文摘要 · Abstract (English)

AI models underpin modern intelligent systems, driving advances across science, medicine, finance, and technology. Yet developing high-performing AI models remains a labor-intensive process that requires expert practitioners to iteratively design architectures, engineer representations, implement training pipelines and refine approaches through empirical evaluation. Existing AutoML methods partially alleviate this burden but remain limited to narrow aspects such as hyperparameter optimization and model selection within predefined search spaces, leaving the full development lifecycle largely dependent on human expertise. To address this gap, we introduce AIBuildAI, an AI agent that automatically builds AI models from a task description and training data. AIBuildAI adopts a hierarchical agent architecture in which a manager agent coordinates three specialized sub-agents: a designer for modeling strategy, a coder for implementation and debugging, and a tuner for training and performance optimization. Each sub-agent is itself a large language model (LLM) based agent capable of multi-step reasoning and tool use, enabling end-to-end automation of the AI model development process that goes beyond the scope of existing AutoML approaches. We evaluate AIBuildAI on MLE-Bench, a benchmark of realistic Kaggle-style AI development tasks spanning visual, textual, time-series and tabular modalities. AIBuildAI ranks first on MLE-Bench with a medal rate of 63.1%, outperforming all existing baseline methods and matching the capability of highly experienced AI engineers. These results demonstrate that hierarchical agent systems can automate the full AI model development process from task specification to deployable model, suggesting a pathway toward broadly accessible AI development with minimal human intervention.

AI代理自动建模AutoML大模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。