Multi-agent cooperation through in-context co-player inference
Organizations: Google, Paradigms of Intelligence Team · Santa Fe Institute
Abstract
Achieving cooperation among self-interested agents remains a fundamental challenge in multi-agent reinforcement learning. Recent work showed that mutual cooperation can be induced between "learning-aware" agents that account for and shape the learning dynamics of their co-players. However, existing approaches typically rely on hardcoded, often inconsistent, assumptions about co-player learning rules or enforce a strict separation between "naive learners" updating on fast timescales and "meta-learners" observing these updates. Here, we demonstrate that the in-context learning capabilities of sequence models allow for co-player learning awareness without requiring hardcoded assumptions or explicit timescale separation. We show that training sequence model agents against a diverse distribution of co-players naturally induces in-context best-response strategies, effectively functioning as learning algorithms on the fast intra-episode timescale. We find that the cooperative mechanism identified in prior work-where vulnerability to extortion drives mutual shaping-emerges naturally in this setting: in-context adaptation renders agents vulnerable to extortion, and the resulting mutual pressure to shape the opponent's in-context learning dynamics resolves into the learning of cooperative behavior. Our results suggest that standard decentralized reinforcement learning on sequence models combined with co-player diversity provides a scalable path to learning cooperative behaviors.
Figures & tables
Appendix figures & tables4 assets
Supplementary material from the paper’s appendix.
Appendix
| Player 2 | |||
| C | D | ||
| Player 1 | C | (1, 1) | (-1, 2) |
| D | (2, -1) | (0, 0) | |
| RL Hyperparameter | Step 1 | Step 2 | Step 3 | Step 4 |
| advantages_normalization | True | False | True | True |
| batch size | 2048 | 2048 | 4096 | 4096 |
| reward_rescaling | ||||
| value_discount ( ) | ||||
| td_lambda ( ) | ||||
| gae_lambda ( ) |
Cooperative MARL
Cooperative multi-agent reinforcement learning (MARL) trains agents to coordinate toward shared goals under partial information and decentralized control.
See all 244 papers