Compositional RL: Evaluating and Training LLMs for Collaborative Reasoning
Organizations: University of Washington · NVIDIA
Abstract
Language models are often evaluated alone but increasingly deployed together, routed between, made to debate, ensembled, or collaboratively trained. For a model inside a system, what matters most is not how well it performs on its own, but how much the system gains from its presence. We call this a model's compositional strength and formalize it using the Shapley value of a cooperative game over coalitions of a model pool, and use a model's leave-one-out contribution to the full coalition as a tractable proxy, requiring n+1 rather than 2^n system evaluations. Evaluating individual vs. compositional strengths across five model pools, twelve collaboration methods, and twelve datasets, we find them largely uncorrelated: in 12 of 15 (pool, method-family) settings, individual strength does not translate into compositional advantages, and the coupling is weakest among general-purpose models that practitioners are most likely to deploy. Standalone benchmarks are therefore often poor proxies for models' value in collaborative systems. We then propose compositional reinforcement learning (C-RL), which trains a policy inside a live multi-model system and rewards the outcome produced by the system as a whole, rather than the policy's correctness in isolation. The leave-one-out baseline in Shapley values cancels under a group policy gradient, so this needs no additional rollout cost beyond RL on the collaborative system's outcome and never requires running a coalition without the policy. Across three collaboration protocols and eight tasks spanning reasoning, coding, and scientific discovery, C-RL produces the strongest collaborator in 21 of 24 settings, lifting collaborative performance by +4.0 to +5.4 pts on average (~3x the gain of baselines), up to +13.1 pts.
Figures & tables
| Method | GPQA-dia | Theorem | KernelB | SMDD | AssayB | ChemB | Culture-H | MMLU-PX | Avg. |
|---|---|---|---|---|---|---|---|---|---|
| Collaboration protocol: LLM-Blender | |||||||||
| Base | .818 | .613 | .144 | .635 | .035 | .720 | .806 | .804 | .572 |
| GRPO-indiv | .849 | .647 | .157 | .635 | .033 | .732 | .825 | .822 | .587 |
| DAPO-indiv | .859 | .657 | .129 | .635 | .041 | .750 | .819 | .812 | .588 |
| prompt-collab | .838 | .630 | .159 | .654 | .036 | .742 | .822 | .827 | .588 |
| RAFT-collab | .808 | .645 | .147 | .652 | .031 | .708 | .827 | .796 | .577 |
| Method | KernelBench | AssayBench | ||
|---|---|---|---|---|
| score | score | |||
| Base | .174 | – | .044 | – |
| GRPO-indiv | .185 | .033 | ||
| DAPO-indiv | .223 | .026 | ||
| prompt-collab | .171 | .043 | ||
| RAFT-collab | .215 | .032 | ||
Appendix figures & tables11 assets
Supplementary material from the paper’s appendix.
Appendix
| Method | GPQA-dia | Theorem | KernelB | SMDD | AssayB | ChemB | Culture-H | MMLU-PX | Avg. |
|---|---|---|---|---|---|---|---|---|---|
| Base | .687 | .547 | .103 | .279 | .023 | .694 | .783 | .758 | .484 |
| GRPO-indiv | .808 | .608 | .111 | .322 | .026 | .714 | .832 | .784 | .526 |
| DAPO-indiv | .758 | .640 | .114 | .261 | .024 | .724 | .819 | .820 | .520 |
| prompt-collab | .697 | .580 | .110 | .260 | .026 | .700 | .818 | .796 | .498 |
| RAFT-collab | .768 | .637 | .108 | .410 | .017 | .684 | .806 | .764 | .524 |
| C-RL , LLM-Blender | .889 | .690 | .097 | .297 | .024 | .710 | .842 | .826 | .547 |
| Method | GPQA-dia | Theorem | KernelB | SMDD | AssayB | ChemB | Culture-H | MMLU-PX | Avg. |
|---|---|---|---|---|---|---|---|---|---|
| Collaboration protocol: LLM-Blender | |||||||||
| Base | .818 | .613 | .144 | .635 | .035 | .720 | .806 | .804 | .572 |
| C-RL (ours) | .879 | .708 ∗ | .183 | .686 | .046 | .776 ∗ | .843 ∗ | .840 ∗ | .620 ∗ |
| [.808,.939] | [.662,.752] | [.120,.256] | [.606,.759] | [.024,.069] | [.738,.812] | [.814,.873] | [.808,.872] | [.602,.638] | |
| Collaboration protocol: Multiagent Debate | |||||||||
| Base | .768 | .623 | .155 | .655 | .036 | .720 | .817 | .818 | .574 |
| LLM-Blender | Multiagent Debate | SLM-MUX | ||||
|---|---|---|---|---|---|---|
| Task | rescue | drag | rescue | drag | rescue | drag |
| GPQA-dia | .162 .040 | .030 .051 | .152 .111 | .071 .030 | .172 .172 | .020 .040 |
| Theorem | .098 .048 | .033 .030 | .100 .052 | .025 .037 | .085 .152 | .065 .033 |
| KernelB | .136 .168 | .064 .032 | .144 .160 | .056 .024 | .140 .220 | .120 .020 |
| SMDD | .438 .438 | .080 .088 | .474 .423 | .095 .095 | .504 .526 | .051 .073 |
| AssayB | .006 .009 | .000 .000 | .000 .000 | .000 .003 | – | – |
| option proposed by each model | system | policy len. | ||||
| Policy | gpt-oss | kimi-k3 | nemotron | |||
| What happens if aromatic nitro compounds are mixed with sodium hydroxide? (A) Nothing happens (B) They may explode (C) They decompose (D) They form a salt Correct: B | ||||||
| Base | A | A | B | A | A | 34,350 |
| C-RL | B | A | B | A | B | 4,059 |
| Which of the following statements is true about halogenated organic compounds? (A) They are incompatible with strong oxidizing and reducing agents. (B) They react slowly with metals. (C) The more halogen atoms, the more reactive they become. (D) They are highly reactive. Correct: A | ||||||
| Base | D | A | A | A | D | 7,292 |
| Q. How many ways are there to arrange the letters in the word BANANA up to the symmetries of the word? | |
| Base | The phrase ”up to the symmetries” implies treating the arrangement as a bracelet, where two arrangements are considered identical if one can be rotated or flipped to match the other (the dihedral group ). Applying Burnside’s Lemma, the identity symmetry fixes all 60 distinct linear permutations, while the five non-trivial rotations fix none because the letter counts (3 A’s, 2 N’s, 1 B) cannot form the required periodic patterns. Among the six reflections, the three axes passing through opposite letter positions each fix 4 arrangements (by assigning the AA and NN pairs to the two swapped slots and the remaining A and B to the fixed slots), while the three axes passing between slots fix none. Summing the fixed counts gives , and dividing by the 12 total symmetries yields 6 distinct arrangements. Final Answer: 6 |
| C-RL | First, calculate the number of distinct linear permutations of the letters in BANANA, which is . The phrase ”up to symmetries” for a linear word means we consider an arrangement equivalent to its reverse. Since the letter counts for ’A’ (3) and ’B’ (1) are odd, no arrangement can be a palindrome, meaning every permutation is distinct from its reverse and forms a unique pair. Dividing the total linear permutations by 2 to account for this reversal symmetry gives 30 distinct arrangements. Final Answer: 30 |
| Q. Determine the product obtained when 3,4-dimethylhexanedial is reacted with the following set of reagents: 1. KOH, H2O, THF, Heat 2. CH3CH2MgBr, H3O+ 3. PCC, CH2Cl2 4. O3, H2O A. 3,4-dimethyl-5,6-dioxooctanoic acid D. 3,4-dimethyl-5,6-dioxooctanal | |
| Base | Base-catalyzed aldol condensation of 3,4-dimethylhexanedial forms a cyclohexene derivative, which subsequently undergoes Grignard addition and PCC oxidation to introduce an ethyl ketone group. Ozonolysis cleaves the ring double bond to yield an acyclic chain containing aldehyde and ketone functionalities. The resulting 8-carbon skeleton and carbonyl positions precisely match 3,4-dimethyl-5,6-dioxooctanal, while ozonolysis of an internal alkene does not produce carboxylic acids, eliminating option A. Answer: D |
| C-RL | 3,4-Dimethylhexanedial undergoes an intramolecular aldol condensation under basic conditions to form a five-membered cyclic , -unsaturated aldehyde. Subsequent Grignard addition and PCC oxidation convert the side-chain aldehyde into an ethyl ketone. Finally, oxidative ozonolysis cleaves the ring double bond, opening the ring to yield a linear dicarbonyl carboxylic acid chain. Counting the carbons and functional groups confirms the product is 3,4-dimethyl-5,6-dioxooctanoic acid. Answer: A |
| Model pool | QA | Reasoning | Coding | Pluralism & Safety | Science |
|---|---|---|---|---|---|
| Specialized Expert | 65.8 72.4 (+6.6) | 32.3 45.5 (+13.1) | 44.1 65.5 (+21.4) | 38.7 54.0 (+15.3) | 41.2 55.4 (+14.2) |
| 8-30B Open | 82.9 94.9 (+12.0) | 24.2 35.6 (+11.4) | 70.7 82.5 (+11.8) | 78.2 81.2 (+3.1) | 57.7 60.8 (+3.1) |
| 100B Open | 80.0 83.7 (+3.7) | 48.1 60.0 (+11.9) | 48.8 71.2 (+22.4) | 77.0 82.0 (+4.9) | 53.7 68.3 (+14.6) |
| Proprietary API | 86.8 90.0 (+3.2) | 58.5 69.4 (+10.9) | 66.7 94.1 (+27.5) | 32.1 37.7 (+5.6) | 73.1 80.6 (+7.5) |
| Diverse Academic | 66.1 70.0 (+3.9) | 32.4 65.3 (+32.9) | 37.4 52.7 (+15.4) | 64.6 71.6 (+7.0) | 50.4 55.1 (+4.7) |
| Pool | Members | |
|---|---|---|
| Specialized Expert | 4 | Qwen2.5-7B-Instruct, Qwen2.5-Math-7B, Qwen2.5-Coder-7B-Instruct, Qwen2.5-7B-Instruct-1M |
| 8–30B Open | 7 | Qwen3-14B, Nemotron-3-Nano-30B-A3B, gpt-oss-20b, Llama-3.1-8B-Instruct, gemma-3-12b-it, phi-4, DeepSeek-R1-Distill-Qwen-14B |
| 100B Open | 4 | gpt-oss-120b, deepseek-v4-pro, kimi-k2.6, nemotron-3-ultra |
| Proprietary API | 3 | gpt-5.5, gemini-3.5-flash, bedrock-claude-opus-4-8 |
| Diverse Academic ( Feng et al., 2026e ) | 8 | parti_0_full, parti_1_full, parti_2_full, parti_3_full, parti_4_full, parti_5_full, parti_6_full, parti_7_full |
| Dataset | Category | Test items |
|---|---|---|
| ARC-Challenge ( Clark et al., 2018 ) | QA | 1172 |
| MMLU-Redux ( Gema et al., 2025 ) | QA | 1000 |
| GPQA-Diamond ( Rein et al., 2024 ) | Reasoning | 99 |
| BBH ( Suzgun et al., 2023 ) | Reasoning | 1000 |
| TheoremQA ( Chen et al., 2023 ) | Reasoning | 400 |
| HumanEval ( Chen et al., 2021 ) | Coding | 114 |
| Dataset | Category | Development | Test |
| GPQA-Diamond ( Rein et al., 2024 ) | Answer-matching | 99 | 99 |
| TheoremQA ( Chen et al., 2023 ) | Answer-matching | 400 | 400 |
| MMLU-ProX ( Xuan et al., 2025 ) | Answer-matching | 500 | 500 |
| CulturalBench-Hard ( Chiu et al., 2024 ) | Answer-matching | 613 | 613 |
| ChemBench ( Mirza et al., 2025 ) | Answer-matching | 498 | 500 |
| KernelBench ( Ouyang et al., 2025 ) | Execution-verified | 125 | 125 |
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 85 papers