One Model, Many Minds: Unlocking Multi-Agent Synergy in a Single Agent via Mixture of Roles
Organizations: University of Illinois Urbana-Champaign · Amazon
Abstract
Specializing Large Language Models (LLMs) toward distinct abilities underpins successes ranging from personalized assistants to multi-agent systems (MAS). Single-agent paradigms rely on pre-defined personas or steering vectors to induce specialization, yet they impose a single fixed specialization that fails to adapt to diverse queries. Conversely, MAS achieves dynamic multi-perspective problem solving by orchestrating agents with distinct text-based roles, but fusing these specializations requires multi-turn interactions that inflate context length and inference cost. To address these limitations, we propose Mixture of Roles (MoRe), which adaptively composes multiple specializations into a single steering vector for single-turn inference. Specifically, MoRe learns a diversified codeboox of steering vectors, each of which encodes a latent role. A query-aware router dynamically fuses the codebook into a steering vector that encompasses multiple roles. By steering the backbone LLM with the composed vector, MoRe enables multi-perspective specialization in a single-agent, single-turn inference process. The proposed MoRe can be efficiently trained via a three-stage SFT curriculum and GRPO post-training, while the backbone LLM remains frozen. Experiments across reasoning and personality benchmarks show that MoRe outperforms single-agent baselines by 2.2% on average, and achieves performance on par with MAS while reducing token cost by 20x.
Figures & tables
| Method | MMLU | TriviaQA | MATH | GSM8K | MedQA | AvgAcc | AvgRank |
|---|---|---|---|---|---|---|---|
| Llama3.1-8B-Instruct | |||||||
| Vanilla | 61.80 | 74.80 | 52.60 | 80.40 | 60.00 | 65.92 | 8.80 |
| Text | 61.00 | 75.57 | 56.09 | 80.06 | 60.31 | 66.61 | 8.60 |
| CAA | 61.34 | 73.77 | 56.31 | 82.69 | 61.06 | 67.03 | 7.60 |
| ActAdd | 61.74 | 79.00 | 56.46 | 82.89 | 61.71 | 68.36 | 5.00 |
| Prompt FT | 62.80 | 77.60 | 58.60 | 78.00 | 63.20 | 68.04 | 4.60 |
| Method | O | C | E | A | N | AvgScore | AvgRank |
|---|---|---|---|---|---|---|---|
| Llama3.1-8B-Instruct | |||||||
| Text | 8.12±1.90 | 7.06±1.25 | 9.21±0.77 | 8.15±1.68 | 9.00±1.76 | 8.31±0.85 | 4.6 |
| CAA | 8.79±0.47 | 7.84±1.02 | 9.56±0.42 | 7.40±1.15 | 8.77±0.88 | 8.47±0.86 | 4.4 |
| NPTI | 9.15±0.71 | 9.86±0.13 | 9.72±0.20 | 9.29±0.40 | 9.84±0.00 | 9.57±0.33 | 2.6 |
| IRIS | 9.29±0.36 | 9.84±0.13 | 9.76±0.27 | 9.06±0.66 | 9.92±0.00 | 9.57±0.38 | 2.4 |
| MoRe | 9.47±0.44 | 9.87±0.14 | 9.81±0.15 | 9.88±0.11 | 9.98±0.02 | 9.80±0.20 | 1.0 |
Appendix figures & tables4 assets
Supplementary material from the paper’s appendix.
Appendix
| Role | MMLU | TriviaQA | MATH | GSM8K | MedQA | AvgAcc |
|---|---|---|---|---|---|---|
| Llama-8B-Instruct | ||||||
| Math. | 64.00 | 73.60 | 58.40 | 81.40 | 61.00 | 67.68 |
| SDE | 59.60 | 75.60 | 55.00 | 79.60 | 54.00 | 64.76 |
| DataSci. | 60.20 | 75.40 | 56.40 | 81.60 | 61.40 | 67.00 |
| Logic. | 61.00 | 72.40 | 54.40 | 77.20 | 55.80 | 64.16 |
| Teacher | 61.80 | 78.20 | 58.20 | 79.80 | 61.00 | 67.80 |
Language Model Steering
Language model steering controls a model’s behavior at inference time by intervening in its representations, components, or decoding process.
See all 275 papers