HGAP: Boosting Permutation Invariant and Permutation Equivariant in Multi-Agent Reinforcement Learning via Graph Attention Network
Bor-Jiun Lin, Chun-Yi Lee
Abstract
Graph representation has gained widespread application across various machine learning domains, attributed to its ability to discern correlations among input nodes. In the realm of Multiagent Reinforcement Learning (MARL), agents are tasked with observing other entities within their environment to determine their behavior. Conventional MARL methodologies often suffer from training difficulties if Permutation Invariant (PI) and Permutation Equivariant (PE) properties are not considered during training. The adoption of graph representation offers a solution to these challenges by conceptualizing observed entities as a graph. In this context, we introduce the Hyper Graphical Attention Policy (HGAP) Network, which employs a graph attention mechanism to fulfill the PI and PE properties, while also understanding inter-entity interactions for decisionmaking. HGAP is assessed across various MARL benchmarks to confirm its effectiveness and efficiency. In addition, a series of ablation studies are provided to demonstrate its adaptability, transferability, and the capability to alleviate the complexities introduced by the POMDP constraint.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers2
- Heterogeneous Agent Q-weighted Policy OptimizationBor-Jiun Lin, Chun-Yi LeeICLR 2026 · 102 citations
- Bayesian Ego-graph Inference for Networked Multi-Agent Reinforcement LearningWei Duan, Jie Lu, Junyu XuanNeurIPS 2025 · 15 citations
Builds on13
- An Image is Worth 16x16 Words: Transformers for Image Recognition at ScaleAlexey Dosovitskiy, Lucas Beyer, Alexander Kolesnikov, Dirk Weissenborn et al.ICLR 2021 · 21,477 citations
- Weighted QMIX: Expanding Monotonic Value Function Factorisation for Deep Multi-Agent Reinforcement LearningTabish Rashid, Gregory Farquhar, Bei Peng, Shimon WhitesonNeurIPS 2020 · 1,960 citations
- How Attentive are Graph Attention Networks?Shaked Brody, Uri Alon, Eran YahavICLR 2022 · 1,717 citations
- QPLEX: Duplex Dueling Multi-Agent Q-LearningJianhao Wang, Zhizhou Ren, Terry Liu, Yang Yu et al.ICLR 2021 · 595 citations
- Graph Convolutional Reinforcement LearningJiechuan Jiang, Chen Dun, Tiejun Huang, Zongqing LuICLR 2020 · 415 citations
Related papers
- Boosting Multiagent Reinforcement Learning via Permutation Invariant and Permutation Equivariant NetworksJianye Hao, Xiaotian Hao, Hangyu Mao, Weixun Wang et al.ICLR 2023
- Multi-Agent Actor-Critic with Hierarchical Graph Attention NetworkHeechang Ryu, Hayong Shin, Jinkyoo ParkAAAI 2020 · 143 citations
- Pairwise is Not Enough: Hypergraph Neural Networks for Multi-Agent PathfindingRishabh Jain, Keisuke Okumura, Michael Amir, Pietro Lio et al.ICLR 2026 · 6 citations
- Concentration Network for Reinforcement Learning of Large-Scale Multi-Agent SystemsQingxu Fu, Tenghai Qiu, Jianqiang Yi, Zhiqiang Pu et al.AAAI 2022 · 22 citations
- Universally Expressive Communication in Multi-Agent Reinforcement LearningMatthew Morris, Thomas D. Barrett, Arnu PretoriusNeurIPS 2022 · 7 citations
