Training LLMs for Multi-Agent Collaboration with Verifiable Information Rewards
cs.AI, cs.CL, cs.LG
Submitted: 2026-10-13
Updated: 2026-10-13
Comments: 23 pages
License: http://creativecommons.org/licenses/by/4.0/
The gist: Effective collaboration among large language model (LLM) agents in multi-agent systems requires sharing relevant evidence, requesting missing information, and incorporating teammates' contributions
Terminology
Abstract
Effective collaboration among large language model (LLM) agents in multi-agent systems requires sharing relevant evidence, requesting missing information, and incorporating teammates' contributions into decisions. Training these behaviors is challenging because final task outcomes provide limited feedback on the value of individual messages. We introduce CoSIR (Collaboration via Surprisal-based Information Rewards), a post-training approach that uses a simple, verifiable environment to teach collaboration behaviors that generalize to more complex, unseen tasks. We propose a Bayesian evaluator that scores shared information by its reduction in team surprisal, defined as the sum of the negative log probabilities assigned to the correct option given each agent's available evidence. Requests are scored by the potential reduction in the same quantity if the requested information were truthfully provided. Separate rewards encourage agents to use the received information in their private choices and calculations. We post-train Qwen3-4B, Qwen3-8B, and Gemma-3-4B-it through self-play and evaluate transfer to seven unseen natural-language collaboration tasks. CoSIR achieves the highest aggregate score, while its performance in general tasks remains competitive. Behavioral analyses show increased disclosure of private evidence and fewer judged failures to use teammates' input.
Sources
- SmolLM2: When Smol Goes Big -- Data-Centric Training of a Small Language Model
- Why Do Multi-Agent LLM Systems Fail?
- HIPO: Instruction Hierarchy via Constrained Reinforcement Learning
- Training Verifiers to Solve Math Word Problems
- Gemma 3 Technical Report
- Systematic Failures in Collective Reasoning under Distributed Information in Multi-Agent LLMs
- SPIRAL: Self-Play on Zero-Sum Games Incentivizes Reasoning via Multi-Agent Multi-Turn Reinforcement Learning
- Learning Decentralized LLM Collaboration with Multi-Agent Actor Critic
- Communication and Verification in LLM Agents towards Collaboration under Information Asymmetry
- Proximal Policy Optimization Algorithms
- The Collaboration Tax: How Much LLM Multi-Agent Systems Pay to Coordinate
- Talk Structurally, Act Hierarchically: A Collaborative Framework for LLM Multi-Agent Systems
- RAGEN: Understanding Self-Evolution in LLM Agents via Multi-Turn Reinforcement Learning
- Do as We Do, Not as You Think: the Conformity of Large Language Models
- MathChat: Converse to Tackle Challenging Math Problems with LLM Agents
- Qwen3 Technical Report
- MARSHAL: Incentivizing Multi-Agent Reasoning via Self-Play with Strategic LLMs
- Instruction-Following Evaluation for Large Language Models
Related papers
- MAVEN-T: Reinforced Heterogeneous Distillation for Real-Time Multi-Agent Trajectory Prediction
- Model Discovery Agent: LLM-assisted Bayesian experiment design for data-efficient discovery of mechanistic world models
- Agents' Last Exam
- The Clinician's Veto: Navigating Trust, Liability, and Uncertainty in Autonomous AI Prescribing
- MindHelper: Closed-Loop Embodied Mental-State Reasoning for Precision Intervention
- Incumbent Advantage: Brand Bias and Cognitive Manipulation Dynamics in LLM Recommendation Systems