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 条
[11]   Graph Representation Learning via Hard and Channel-Wise Attention Networks [J].
Gao, Hongyang ;
Ji, Shuiwang .
KDD'19: PROCEEDINGS OF THE 25TH ACM SIGKDD INTERNATIONAL CONFERENCCE ON KNOWLEDGE DISCOVERY AND DATA MINING, 2019, :741-749
[12]   Large-Scale Learnable Graph Convolutional Networks [J].
Gao, Hongyang ;
Wang, Zhengyang ;
Ji, Shuiwang .
KDD'18: PROCEEDINGS OF THE 24TH ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY & DATA MINING, 2018, :1416-1424
[13]  
Gilmer J, 2017, PR MACH LEARN RES, V70
[14]  
Hamilton WL, 2017, ADV NEUR IN, V30
[15]  
Hu Weihua, 2020, Advances in Neural Information Processing Systems, V33
[16]   Node Similarity Preserving Graph Convolutional Networks [J].
Jin, Wei ;
Derr, Tyler ;
Wang, Yiqi ;
Ma, Yao ;
Liu, Zitao ;
Tang, Jiliang .
WSDM '21: PROCEEDINGS OF THE 14TH ACM INTERNATIONAL CONFERENCE ON WEB SEARCH AND DATA MINING, 2021, :148-156
[17]  
Knyazev B, 2019, ADV NEUR IN, V32
[18]   ImageNet Classification with Deep Convolutional Neural Networks [J].
Krizhevsky, Alex ;
Sutskever, Ilya ;
Hinton, Geoffrey E. .
COMMUNICATIONS OF THE ACM, 2017, 60 (06) :84-90
[19]   Gradient-based learning applied to document recognition [J].
Lecun, Y ;
Bottou, L ;
Bengio, Y ;
Haffner, P .
PROCEEDINGS OF THE IEEE, 1998, 86 (11) :2278-2324
[20]  
Lee J, 2019, PR MACH LEARN RES, V97