Multiagent Reinforcement Learning With Heterogeneous Graph Attention Network

被引:17
|
作者
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
相关论文
共 50 条
  • [1] Cascaded Attention: Adaptive and Gated Graph Attention Network for Multiagent Reinforcement Learning
    Qi, Shuhan
    Huang, Xinhao
    Peng, Peixi
    Huang, Xuzhong
    Zhang, Jiajia
    Wang, Xuan
    IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2024, 35 (03) : 3769 - 3779
  • [2] VGN: Value Decomposition With Graph Attention Networks for Multiagent Reinforcement Learning
    Wei, Qinglai
    Li, Yugu
    Zhang, Jie
    Wang, Fei-Yue
    IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2024, 35 (01) : 182 - 195
  • [3] Attention-Based Multiagent Graph Reinforcement Learning for Service Restoration
    Fan B.
    Liu X.
    Xiao G.
    Kang Y.
    Wang D.
    Wang P.
    IEEE Transactions on Artificial Intelligence, 2024, 5 (05): : 2163 - 2178
  • [4] Heterogeneous-graph Attention Reinforcement Learning for Football Matches
    Wang, Shijie
    Pan, Yi
    Pu, Zhiqiang
    Yi, Jianqiang
    Liang, Yanyan
    Zhang, Du
    2023 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS, IJCNN, 2023,
  • [5] Heterogeneous Graph Attention Network
    Wang, Xiao
    Ji, Houye
    Shi, Chuan
    Wang, Bai
    Cui, Peng
    Yu, P.
    Ye, Yanfang
    WEB CONFERENCE 2019: PROCEEDINGS OF THE WORLD WIDE WEB CONFERENCE (WWW 2019), 2019, : 2022 - 2032
  • [6] Heterogeneous Information Network Representation Learning Framework Based on Graph Attention Network
    Kang Shize
    Ji Lixin
    Zhang Jianpeng
    JOURNAL OF ELECTRONICS & INFORMATION TECHNOLOGY, 2021, 43 (04) : 915 - 922
  • [7] Multimodal heterogeneous graph attention network
    Jia, Xiangen
    Jiang, Min
    Dong, Yihong
    Zhu, Feng
    Lin, Haocai
    Xin, Yu
    Chen, Huahui
    NEURAL COMPUTING & APPLICATIONS, 2023, 35 (04): : 3357 - 3372
  • [8] Multimodal heterogeneous graph attention network
    Xiangen Jia
    Min Jiang
    Yihong Dong
    Feng Zhu
    Haocai Lin
    Yu Xin
    Huahui Chen
    Neural Computing and Applications, 2023, 35 : 3357 - 3372
  • [9] Heterogeneous Graph Gated Attention Network
    Ma, Shuai
    Liu, Jian-wei
    Zuo, Xin
    Li, Wei-min
    2021 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2021,
  • [10] Heterogeneous Dynamic Graph Attention Network
    Li, Qiuyan
    Shang, Yanlei
    Qiao, Xiuquan
    Dai, Wei
    11TH IEEE INTERNATIONAL CONFERENCE ON KNOWLEDGE GRAPH (ICKG 2020), 2020, : 404 - 411