Multi-head multi-order graph attention networks

被引:5
作者
Ben, Jie [1 ]
Sun, Qiguo [1 ]
Liu, Keyu [2 ]
Yang, Xibei [1 ]
Zhang, Fengjun [1 ]
机构
[1] Jiangsu Univ Sci & Technol, Sch Comp, Zhenjiang 212100, Jiangsu, Peoples R China
[2] Southwest Jiaotong Univ, Sch Comp & Artificial Intelligence, Chengdu 611756, Sichuan, Peoples R China
基金
中国国家自然科学基金;
关键词
Semi-supervised learning; Graph convolutional network; Attention mechanism; Multi-order information; CONVOLUTIONAL NETWORKS;
D O I
10.1007/s10489-024-05601-z
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The Graph Attention Network (GAT) is a type of graph neural network (GNN) that uses attention mechanisms to weigh the importance of nodes' neighbors, demonstrating flexibility and power in representation learning. However, GAT and its variants still face common challenges in GNNs, such as over-smoothing and over-squashing. To address this, we propose Multi-Head Multi-Order Graph Attention Networks (MHMOGAT) as an enhanced GAT layer. MHMOGAT is built based on multi-head attention and adjacency matrices of different orders, aiming to expand the receptive field of GAT to effectively capture long-distance dependencies. Moreover, Bayesian optimization is employed to determine optimal hyperparameter combinations for different datasets. Experimental results on six prevailing datasets demonstrate that MHMOGAT improves GAT accuracy by approximately 2-5% across various datasets with different label rates, indicating its effectiveness. Additionally, MHMOGAT exhibits potential in handling large and complex graphs with low label rates.
引用
收藏
页码:8092 / 8107
页数:16
相关论文
共 53 条
[1]  
Abu-El-Haifa S, 2019, PR MACH LEARN RES, V97
[2]  
Alon U., 2021, INT C LEARN REPR
[3]   Semi-supervised active learning for support vector machines: A novel approach that exploits structure information in data [J].
Calma, Adrian ;
Reitmaier, Tobias ;
Sick, Bernhard .
INFORMATION SCIENCES, 2018, 456 :13-33
[4]  
Chai Ziwei, 2023, arXiv
[5]  
Chamberlain B., 2021, P INT C MACH LEARN, P1407
[6]   Global Attention-Based Graph Neural Networks for Node Classification [J].
Chen, Jiusheng ;
Fang, Chengyuan ;
Zhang, Xiaoyu .
NEURAL PROCESSING LETTERS, 2023, 55 (04) :4127-4150
[7]  
Chen Ming, 2020, P MACHINE LEARNING R, V119
[8]  
Chien Eli, 2021, INT C LEARN REPR
[9]  
Cozman FabioG., 2002, P 15 INT FLORIDA ART, P327
[10]  
Dai Mingzhi, 2020, Data Science: 6th International Conference of Pioneering Computer Scientists, Engineers and Educators, ICPCSEE 2020. Communications in Computer and Information Science (1258), P197, DOI 10.1007/978-981-15-7984-4_16