论文
AIBuildAI:一个自动构建AI模型的AI智能体
AIBuildAI: An AI Agent for Automatically Building AI Models
摘要
AI模型是现代智能系统的基石,推动着科学、医学、金融和技术领域的进步。然而,开发高性能AI模型仍是一个劳动密集的过程,需要专家从业者迭代设计架构、工程化表征、实现训练流水线,并通过实证评估不断改进方法。现有AutoML方法部分缓解了这一负担,但仍局限于预定义搜索空间内的超参数优化和模型选择等狭窄环节,整个开发生命周期在很大程度上仍依赖人类专长。为弥合这一差距,我们提出AIBuildAI,一个能从任务描述和训练数据出发自动构建AI模型的AI智能体。AIBuildAI采用分层智能体架构:一个管理者智能体协调三个专职子智能体——负责建模策略的设计者(designer)、负责实现与调试的编码者(coder)、负责训练与性能优化的调优者(tuner)。每个子智能体本身都是基于大语言模型(LLM)的智能体,具备多步推理和工具使用能力,实现了超出现有AutoML方法范围的AI模型开发全流程端到端自动化。我们在MLE-Bench上评估AIBuildAI,该基准涵盖视觉、文本、时间序列和表格模态的真实Kaggle式AI开发任务。AIBuildAI以63.1%的奖牌率在MLE-Bench上排名第一,超越所有现有基线方法,并达到资深AI工程师的能力水平。这些结果表明,分层智能体系统可以自动化从任务说明到可部署模型的完整AI模型开发过程,为以最少人工干预实现广泛可及的AI开发提示了一条路径。