用多个小模型协作提升推理能力,效果超越大模型。
Slm-mux: Orchestrating small language models for reasoning
- 设计多模型架构SLM-MUX,动态协调多个小模型
- 在MATH等任务上比现有方法最高提升13.4%
- 仅用两个小模型就超过720亿参数的大模型
随着语言模型快速发展,小型语言模型(SLMs)数量显著增长。尽管它们无法达到顶尖精度,但效率更高,且在特定任务中表现优异。这引发一个关键问题:能否将多个SLMs协同成系统,使每个模型各展所长,整体性能优于任一单个模型?现有协同方法主要针对前沿大模型(如GPT-4),在应用于SLMs时表现不佳。为此,我们提出三阶段协同方案:首先引入SLM-MUX多模型架构,有效协调多个SLMs;在此基础上,提出两种优化策略:(i) 模型选择搜索,从池中选出互补性最强的SLMs;(ii) 针对SLM-MUX设计的测试时扩展方法。实验表明,相比现有方法,该方案在MATH上最高提升13.4%,在GPQA上提升8.8%,在GSM8K上提升7.0%。仅使用两个SLMs,SLM-MUX即在GPQA和GSM8K上超越Qwen 2.5 72B,在MATH上持平。理论分析验证了方法优势。额外实验显示,SLM-MUX核心原理可拓展至开放式生成任务(如HumanEval),并适用于其他模型类型,包括前沿大模型与领域微调的SLMs。结果证明,通过该方法,可将SLMs高效协同为更准确、高效的系统。
原文摘要 · Abstract (English)
With the rapid development of language models, the number of small language models (SLMs) has grown significantly. Although they do not achieve state-of-the-art accuracy, they are more efficient and often excel at specific tasks. This raises a natural question: can multiple SLMs be orchestrated into a system where each contributes effectively, achieving higher accuracy than any individual model? Existing orchestration methods have primarily targeted frontier models (e.g., GPT-4) and perform suboptimally when applied to SLMs. To address this gap, we propose a three-stage approach for orchestrating SLMs. First, we introduce SLM-MUX, a multi-model architecture that effectively coordinates multiple SLMs. Building on this, we develop two optimization strategies: (i) a model selection search that identifies the most complementary SLMs from a given pool, and (ii) test-time scaling tailored to SLM-MUX. Our approach delivers strong results: Compared to existing orchestration methods, our approach achieves up to 13.4% improvement on MATH, 8.8% on GPQA, and 7.0% on GSM8K. With just two SLMs, SLM-MUX outperforms Qwen 2.5 72B on GPQA and GSM8K, and matches its performance on MATH. We further provide theoretical analyses to substantiate the advantages of our method. Additional experiments show that the core principle of SLM-MUX extends to open-ended generation tasks (e.g., HumanEval) and benefits other model classes, including frontier LLMs and domain-specific fine-tuned SLMs. In summary, we demonstrate that SLMs can be effectively orchestrated into more accurate and efficient systems through the proposed approach. The project page is available at https://slm-mux.github.io/.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。