Non-Local Graph Neural Networks

被引:90
作者
Liu, Meng [1 ]
Wang, Zhengyang [1 ,2 ]
Ji, Shuiwang [1 ]
机构
[1] Texas A&M Univ, Dept Comp Sci & Engn, College Stn, TX 77843 USA
[2] Amazon Com Serv LLC, Seattle, WA 98109 USA
基金
美国国家科学基金会;
关键词
Sorting; Task analysis; Graph neural networks; Convolution; Aggregates; Nonhomogeneous media; Calibration; non-local aggregation; attention mechanism; disassortative graphs;
D O I
10.1109/TPAMI.2021.3134200
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Modern graph neural networks (GNNs) learn node embeddings through multilayer local aggregation and achieve great success in applications on assortative graphs. However, tasks on disassortative graphs usually require non-local aggregation. In addition, we find that local aggregation is even harmful for some disassortative graphs. In this work, we propose a simple yet effective non-local aggregation framework with an efficient attention-guided sorting for GNNs. Based on it, we develop various non-local GNNs. We perform thorough experiments to analyze disassortative graph datasets and evaluate our non-local GNNs. Experimental results demonstrate that our non-local GNNs significantly outperform previous state-of-the-art methods on seven benchmark datasets of disassortative graphs, in terms of both model performance and efficiency.
引用
收藏
页码:10270 / 10276
页数:7
相关论文
共 53 条
[1]  
Alon Uri, 2020, INT C LEARN REPR
[2]  
Bianchi FM, 2020, PR MACH LEARN RES, V119
[3]  
Bo DY, 2021, AAAI CONF ARTIF INTE, V35, P3950
[4]   Geometric Deep Learning Going beyond Euclidean data [J].
Bronstein, Michael M. ;
Bruna, Joan ;
LeCun, Yann ;
Szlam, Arthur ;
Vandergheynst, Pierre .
IEEE SIGNAL PROCESSING MAGAZINE, 2017, 34 (04) :18-42
[5]  
Chen DL, 2020, AAAI CONF ARTIF INTE, V34, P3438
[6]   Xception: Deep Learning with Depthwise Separable Convolutions [J].
Chollet, Francois .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :1800-1807
[7]  
Defferrard M, 2016, ADV NEUR IN, V29
[8]  
Errica F., 2019, ARXIV191209893
[9]  
Fey M., 2019, P WORKSH REPR LEARN, P1
[10]  
Gao HY, 2019, PR MACH LEARN RES, V97