GCS: Graph-Based Coordination Strategy for Multi-Agent Reinforcement Learning

Jingqing Ruan (Institute of Automation, Chinese Academy of Sciences & University of Chinese Academy of Sciences), Yali Du (King's College London), Xuantang Xiong (Institute of Automation, Chinese Academy of Sciences), Dengpeng Xing (Institute of Automation, Chinese Academy of Sciences), Xiyun Li (Institute of Automation, Chinese Academy of Sciences), Linghui Meng (Institute of Automation, Chinese Academy of Sciences), Haifeng Zhang (Institute of Automation, Chinese Academy of Sciences), Jun Wang (University College London), Bo Xu (Institute of Automation, Chinese Academy of Sciences)

Abstract

Many real-world scenarios involve a team of agents that have to coordinate their policies to achieve a shared goal. Previous studies mainly focus on decentralized control to maximize a common reward and barely consider the coordination among control policies, which is critical in dynamic and complicated environments. In this work, we propose factorizing the joint team policy into a graph generator and graph-based coordinated policy to enable coordinated behaviours among agents. The graph generator adopts an encoder-decoder framework that outputs directed acyclic graphs (DAGs) to capture the underlying dynamic decision structure. We also apply the DAGness-constrained and DAG depth-constrained optimization in the graph generator to balance efficiency and performance. The graph-based coordinated policy exploits the generated decision structure. The graph generator and coordinated policy are trained simultaneously to maximize the discounted return. Empirical evaluations on Collaborative Gaussian Squeeze, Cooperative Navigation, and Google Research Football demonstrate the superiority of the proposed method. The code is available at https://github.com/Amanda-1997/GCS_aamas337.