Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Model Swarms: Collaborative Search to Adapt LLM Experts via Swarm Intelligence

About

We propose Model Swarms, a collaborative search algorithm to adapt LLMs via swarm intelligence, the collective behavior guiding individual systems. Specifically, Model Swarms starts with a pool of LLM experts and a utility function. Guided by the best-found checkpoints across models, diverse LLM experts collaboratively move in the weight space and optimize a utility function representing model adaptation objectives. Compared to existing model composition approaches, Model Swarms offers tuning-free model adaptation, works in low-data regimes with as few as 200 examples, and does not require assumptions about specific experts in the swarm or how they should be composed. Extensive experiments demonstrate that Model Swarms could flexibly adapt LLM experts to a single task, multi-task domains, reward models, as well as diverse human interests, improving over 12 model composition baselines by up to 21.0% across tasks and contexts. Further analysis reveals that LLM experts discover previously unseen capabilities in initial checkpoints and that Model Swarms enable the weak-to-strong transition of experts through the collaborative search process.

Shangbin Feng, Zifeng Wang, Yike Wang, Sayna Ebrahimi, Hamid Palangi, Lesly Miculicich, Achin Kulshrestha, Nathalie Rauschmayr, Yejin Choi, Yulia Tsvetkov, Chen-Yu Lee, Tomas Pfister• 2024

Related benchmarks

TaskDatasetResultRank
General KnowledgeMMLU
MMLU General Knowledge Accuracy52.74
373
MathematicsMATH
MATH Accuracy10.25
172
Language UnderstandingMMLU (test)--
167
Reading ComprehensionDROP
DROP Accuracy31.7
138
Language UnderstandingMMLU-Pro
Accuracy26.8
130
Mathematical ReasoningGSM8K (val)
Accuracy41
115
Multi-task Language UnderstandingMMLU (test)
Normalized Accuracy57.3
107
Multitask Language UnderstandingMMLU (val)
Accuracy64
94
Common Sense ReasoningHELLASWAG (test)
Accuracy58.7
86
Mathematical ReasoningGSM8K (test)
Accuracy (ACC)24.6
76
Showing 10 of 44 rows

Other info

Follow for update