Compositional RL: Evaluating and Training LLMs for Collaborative Reasoning
cs.CL, cs.AI, cs.LG
Submitted: 2026-10-12
Updated: 2026-10-12
Code: https://github.com/BunsenFeng/model_collaboration
License: http://creativecommons.org/licenses/by/4.0/
The gist: Language models are often evaluated alone but increasingly deployed together, routed between, made to debate, ensembled, or collaboratively trained.
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.
Sources
- Program Synthesis with Large Language Models
- CulturalBench: A Robust, Diverse, and Challenging Cultural Benchmark by Human-AI CulturalTeaming
- Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challenge
- How to Train a Leader: Hierarchical Reasoning in Multi-Agent LLMs
- Heterogeneous Swarms: Jointly Optimizing Model Roles and Weights for Multi-LLM Systems
- Don't Throw Away Your Pretrained Model
- MoCo: A One-Stop Shop for Model Collaboration Research
- The Single-Multi Evolution Loop for Self-Improving Model Collaboration Systems
- Prompt-to-Leaderboard
- DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learning
- SPARTA ALIGNMENT: Collectively Aligning Multiple Language Models through Combat
- Tulu 3: Pushing Frontiers in Open Language Model Post-Training
- Chain-of-Agents: End-to-End Agent Foundation Models via Multi-Agent Distillation and Agentic RL
- Agents that Matter: How Subtle Choices Shape Agent Attribution in Multi-Agent Systems
- LLM Chemistry Estimation for Multi-LLM Recommendation
- Proximal Policy Optimization Algorithms
- DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Models
- Composition-RL: Compose Your Verifiable Prompts for Reinforcement Learning of Large Language Models
- Understanding and Optimizing Agentic Workflows via Shapley value
- Harness-Bench: Evaluating Agent Harnesses across Models and Realistic Workflows
Related papers
- Exploring Solution Divergence and Its Effect on Large Language Model Problem Solving
- Ishigaki-IDS-Bench: A Benchmark for Generating Information Delivery Specification from BIM Information Requirements
- Subliminal Steering: Stronger Encoding of Hidden Signals
- MedStruct-S: A Benchmark for Key Discovery, Key-Conditioned QA and Semi-Structured Extraction from OCR Clinical Reports
- The End of Transformers? On Challenging Attention and the Rise of Sub-Quadratic Architectures
- Untangling the Mechanisms of Misleading Context in Medical Question Answering