用低成本模型+多智能体协作,让自动机器学习更便宜高效
BudgetMLAgent: A Cost-Effective LLM Multi-Agent system for Automating Machine Learning Tasks
- 构建多智能体系统,融合专家协作与历史检索提升决策能力
- 成本降94.2%,成功率从22.72%提升至32.95%
- 适合预算有限但需自动化复杂机器学习任务的研究者
大型语言模型(LLMs)在代码生成方面表现优异,但在复杂机器学习(ML)任务中常表现不佳。现有基于单个LLM的系统依赖昂贵模型如GPT-4,而低成本模型(如Gemini-Pro、Mixtral、CodeLlama)在单智能体设置下性能远逊于GPT-4。为实现成本高效的LLM解决方案,我们提出一个基于多智能体的系统,结合专家组合、动态检索过往观察、LLM级联和调用专家请求机制。在MLAgentBench基准上对机器学习工程任务进行实证分析表明,该系统仅使用无成本的Gemini作为基础模型,搭配GPT-4级联及专家调用以处理规划任务,使每轮运行成本从GPT-4单智能体系统的0.931美元降至0.054美元(降幅94.2%),同时平均成功率从22.72%提升至32.95%。
原文摘要 · Abstract (English)
Large Language Models (LLMs) excel in diverse applications including generation of code snippets, but often struggle with generating code for complex Machine Learning (ML) tasks. Although existing LLM single-agent based systems give varying performance depending on the task complexity, they purely rely on larger and expensive models such as GPT-4. Our investigation reveals that no-cost and low-cost models such as Gemini-Pro, Mixtral and CodeLlama perform far worse than GPT-4 in a single-agent setting. With the motivation of developing a cost-efficient LLM based solution for solving ML tasks, we propose an LLM Multi-Agent based system which leverages combination of experts using profiling, efficient retrieval of past observations, LLM cascades, and ask-the-expert calls. Through empirical analysis on ML engineering tasks in the MLAgentBench benchmark, we demonstrate the effectiveness of our system, using no-cost models, namely Gemini as the base LLM, paired with GPT-4 in cascade and expert to serve occasional ask-the-expert calls for planning. With 94.2\% reduction in the cost (from \$0.931 per run cost averaged over all tasks for GPT-4 single agent system to \$0.054), our system is able to yield better average success rate of 32.95\% as compared to GPT-4 single-agent system yielding 22.72\% success rate averaged over all the tasks of MLAgentBench.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。