Community-based Multi-Agent Reinforcement Learning with Transfer and Active Exploration
Zhaoyang Shi
TL;DR
The paper addresses coordination in multi-agent reinforcement learning (MARL) with heterogeneous agents by introducing a community-based MARL framework that uses latent overlapping communities and mixed memberships. It develops a two-time-scale actor-critic algorithm where each community maintains a shared $Q$-function and agents aggregate these estimates according to personalized membership weights, enabling transfer learning and active exploration with provable convergence under linear function approximation. A spectral MSCORE method estimates mixed memberships from observed interactions, enabling rapid transfer to new agents or tasks through $Q^{new}(s,a)=\sum_k \gamma^{new}(k) Q(s,a;\omega^{(k)})$. Active learning prioritizes uncertain communities to improve sample efficiency. Empirical results show improved performance over fixed-neighborhood MARL and support the theoretical convergence claims, highlighting the framework's scalability, transferability, and structured exploration benefits in coordinated multi-agent systems.
Abstract
We propose a new framework for multi-agent reinforcement learning (MARL), where the agents cooperate in a time-evolving network with latent community structures and mixed memberships. Unlike traditional neighbor-based or fixed interaction graphs, our community-based framework captures flexible and abstract coordination patterns by allowing each agent to belong to multiple overlapping communities. Each community maintains shared policy and value functions, which are aggregated by individual agents according to personalized membership weights. We also design actor-critic algorithms that exploit this structure: agents inherit community-level estimates for policy updates and value learning, enabling structured information sharing without requiring access to other agents' policies. Importantly, our approach supports both transfer learning by adapting to new agents or tasks via membership estimation, and active learning by prioritizing uncertain communities during exploration. Theoretically, we establish convergence guarantees under linear function approximation for both actor and critic updates. To our knowledge, this is the first MARL framework that integrates community structure, transferability, and active learning with provable guarantees.
