Multi-Agent & Goal-Conditioned RL
Policy Generalization Across Large Agent-Count Shifts in Multi-Agent Reinforcement Learning
Generated automatically from the limitations stated in 3 papers (ICLR), listed under Evidence. It is not a paper, and it does not come from papers submitted to CSPaper.
The problem
Multi-agent and goal-conditioned reinforcement learning policies trained on small team sizes fail catastrophically when evaluated on larger agent counts, collapsing to near-zero success rates on unseen large-scale tasks. Existing architectures and skill representations fail to capture distribution shifts under scale, forcing practitioners to retrain separate policies from scratch for every team size. This inability to transfer learned coordination across team scales prevents the deployment of multi-agent policies to dynamic environments where entity counts vary at runtime.
Why it matters
Enables multi-agent policies trained in lightweight, few-agent simulations to transfer zero-shot or with minimal fine-tuning to large-scale coordination problems.
Ways to approach it
Prior-work checks are free with an account. Results someone already ran are shown to everyone.
- 1
Benchmark permutation-invariant and graph-based policy representations trained on 3-5 agents across standard multi-agent testbeds (e.g., SMAC 10m12m, 13m15m), measuring zero-shot and few-shot win rates as agent scale increases.
- 2
Evaluate entity-centric goal conditioning and attention-pooling mechanisms under varying agent counts, measuring task-specific embedding overlap and policy degradation during scale shifts.
- 3
Test curriculum-based fine-tuning schedules that progressively expand the observation and action masking space, quantifying the sample efficiency required to recover baseline performance at 10+ agents.
Have a different approach?
Describe how you would tackle this problem and we'll look for papers that already do it. Free; your text stays private.
Why it might fail
If the coordination dynamics fundamentally change at scale such that few-agent coordination strategies share zero actionable structure with large-team strategies, zero-shot or few-shot transfer across large agent-count shifts will remain intractable.
Evidence
Each paper's own statement of the limitation, verbatim.
- STAIRS-Former: Spatio-Temporal Attention with Interleaved Recursive Structure TransFormer for Offline Mulit-task Multi-agent Reinforcement LearningICLR 2026
Near-zero performance remains on very large unseen tasks (10m12m, 13m15m ~0-5%), so generalization still fails at large agent-count shifts
- BRIDGE: Bi-level Reinforcement Learning for Dynamic Group Structure in Coalition Formation GamesICLR 2026
Generalization quality decays quickly with agent count: pretraining on 3 agents yields only ~48% of optimum at 10 agents even with 200 fine-tuning episodes
- Learning Generalizable Skills from Offline Multi-Task Data for Multi-Agent CooperationICLR 2025
Performance on hard unseen tasks with many agents is near-zero (e.g., 10m12m: 0.6 win rate, 13m15m: 1.4), and task-specific skill embeddings for large-scale tasks (10m, 12m) overlap, indicating failure to capture distribution shifts at scale
Nearest existing work
- A Structured Prediction Approach for Generalization in Cooperative Multi-Agent Reinforcement LearningNeurIPS 2019
- Self-Organized Group for Cooperative Multi-agent Reinforcement LearningNeurIPS 2022
- Celebrating Diversity in Shared Multi-Agent Reinforcement LearningNeurIPS 2021
- Coach-Player Multi-agent Reinforcement Learning for Dynamic Team CompositionICML 2021
- Revisiting Cooperative Off-Policy Multi-Agent Reinforcement LearningICML 2025
- Complementary Attention for Multi-Agent Reinforcement LearningICML 2023
- Multi-Agent Actor-Critic for Mixed Cooperative-Competitive EnvironmentsNeurIPS 2017
- Evolutionary Population Curriculum for Scaling Multi-Agent Reinforcement LearningICLR 2020
- Variational Automatic Curriculum Learning for Sparse-Reward Cooperative Multi-Agent ProblemsNeurIPS 2021
- Role-Level Inductive Bias for Cross-Task Generalization in Multi-Agent Reinforcement LearningICML 2026
- Scalable Safe Policy Improvement for Factored Multi-Agent MDPsICML 2024
- Actor-Attention-Critic for Multi-Agent Reinforcement LearningICML 2019
- Promoting Coordination through Policy Regularization in Multi-Agent Deep Reinforcement LearningNeurIPS 2020
- Scalable Multi-Agent Reinforcement Learning through Intelligent Information AggregationICML 2023
- Discovering Generalizable Multi-agent Coordination Skills from Multi-task Offline DataICLR 2023