用10个小型开源模型组合,超越GPT-4系列
The Avengers: A Simple Recipe for Uniting Smaller Language Models to Challenge Proprietary Giants
- 将查询聚类后,选各簇表现最好模型协同生成
- 在15个数据集上平均性能超GPT-4o、4.1和4.5
- 对参数、嵌入模型等变化鲁棒,适合资源有限者
开源小模型能否挑战大厂闭源巨头?本文提出Avengers——一种简单方法,整合10个约70亿参数的开源模型。通过嵌入、聚类、打分、投票四步:先用文本嵌入编码查询,按语义聚类;再评估各模型在每类中的表现;推理时将查询分配至最近聚类,选取该簇最优模型重复采样并投票输出。实验显示,该方法在15个涵盖数学、编程、逻辑推理、常识与情感任务的数据集上,平均性能超越GPT-4o、4.1和4.5。尤其在数学任务上比GPT-4.1高18.21%,代码任务高7.46%。且对嵌入模型、聚类算法、集成策略及簇数变化均保持稳定,具备强泛化能力。
原文摘要 · Abstract (English)
Proprietary giants are increasingly dominating the race for ever-larger language models. Can open-source, smaller models remain competitive across a broad range of tasks? In this paper, we present the Avengers -- a simple recipe that leverages the collective intelligence of these smaller models. The Avengers builds upon four lightweight operations: (i) embedding: encode queries using a text embedding model; (ii) clustering: group queries based on their semantic similarity; (iii) scoring: scores each model's performance within each cluster; and (iv) voting: improve outputs via repeated sampling and voting. At inference time, each query is embedded and assigned to its nearest cluster. The top-performing model(s) within that cluster are selected to generate the response with repeated sampling. Remarkably, with 10 open-source models (~7B parameters each), the Avengers surpasses GPT-4o, 4.1, and 4.5 in average performance across 15 diverse datasets spanning mathematics, coding, logical reasoning, general knowledge, and affective tasks. In particular, it surpasses GPT-4.1 on mathematics tasks by 18.21% and on code tasks by 7.46%. Furthermore, the Avengers delivers superior out-of-distribution generalization, and remains robust across various embedding models, clustering algorithms, ensemble strategies, and values of its sole parameter -- the number of clusters.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。