TeamGym: A Gym for Team Decision-Making by AI Agents with Distributed Information
Organizations: MIT · KAIST · UT Austin · University of Maryland
Abstract
Many real-world decisions require teams to integrate information distributed across participants. Yet AI agents are still primarily trained and benchmarked in isolation. We introduce TeamGym, an environment for evaluating and training language model-based agents on team decision-making problems constructed from real-world legal, finance, medical, and software engineering corpora. TeamGym introduces a method to identify a minimal set of facts, each element of which is necessary and all of which are collectively sufficient for a correct final decision, which we call pivotal information. This enables TeamGym to both control the distribution of pivotal information and track how well agents identify and integrate pivotal information when making team decisions. Examining 8 language models on 12 source corpora, we find that current AI agents often fail to effectively share information: Distributing pivotal information across participants lowers success rate by 22.9 percentage points on average relative to giving the same information to a single model (a communication gap). We also document that participants do not effectively retrieve decision-relevant information on their own: A team's success rate is lowered by a further 19.9 points when participants must identify and extract pivotal information from source material rather than receiving it directly (a retrieval gap). To help remedy these deficits, we use TeamGym to train policies through reinforcement learning on team decision loss. Across two policy-training experiments, we find that training improves final-decision quality and induces emergent coordination in participation, speaker selection, and communication, even though the reward evaluates only the final decision.
Figures & tables
| Domain | Source corpora | Decisions |
| Legal | CUAD, MAUD, eCFR Casebook, MSLR, Multi-LexSum, SARA | Contract profiles, merger reviews, compliance and enforcement, legal and procedural conclusions, and tax calculations |
| Finance | FinDVer, FinLongDocQA | SEC-filing findings, calculations, comparisons, and classifications |
| Medical | MLCR, EvidenceBench | Evidence-grounded medical reviews and biomedical answers |
| Software engineering | SWE Atlas–Codebase QnA, DeepSWE | Codebase QA and multi-turn, long-horizon repository-level code repair with functionally evaluated patches |
| Legal | Finance, medical, and software engineering | |||||
| Source corpus | Constructed problem instances | Evaluated problem instances | Domain | Source corpus | Constructed problem instances | Evaluated problem instances |
| CUAD | 538 | 30 | Finance | FinDVer | 389 | 30 |
| MAUD | 557 | 30 | FinLongDocQA | 249 | 30 | |
| eCFR Casebook | 18 | 18 | Medical | MLCR | 29 | 29 |
| MSLR | 11 | 11 | EvidenceBench | 45 | 45 | |
| Multi-LexSum | 29 | 29 | Software engineering | SWE Atlas–Codebase QnA | 32 | 32 |
| Measurement | Base policies | Two-stage training | Joint two-policy training |
| Every participant queried | 80.6 | 66.7 | 100.0 |
| Mean words per query | 14.0 | 48.0 | 12.5 |
| Question copied in first query | 0.0 | 88.6 | 0.0 |
| Later queries reusing preceding-reply terms | 3.2 | 42.9 | 13.5 |
Appendix figures & tables47 assets
Supplementary material from the paper’s appendix.
Appendix
| LLM-judged score | Final-decision completion | Same final- decision policy | PIUs appearing in the communication transcript | Exactly repeated public messages | |
| FinDVer validation | 21.85 41.08 | 51.3 84.6 | 34.10 46.72 | 38.0 53.9 | 18.9 50.9 |
| FinLongDocQA test | 33.50 41.13 | 65.0 92.5 | 49.60 44.82 | 63.1 63.1 | 23.1 48.6 |
| Source corpus | Underlying source materials | Reference decision | Problem family |
| CUAD | Expert-annotated commercial contracts | Constructed alternative | Contract-profile decision |
| MAUD | Expert-curated merger-agreement provisions | Constructed alternative | Merger-review decision |
| eCFR Casebook | Regulations and controlled transaction facts | Written conclusion | Compliance analysis |
| MSLR | Anonymized enforcement case files | Final sanction record | Enforcement conclusion |
| Multi-LexSum | Documents from a litigation collection | Case-grounded decision | Procedural analysis |
| SARA | Tax facts and supplied statutory provisions | Numeric result | Tax calculation |
| Source corpus | Problem instances | PIU distributions | Participants/ distribution | PIUs | Source-material tok./ distribution | Source-material tok./ participant | PIU tok./ unit |
| CUAD | 30 | 42 | 3.0 | 2.0 | 15,644.9 | 5,215.0 | 116.4 |
| MAUD | 30 | 53 | 3.5 | 3.4 | 70,859.8 | 20,759.0 | 174.8 |
| eCFR Casebook | 18 | 61 | 6.7 | 6.3 | 68,383.3 | 10,270.3 | 96.3 |
| MSLR | 11 | 20 | 6.3 | 6.0 | 35,948.0 | 5,688.9 | 141.7 |
| Multi-LexSum | 29 | 113 | 10.4 | 10.4 | 15,621.4 | 1,535.4 | 69.2 |
| SARA | 45 | 225 | 8.9 | 8.7 | 97,936.4 | 10,969.9 | 64.3 |
| CUAD label family | Information in annotated spans | Decision objective |
| Source Code Escrow | deposit, release trigger, post-release use right, update duty, cost | source-code continuity protection strength |
| ROFR/ROFO/ROFN | covered opportunity, notice, exercise period, matching or negotiation right, expiration | transaction restriction strength |
| Insurance | coverage type, coverage limit, certificate, cancellation notice, additional insured, remedy | insurance burden or protection level |
| Audit Rights | recordkeeping, records review, facility or system inspection, notice, frequency, cost | audit burden level |
| License Grant | grant, field of use, territory, duration, sublicense, use restriction | license sufficiency for intended use |
| Minimum Commitment | amount, period, purchase or revenue metric, shortfall remedy, exceptions | commercial commitment materiality |
| CUAD label family | Example participant descriptions |
| Source Code Escrow | deposit reviewer; release-condition reviewer; use-right reviewer; update/cost reviewer |
| ROFR/ROFO/ROFN | covered-opportunity reviewer; notice reviewer; exercise-period reviewer; post-expiration reviewer |
| Insurance | coverage-type reviewer; coverage-limit reviewer; certificate/notice reviewer; special-coverage reviewer |
| Audit Rights | records reviewer; access reviewer; operations-inspection reviewer; process/cost reviewer |
| MAUD category | Information in provision excerpts | Decision objective |
| Deal protection and related provisions | no-shop limits, fiduciary-out conditions, matching periods, superior-proposal conditions, and termination-fee provisions | board-process or termination-trigger profile |
| Conditions to closing | representation bring-down timing and standard, materiality treatment, and fundamental-representation treatment | closing-condition risk profile |
| Material adverse effect | base definition, carveouts, disproportionate-impact treatment, and pandemic or market exceptions | material-adverse-effect risk-allocation profile |
| Operating and efforts covenant | ordinary-course covenant, negative covenants, buyer consent rights, regulatory-efforts language, and emergency exceptions | pre-closing control profile |
| Ordering | Mean adjacent cosine similarity | Episode success (%) | LLM-judged score (%) | Answer-component coverage (%) |
| Seeded random | .449 | 13.8 6.4 | 65.2 4.3 | 80.7 3.4 |
| Maximum similarity | .598 | 27.6 8.3 | 71.9 3.5 | 87.4 2.8 |
| Participant descriptions | Episode success rate (%) | LLM-judged score (%) | Answer-component coverage (%) |
| Constructed from participants’ assigned PIUs | 24.1 8.1 | 77.6 1.9 | 84.8 2.4 |
| Same descriptions permuted across participants | 0.0 0.0 | 46.9 4.8 | 67.2 3.6 |
| Source corpus | Underlying source materials | Reference decision | Problem family |
| FinDVer | SEC filings with source questions and annotated contexts | Written filing-review answer | Filing-review reasoning |
| FinLongDocQA | Company-year annual reports with source questions and answers | Written annual-report answer | Annual-report reasoning |
| PIU-distribution condition | FinDVer | FinLongDocQA |
| One PIU per participant | 30 | 30 |
| Pairwise PIU overlap | 26 | 23 |
| Extra participant without a PIU | 30 | 28 |
| One common PIU | 28 | 23 |
| All-but-one PIU concentration | 30 | 29 |
| PIU distributions, total | 144 | 133 |
| Statistic | FinDVer | FinLongDocQA |
| Source questions in the complete release | 2,400 | 7,527 |
| Source questions considered during construction | 700 | 7,527 |
| Available underlying source materials | 255 filings | 1,329 company-years |
| Constructed problem instances | 389 | 249 |
| Source corpus | Problem instances | PIU distributions | Participants/ distribution | PIUs | Source-material tok./ distribution | Source-material tok./ participant | PIU tok./ unit |
| FinDVer | 30 | 144 | 4.5 | 5.1 | 44,038.1 | 11,887.5 | 271.3 |
| FinLongDocQA | 30 | 133 | 3.1 | 3.3 | 91,711.8 | 31,168.6 | 546.0 |
| All source corpora | 60 | 277 | 3.8 | 4.2 | 67,874.9 | 21,528.1 | 408.6 |
| Source corpus | Underlying source materials | Reference decision | Problem family |
| MLCR | Case-level visit-summary documents with source questions and answers | Written medical-record review | Medical-record reasoning |
| EvidenceBench | Biomedical-paper sentence pools with hypotheses and evidence annotations | Written biomedical-evidence answer | Biomedical-evidence reasoning |
| PIU-distribution condition | MLCR | EvidenceBench |
| One PIU per participant | 29 | 45 |
| Pairwise PIU overlap | 19 | 36 |
| Extra participant without a PIU | 28 | 45 |
| One common PIU | 19 | 36 |
| All-but-one PIU concentration | 19 | 45 |
| PIU distributions, total | 114 | 207 |
| Statistic | MLCR | EvidenceBench |
| Source questions or hypotheses in the construction pool | 205 source questions | 293 test-split hypotheses |
| Available underlying source materials | 1,071 visit-summary documents | 293 biomedical-paper sentence pools |
| Evaluated problem instances | 29 | 45 |
| Evaluated PIU distributions | 114 | 207 |
| Source corpus | Problem instances | PIU distributions | Participants/ distribution | PIUs | Source-material tok./ distribution | Source-material tok./ participant | PIU tok./ unit |
| MLCR | 29 | 114 | 3.2 | 3.2 | 38,119.9 | 13,080.7 | 208.5 |
| EvidenceBench | 45 | 207 | 4.0 | 4.0 | 5,678.0 | 1,727.4 | 62.7 |
| All source corpora | 74 | 321 | 3.7 | 3.7 | 18,391.7 | 6,176.7 | 119.8 |
| Source corpus | Problem family | Evaluation criteria | Evaluated problem instances |
| SWE Atlas Codebase QnA | Multi-file and runtime code comprehension | Every answer component is complete and correct | 32 |
| DeepSWE | Long-horizon repository-level code repair | Official repository verifier passes | 12 |
| Source corpus | Problem instances | PIUs/ instance | Code/source units/instance | Runtime observations/instance | Full source-material tok./instance | PIU tok./ unit |
| SWE Atlas | 32 | 7.3 | 236.5 | 0.0 | 261,951.8 | 130.2 |
| DeepSWE | 12 | 3.6 | 16.8 | 7.2 | 1,140,635.7 | 3570.6 |
| Source corpus | Leave-one-out pivotality | Participant-level source-material check |
| MAUD | 64/67 (95.5%) [89.5, 100.0] | 18/20 (90.0%) [69.9, 97.2] |
| FinLongDocQA | 65/67 (97.0%) [92.3, 100.0] | 20/20 (100.0%) [83.9, 100.0] |
| Domain | Source corpus | Problem instances | Models | Single-Agent All-PIU | Multi-Agent PIU-Only | Multi-Agent Full-Context |
| Legal | CUAD | 30 | 8 | 90.4 1.7 | 72.5 2.7 | 62.9 2.8 |
| MAUD | 30 | 8 | 88.3 1.8 | 67.9 2.2 | 60.0 2.7 | |
| eCFR Casebook | 18 | 8 | 88.9 2.2 | 47.9 3.5 | 18.1 2.5 | |
| MSLR | 11 | 8 | 57.9 4.0 | 45.5 4.6 | 8.0 2.6 | |
| Multi-LexSum | 29 | 8 | 62.5 2.3 | 30.2 2.3 | 20.2 2.1 | |
| SARA | 45 | 8 | 63.6 1.9 | 37.2 2.0 | 8.1 1.4 |
| Domain | Source corpus | Model | Problem instances | Single-Agent All-PIU | Single-Agent Full-Context | Multi-Agent PIU-Only | Multi-Agent Full-Context |
|---|---|---|---|---|---|---|---|
| Legal | CUAD | GPT-5.5 | 30 | 96.7 3.3 | 86.7 6.2 | 93.3 4.6 | 90.0 5.5 |
| Claude Sonnet 4.5 | 30 | 100.0 0.0 | 90.0 5.5 | 53.3 9.1 | 43.3 9.0 | ||
| Claude Haiku 4.5 | 30 | 90.0 5.5 | 93.3 4.6 | 70.0 8.4 | 60.0 8.9 | ||
| Gemma 3 4B IT | 30 | 60.0 8.9 | 30.0 8.4 | 50.0 9.1 | 40.0 8.9 | ||
| Gemma 3 12B IT | 30 | 90.0 5.5 | 53.3 9.1 | 70.0 8.4 | 36.7 8.8 | ||
| Gemma 3 27B IT | 30 | 93.3 4.6 | 50.0 9.1 | 80.0 7.3 | 70.0 8.4 |
| Evaluation condition | Strict success | Valid final decision, unsuccessful | Structured-interface failure | Success given a valid final decision |
| Single-Agent All-PIU | 982/1,304 (75.3%) | 318/1,304 (24.4%) | 4/1,304 (0.3%) | 75.5% |
| Multi-Agent PIU-Only | 650/1,304 (49.8%) | 531/1,304 (40.7%) | 123/1,304 (9.4%) | 55.0% |
| Multi-Agent Full-Context | 404/1,304 (31.0%) | 747/1,304 (57.3%) | 153/1,304 (11.7%) | 35.1% |
| Domain | Model | Planned | Evaluated, (%) | Context unavailable, (%) | Output failures, (%) | Other failures, (%) |
| Legal | GPT-5.5 | 163 | 163 (100.0) | 0 (0.0) | 0 (0.0) | 0 (0.0) |
| Legal | Claude Sonnet 4.5 | 163 | 156 (95.7) | 0 (0.0) | 7 (4.3) | 0 (0.0) |
| Legal | Claude Haiku 4.5 | 163 | 161 (98.8) | 0 (0.0) | 2 (1.2) | 0 (0.0) |
| Legal | Gemma 3 4B IT | 163 | 147 (90.2) | 16 (9.8) | 0 (0.0) | 0 (0.0) |
| Legal | Gemma 3 12B IT | 163 | 147 (90.2) | 16 (9.8) | 0 (0.0) | 0 (0.0) |
| Legal | Gemma 3 27B IT | 163 | 147 (90.2) | 16 (9.8) | 0 (0.0) | 0 (0.0) |
| Participant model | Final decision maker | Effect | 95% CI | ||
| Qwen3-32B | Qwen3-32B | 54.4% | 69.0% | ||
| Qwen3-32B | GPT-5.5 | 69.6% | 75.3% | ||
| Qwen3-235B-A22B | Qwen3-235B-A22B | 60.1% | 77.8% | ||
| Qwen3-235B-A22B | GPT-5.5 | 78.5% | 86.7% |
| Participant model | Final decision maker | Effect | 95% CI |
| Qwen3-32B | Qwen3-32B | ||
| Qwen3-32B | GPT-5.5 | ||
| Qwen3-235B-A22B | Qwen3-235B-A22B | ||
| Qwen3-235B-A22B | GPT-5.5 |
| Evaluation condition | PIU-distribution condition | Broadcast query | Round robin | Current-speaker selection | Simultaneous responses |
| Multi-Agent PIU-Only joint | Pairwise PIU overlap | 1.4 * | 1.5 * | 1.9 * | 1.5 |
| Extra participant without a PIU | 1.5 | 1.3 | 1.6 | 1.6 | |
| One common PIU | 1.5 | 1.5 | 1.8 | 1.5 * | |
| All-but-one PIU concentration | 1.6 | 1.5 * | 2.1 * | 1.6 | |
| Multi-Agent Full-Context joint | Pairwise PIU overlap | 1.2 | 1.2 | 1.5 | 1.3 |
| Extra participant without a PIU | 1.3 | 1.2 | 1.3 | 1.1 |
| Evaluation condition | PIU-distribution condition | Broadcast query | Round robin | Current-speaker selection | Simultaneous responses |
| Multi-Agent PIU-Only joint | Pairwise PIU overlap | 1.5 * | 1.5 * | 1.9 * | 1.5 |
| Extra participant without a PIU | 1.5 | 1.3 | 1.6 | 1.6 | |
| One common PIU | 1.5 | 1.5 | 1.8 | 1.6 | |
| All-but-one PIU concentration | 1.6 | 1.5 * | 2.1 * | 1.6 | |
| Multi-Agent Full-Context joint | Pairwise PIU overlap | 1.2 | 1.2 * | 1.6 | 1.3 |
| Extra participant without a PIU | 1.3 | 1.2 | 1.2 | 1.1 |
| PIU distribution | Coordinator-selected speaker | Round robin | Difference |
| One PIU per participant | 62.7% | 80.5% | |
| All-but-one PIU concentration | 36.4% | 67.8% | |
| Interaction (95% CI) | |||
| Coordinator query condition | Interaction (95% CI) | Change (95% CI) |
| Original query without additions | Reference | |
| Original query with additions | ||
| Query removed without additions | ||
| Query removed with additions |
| Original coordinator query | Coordinator query removed | |||
| PIU distribution | Without additions | With additions | Without additions | With additions |
| One PIU per participant | 65.5% | 69.2% | 71.5% | 74.3% |
| All-but-one PIU concentration | 37.3% | 50.6% | 64.1% | 70.9% |
| Comparison | Interaction | 95% CI |
| Add labels and instruction to original coordinator query | ||
| Remove coordinator query without additions | ||
| Remove coordinator query with additions |
| Coordinator- selected speaker | Broadcast query | Round robin | Current-speaker selection | Simultaneous responses | ||
| PIUs only | One PIU per participant | 69.7% | 59.8% | 73.5% | 75.3% | 60.9% |
| All-but-one PIU concentration | 67.9% | 60.9% | 78.0% | 78.5% | 69.7% | |
| PIUs and source material | One PIU per participant | 35.9% | 41.2% | 39.6% | 46.0% | 28.5% |
| All-but-one PIU concentration | 35.6% | 28.0% | 27.8% | 37.1% | 23.2% |
| Communication structure | PIUs only | PIUs and source material | Interaction |
| Coordinator-selected speaker | |||
| Broadcast query | |||
| Round robin | |||
| Current-speaker selection | |||
| Simultaneous responses |
| Communication structure | None | Assigned PIUs only | Assigned no PIU only | Both groups |
| Coordinator-selected speaker | 67.9% | 53.3% | 38.1% | 35.6% |
| Broadcast query | 60.9% | 47.2% | 28.0% | 28.0% |
| Simultaneous responses | 69.7% | 54.8% | 22.5% | 23.2% |
| Source material for participants assigned no PIU | Complete transcript | Messages from participants assigned PIUs only | Effect | 95% CI |
| Absent | 69.9% | 68.9% | ||
| Present | 21.2% | 68.9% | ||
| Interaction | ||||
| Private-note tokens | Message tokens | GPT-5.5 CUAD | GPT-5.5 Multi-LexSum | Qwen3-32B CUAD | Qwen3-32B Multi-LexSum |
| 64 | 32 | 70.8 9.3 | 24.1 7.9 | 79.2 8.3 | 3.4 3.4 |
| 64 | 64 | 70.8 9.3 | 65.5 8.8 | 75.0 8.8 | 0.0 0.0 |
| 64 | 128 | 70.8 9.3 | 72.4 8.3 | 91.7 5.6 | 3.4 3.4 |
| 64 | 256 | 70.8 9.3 | 69.0 8.6 | 83.3 7.6 | 0.0 0.0 |
| 128 | 32 | 79.2 8.3 | 44.8 9.2 | 75.0 8.8 | 3.4 3.4 |
| 128 | 64 | 79.2 8.3 | 62.1 9.0 | 75.0 8.8 | 6.9 4.7 |
| Local-transcript tokens | FIFO | Change from baseline | Random contiguous window | Change from baseline |
| 16 | 82.8 7.0 | -3.4 | 75.9 3.4 | -10.3 |
| 32 | 82.8 7.0 | -3.4 | 83.9 2.0 | -2.3 |
| 64 | 86.2 6.4 | +0.0 | 81.6 5.3 | -4.6 |
| 128 | 79.3 7.5 | -6.9 | 85.1 8.0 | -1.1 |
| 256 | 86.2 6.4 | +0.0 | 82.8 6.9 | -3.4 |
| Unlimited | 86.2 6.4 | 0.0 | — | — |
| Use | CUAD-derived | MAUD-derived | Total |
| Training | 334 | 0 | 334 |
| Validation | 36 | 0 | 36 |
| MAUD-derived evaluation | 0 | 26 | 26 |
| Setting | Participant-only stage | Coordinator-only continuation | Joint two-policy training |
| Updated policy | Participant policy | Coordinator policy | Participant and coordinator policies |
| Fixed policy | Base coordinator policy | Trained participant policy | Neither |
| Training stage | LLM-judged score |
| Base policies | 12.14 |
| Participant-only stage | 16.84 |
| Coordinator-only continuation | 23.33 |
| Case | Base participant policy | Trained participant policy | Observed change |
| Liability limits | Both participant messages deny that the agreement contains liability limits; the final decision receives 0.00. | One participant message includes the mutual exclusion of lost profits and indirect, special, consequential, and punitive damages, including the product context and even-if-advised condition; the final decision receives 0.33. | Participant message includes information needed for a correct final decision |
| Compliance audit and access | Participant messages leave the right holder unspecified and substitute unrelated consumer-credit services; the final decision receives 0.00. | The two participant messages separately include Equidata’s at-any-time site-inspection right and its compliance audit at Equidata’s expense; the final decision receives 0.33 but still merges the two rights. | Different PIUs appear across participant messages |
| Insurance obligations | The first participant message includes the $1M coverage, but the second coordinator query repeats participant A and obtains an unrelated fiber-rights clause; the final decision receives 0.40. | After the trained participant policy produces a different first message, the fixed coordinator observes the changed transcript and queries participant B, which supplies the cancellation, non-renewal, modification, and 30-day notice condition; the final decision receives 0.80. | Changed participant message leads to a different next coordinator query |
| Case | Base coordinator policy | Trained coordinator policy | Observed change |
| Distributor insurance | Participant messages include unrelated regulatory and committee-cost facts; the final decision receives 0.00. | Changed coordinator queries cause participant messages to include both the certificate-on-request clause and the reciprocal insurance obligation; the final decision receives 1.00. | Changed coordinator queries |
| Co-brand minimum commitment | Both participant messages contain the 5,000-unit initial order and the 10,000-unit/32-month guarantee, but the final decision confuses their roles; the final decision receives 0.33. | With substantively unchanged participant messages, the final decision correctly separates the initial order from the longer-term guarantee and receives 1.00. | Changed final decision from the same participant messages |
| FinDVer validation | FinLongDocQA test | |||||
| Measurement | Base | Trained | Base | Trained | ||
| Same final-decision policy: LLM-judged score | 34.10 | 46.72 | +12.62 | 49.60 | 44.82 | |
| Public messages | 7.28 | 12.46 | +5.18 | 9.30 | 13.43 | +4.12 |
| PIUs appearing in the communication transcript (%) | 38.0 | 53.9 | +15.9 | 63.1 | 63.1 | 0.0 |
| PIUs appearing in the communication transcript with no incompatible message (%) | 23.3 | 26.3 | +3.0 | 55.6 | 58.8 | +3.1 |
| Answer components fully supported (%) | 23.5 | 33.0 | +9.5 | 26.2 | 25.0 | |
Multi-Agent Collaboration
Multi-agent collaboration studies how autonomous agents coordinate, share information, and combine capabilities to solve tasks together.
See all 312 papers