Attention-Guided Contrastive Role Representations for Multi-Agent Reinforcement Learning

Zican Hu,Zongzhang Zhang,Huaxiong Li,Chunlin Chen,Hongyu Ding,Zhi Wang
2024-01-01
Abstract:Real-world multi-agent tasks usually involve dynamic team composition withthe emergence of roles, which should also be a key to efficient cooperation inmulti-agent reinforcement learning (MARL). Drawing inspiration from thecorrelation between roles and agent's behavior patterns, we propose a novelframework of **A**ttention-guided **CO**ntrastive **R**ole representationlearning for **M**ARL (**ACORM**) to promote behavior heterogeneity, knowledgetransfer, and skillful coordination across agents. First, we introduce mutualinformation maximization to formalize role representation learning, derive acontrastive learning objective, and concisely approximate the distribution ofnegative pairs. Second, we leverage an attention mechanism to prompt the globalstate to attend to learned role representations in value decomposition,implicitly guiding agent coordination in a skillful role space to yield moreexpressive credit assignment. Experiments on challenging StarCraft IImicromanagement and Google research football tasks demonstrate thestate-of-the-art performance of our method and its advantages over existingapproaches. Our code is available at[https://github.com/NJU-RL/ACORM](https://github.com/NJU-RL/ACORM).
What problem does this paper attempt to address?