arXiv:2502.14815cs.AIcs.CL2025-02被引 33

用智能选型框架提升多模型协作系统性能,效果显著。

Optimizing Model Selection for Compound AI Systems

  • 基于模块表现估计,逐个优化各环节的模型选择
  • 实测在多智能体辩论等系统中提升5%-70%准确率
  • 适合需要高效组合多个大模型的任务场景

复合型AI系统通过整合多个大语言模型(LLM)调用,在诸多任务中表现出色。本文聚焦于核心问题:在系统中每个调用或模块应如何选择合适的LLM?我们发现模型选择对最终质量影响巨大,但搜索空间呈指数级增长。为此提出LLMSelector框架,利用两个关键经验观察:(i) 整体性能通常随单个模块表现单调上升,其他模块固定;(ii) 单个模块性能可由一个LLM准确估算。基于此,LLMSelector迭代选择模块并分配预测表现最优的模型,直至无进一步提升。该方法适用于任意模块数有限的复合系统,API调用次数与模块数线性相关,兼具实证与理论优势。在GPT-4o、Claude 3.5 Sonnet和Gemini 1.5等模型上,针对多智能体辩论和自修正等系统进行实验,结果显示相较统一使用单一模型,准确率提升5%-70%。

原文摘要 · Abstract (English)

Compound AI systems that combine multiple LLM calls, such as self-refine and multi-agent-debate, achieve strong performance on many AI tasks. We address a core question in optimizing compound systems: for each LLM call or module in the system, how should one decide which LLM to use? We show that these LLM choices have a large effect on quality, but the search space is exponential. We propose LLMSelector, an efficient framework for model selection in compound systems, which leverages two key empirical insights: (i) end-to-end performance is often monotonic in how well each module performs, with all other modules held fixed, and (ii) per-module performance can be estimated accurately by an LLM. Building upon these insights, LLMSelector iteratively selects one module and allocates to it the model with the highest module-wise performance, as estimated by an LLM, until no further gain is possible. LLMSelector is applicable to any compound system with a bounded number of modules, and its number of API calls scales linearly with the number of modules, achieving high-quality model allocation both empirically and theoretically. Experiments with popular compound systems such as multi-agent debate and self-refine using LLMs such as GPT-4o, Claude 3.5 Sonnet and Gemini 1.5 show that LLMSelector confers 5%-70% accuracy gains compared to using the same LLM for all modules.

模型选择多模型协同大模型优化

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