Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

SMoA: Improving Multi-agent Large Language Models with Sparse Mixture-of-Agents

About

While multi-agent systems have been shown to significantly enhance the performance of Large Language Models (LLMs) across various tasks and applications, the dense interaction between scaling agents potentially hampers their efficiency and diversity. To address these challenges, we draw inspiration from the sparse mixture-of-agents (SMoE) and propose a sparse mixture-of-agents (SMoA) framework to improve the efficiency and diversity of multi-agent LLMs. Unlike completely connected structures, SMoA introduces novel Response Selection and Early Stopping mechanisms to sparsify information flows among individual LLM agents, striking a balance between performance and efficiency. Additionally, inspired by the expert diversity principle in SMoE frameworks for workload balance between experts, we assign distinct role descriptions to each LLM agent, fostering diverse and divergent thinking. Extensive experiments on reasoning, alignment, and fairness benchmarks demonstrate that SMoA achieves performance comparable to traditional mixture-of-agents approaches but with significantly lower computational costs. Further analysis reveals that SMoA is more stable, has a greater capacity to scale, and offers considerable potential through hyper-parameter optimization. Code and data will be available at: https://github.com/David-Li0406/SMoA.

Dawei Li, Zhen Tan, Peijia Qian, Yifan Li, Kumar Satvik Chaudhary, Lijie Hu, Jiayi Shen• 2024

Related benchmarks

TaskDatasetResultRank
Mathematical ReasoningMATH
Accuracy78.08
535
Reading ComprehensionRACE high
Accuracy84
295
Instruction FollowingAlpacaEval 2.0--
281
Comprehensive ExaminationAGIEval (test)--
34
Mathematical ReasoningMATH 500
Accuracy73.5
26
General KnowledgeMMLU-Redux
Accuracy84.94
20
Code ReasoningCRUX
Accuracy86.93
16
ReasoningARC Challenge
Accuracy89.4
11
Code GenerationMBPP
Accuracy79.4
11
Biomedical KnowledgeMMLU bio
Accuracy75.7
11
Showing 10 of 15 rows

Other info

Follow for update