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
Language UnderstandingMMLU (test)--
167
Mathematical ReasoningGSM8K (val)
Accuracy41
108
Multitask Language UnderstandingMMLU (val)
Accuracy64
94
Multi-task Language UnderstandingMMLU (test)
Normalized Accuracy57.3
87
Mathematical ReasoningGSM8K (test)
Accuracy (ACC)24.6
62
Common Sense ReasoningHELLASWAG (test)
Accuracy58.7
56
Commonsense ReasoningHellaSwag (val)
Accuracy63.5
54
Truthfulness EvaluationTruthfulQA (test)--
30
Knowledge ReasoningK-Cross (test)
Accuracy41.1
22
Natural Language Graph ReasoningNLGraph (test)
Accuracy29.7
22
Showing 10 of 26 rows

Other info

Follow for update