AIBuildAI：用于自动构建AI模型的AI代理
人工智能
2026-04-17 v1
摘要
AI模型支撑着现代智能系统的发展，推动着医学、金融、科技等领域的进步。然而，开发高性能AI模型仍是一项耗时的工作，需要专家实践者迭代设计网络结构、工程表示、实现训练管道并通过经验评估不断完善。现有的AutoML方法仅在狭窄方面（如超参数优化和预定义搜索空间内的模型选择）缓解了这一负担，使完整开发生命周期仍依赖人类专长。为填补这一空白，我们引入AIBuildAI，一个从任务描述和训练数据中自动构建AI模型的AI代理。AIBuildAI采用分层代理架构，其中经理人协调三个专门的子代理：用于建模策略的设计师、用于实现和调试的编码员、用于训练和性能优化的调谐师。每个子代理本身都是基于大型语言模型（LLM）的代理，具备多步推理和工具使用能力，实现了从任务规格到可部署模型的端到端AI模型开发自动化，超越了现有AutoML方法的范围。我们在MLE-Bench上评估了AIBuildAI，这是一个包含真实Kaggle式AI开发任务的基准，涵盖视觉、文本、时间序列和表格等模态。AIBuildAI在MLE-Bench中名列前茅，获奖率达63.1%，超越所有现有基线方法，与高度经验丰富的AI工程师能力相当。这些结果表明，分层代理系统可以自动化完成从任务规格到可部署模型的完整AI模型开发过程，为实现广泛可及的AI开发指明了道路，最低人工干预即可实现。
引用
@article{arxiv.2604.14455,
title = {AIBuildAI: An AI Agent for Automatically Building AI Models},
author = {Ruiyi Zhang and Peijia Qin and Qi Cao and Li Zhang and Pengtao Xie},
journal= {arXiv preprint arXiv:2604.14455},
year = {2026}
}