Multiagent Reinforcement Learning With Heterogeneous Graph Attention Network

被引:18
作者
Du, Wei [1 ,2 ]
Ding, Shifei [1 ,2 ]
Zhang, Chenglong [1 ,2 ]
Shi, Zhongzhi [3 ]
机构
[1] China Univ Min & Technol, Sch Comp Sci & Technol, Xuzhou 221116, Jiangsu, Peoples R China
[2] China Univ Min & Technol, Minist Educ, Mine Digitizat Engn Res Ctr, Xuzhou 221116, Jiangsu, Peoples R China
[3] Chinese Acad Sci, Inst Comp Technol, Key Lab Intelligent Informat Proc, Beijing 100190, Peoples R China
基金
中国国家自然科学基金;
关键词
Reinforcement learning; Multi-agent systems; Aggregates; Task analysis; Scalability; Marine vehicles; Learning systems; Graph attention network; heterogeneous agents; multiagent reinforcement learning (MARL); relationship-level attention;
D O I
10.1109/TNNLS.2022.3215774
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Most recent research on multiagent reinforcement learning (MARL) has explored how to deploy cooperative policies for homogeneous agents. However, realistic multiagent environments may contain heterogeneous agents that have different attributes or tasks. The heterogeneity of the agents and the diversity of relationships cause the learning of policy excessively tough. To tackle this difficulty, we present a novel method that employs a heterogeneous graph attention network to model the relationships between heterogeneous agents. The proposed method can generate an integrated feature representation for each agent by hierarchically aggregating latent feature information of neighbor agents, with the importance of the agent level and the relationship level being entirely considered. The method is agnostic to specific MARL methods and can be flexibly integrated with diverse value decomposition methods. We conduct experiments in predator-prey and StarCraft Multiagent Challenge (SMAC) environments, and the empirical results demonstrate that the performance of our method is superior to existing methods in several heterogeneous scenarios.
引用
收藏
页码:6851 / 6860
页数:10
相关论文
共 33 条
[1]  
Agarwal Alekh, 2019, ARXIV
[2]  
Böhmer W, 2020, PR MACH LEARN RES, V119
[3]  
Das A, 2019, PR MACH LEARN RES, V97
[4]   A survey on multi-agent deep reinforcement learning: from the perspective of challenges and applications [J].
Du, Wei ;
Ding, Shifei .
ARTIFICIAL INTELLIGENCE REVIEW, 2021, 54 (05) :3215-3238
[5]  
Foerster JN, 2018, AAAI CONF ARTIF INTE, P2974
[6]  
Gilmer J, 2017, PR MACH LEARN RES, V70
[7]  
Iqbal S, 2019, PR MACH LEARN RES, V97
[8]  
Jiang J., 2018, ARXIV
[9]  
Jiang JC, 2018, ADV NEUR IN, V31
[10]  
Liu Y, 2020, AAAI CONF ARTIF INTE, V34, P7211