Heterogeneous Multi-Agent Communication Learning Via Graph Information Maximization.

Wei Du,Shifei Ding
DOI: https://doi.org/10.18293/seke2023-099
2023-01-01
Abstract:Communication learning is an effective way to solve complicated cooperative tasks in multi-agent reinforcement learning (MARL) domain.Graph neural network (GNN) has been widely adopt for learning the multi-agent communication and various GNN-based MARL methods have emerged.However, most of these methods are not specially designed for heterogeneous multi-agent scenarios, where agents have heterogeneous attributes or features based on different observation spaces or action sets.Without effective processing and transmission of heterogeneous feature information, communication learning will be useless and even reduce the performance of cooperation.To solve this problem, we propose a communication learning mechanism based on heterogeneous GNN and graph information maximization to learn effective communication for heterogeneous agents.Specifically, we use heterogeneous GNN for learning the efficient message representations, which aggregate the local feature information of neighboring agents.Furthermore, we maximize the mutual information (MI) between message representations and local values to make efficient use of information.Besides, we present a MARL framework that can flexibly integrate the proposed communication mechanism with existing value factorization methods.Experiments on various heterogeneous multi-agent scenarios demonstrate the effectiveness and superiority of the proposed method compared with baselines.
What problem does this paper attempt to address?