MoCo: A One-Stop Shop for Model Collaboration Research
Organizations: University of Washington · Texas A&M University · University of Notre Dame · University of Virginia ·
Abstract
Advancing beyond single monolithic language models (LMs), recent research increasingly recognizes the importance of model collaboration, where multiple LMs collaborate, compose, and complement each other. Existing research on this topic has mostly been disparate and disconnected, from different research communities, and lacks rigorous comparison. To consolidate existing research and establish model collaboration as a school of thought, we present MoCo: a one-stop Python library of executing, benchmarking, and comparing model collaboration algorithms at scale. MoCo features 26 model collaboration methods, spanning diverse levels of cross-model information exchange such as routing, text, logit, and model parameters. MoCo integrates 25 evaluation datasets spanning reasoning, QA, code, safety, and more, while users could flexibly bring their own data. Extensive experiments with MoCo demonstrate that most collaboration strategies outperform models without collaboration in 61.0% of (model, data) settings on average, with the most effective methods outperforming by up to 25.8%. We further analyze the scaling of model collaboration strategies, the training/inference efficiency of diverse methods, highlight that the collaborative system solves problems where single LMs struggle, and discuss future work in model collaboration, all made possible by MoCo. We envision MoCo as a valuable toolkit to facilitate and turbocharge the quest for an open, modular, decentralized, and collaborative AI future.
Figures & tables
| Model Pool 1: Specialized LMs | Model Pool 2: General-Purpose LMs | |||||||||||||
| QA | math | reason | safety | code | IF | avg. | QA | math | reason | safety | code | IF | avg. | |
| Best Single | 0.588 | 0.774 | 0.418 | 0.586 | 0.553 | 1.387 | 0.549 | 0.504 | 0.834 | 0.413 | 0.495 | 0.614 | 8.402 | 0.582 |
| Cascade | 0.597 | 0.805 | 0.387 | 0.631 | 0.597 | 1.385 | 0.565 | 0.540 | 0.831 | 0.423 | 0.457 | 0.632 | 9.289 | 0.591 |
| Graph Routing | 0.610 | 0.780 | 0.387 | 0.638 | 0.579 | 1.676 | 0.565 | 0.592 | 0.836 | 0.487 | 0.567 | 0.790 | 7.516 | 0.644 |
| Prompt Routing | 0.596 | 0.803 | 0.411 | 0.571 | 0.597 | 1.381 | 0.558 | 0.579 | 0.837 | 0.461 | 0.557 | 0.754 | 10.276 | 0.649 |
| Switch Generation | 0.587 | 0.733 | 0.382 | 0.652 | 0.412 | -1.060 | 0.493 | 0.491 | 0.860 | 0.481 | 0.521 | 0.395 | 17.234 | 0.625 |
Appendix figures & tables6 assets
Supplementary material from the paper’s appendix.
Appendix
| mmlu-redux | gsm8k | bbh | |
|---|---|---|---|
| w/o model 1 | 0.572 | 0.729 | 0.539 |
| w/o model 2 | 0.581 | 0.759 | 0.561 |
| w/o model 3 | 0.589 | 0.822 | 0.565 |
| w/o model 4 | 0.593 | 0.828 | 0.588 |
| w/o model 5 | 0.598 | 0.872 | 0.598 |
| avg | 0.587 | 0.802 | 0.570 |
| Dataset | Source | Size | |
|---|---|---|---|
| dev | test | ||
| AGIeval | ( Zhong et al., 2024 ) | 1156 | 1156 |
| Alpaca | ( Dubois et al., 2023 ) | 2000 | 1000 |
| ARC-Challenge | ( Clark et al., 2018 ) | 299 | 1172 |
| BigBench-hard | ( Suzgun et al., 2023 ) | 1000 | 1000 |
| BLEND | ( Myung et al., 2024 ) | 1000 | 1000 |
| Method | Training | Inference | Notes |
| Cascade | / | Each level defer 50% | |
| Graph Router | Omit GNN training | ||
| Prompt Router | / | Router | |
| LLM Router | Router | ||
| Switch Generate | Switcher , rounds , path size patch | ||
| Mentor Collab | Omit MLP training |
Multi-LLM Collaboration
Multi-LLM collaboration studies how large language models share work, exchange information, and combine outputs to improve capability, reliability, or cost.
See all 79 papers