Neighbor2Seq: Deep Learning on Massive Graphs by Transforming Neighbors to Sequences

Meng Liu, Shuiwang Ji

Main figure for Neighbor2Seq: Deep Learning on Massive Graphs by Transforming Neighbors to Sequences

Abstract

=9pt Modern graph neural networks (GNNs) use a message passing scheme and have achieved great success in many fields. However, this recursive design inherently leads to excessive computation and memory requirements, making it not applicable to massive real-world graphs. In this work, we propose the Neighbor2Seq to transform the hierarchical neighborhood of each node into a sequence. This novel transformation enables the subsequent mini-batch training for general deep learning operations, such as convolution and attention, that are designed for grid-like data and are shown to be powerful in various domains. Therefore, our Neighbor2Seq naturally endows GNNs with the efficiency and advantages of deep learning operations on grid-like data by precomputing the Neighbor2Seq transformations. We evaluate our method on a massive graph, with more than $111$ million nodes and $1.6$ billion edges, as well as several medium-scale graphs. Results show that our proposed method is scalable to massive graphs and achieves superior performance across massive and medium-scale graphs. Our code is available at.

Benchmark Results

Tested on 3 datasets 9 baseline values reported too low

Tables show presented results in global context (best 3 papers + GCN + MLP).

Classic

#ModelF1 ↑Paper
1
EnGCN
0.9714 ±0.0003
2
SAGN+SLE
0.9710
3
GAMLP(JK)
0.9704 ±0.0001
4
GraphSAINT
actual #4 vs paper #13 9 places worse
0.9700
best reported
actual #4 vs paper #13 9 places worse
paper 0.9660 ±0.0010
8
SIGN
0.9680
This paper
10
Neighbor2Seq+Conv
0.9670
This paper
11
Neighbor2Seq+Attn
0.9670
This paper
14
GraphSAGE
actual #14 vs paper #57 43 places worse
0.9660
best reported
actual #14 vs paper #57 43 places worse
paper 0.9530 ±0.0010
paper 0.9530 ±0.0010 Δ at pub 0.0130
25
VR-GCN
actual #25 vs paper #25
0.9640 ±0.0010
actual #25 vs paper #25
paper 0.9640 ±0.0010

Improvement within reported uncertainty; not highlighted.

paper 0.9640 ±0.0010 Δ at pub 0.0010
23
AS-GCN
actual #23 vs paper #43 20 places worse
0.9640 ±0.0010
best reported
actual #23 vs paper #43 20 places worse
paper 0.9580 ±0.0010
48
ClusterGCN
actual #48 vs paper #52 4 places worse
0.9570
best reported
actual #48 vs paper #52 4 places worse
paper 0.9540 ±0.0010
58
GCN
actual #58 vs paper #101 43 places worse
0.9530 ±0.0010
best reported
actual #58 vs paper #101 43 places worse
paper 0.9330
paper 0.9330 Δ today 0.0200
73
SGC
0.9490
This paper
91
FastGCN
actual #91 vs paper #111 20 places worse
0.9370
best reported
actual #91 vs paper #111 20 places worse
paper 0.9240 ±0.0010

OGB

#ModelAccuracy ↑Paper
1
Jacobi
0.8968
2
GPR
0.8953
3
LEMP4HG
0.8939
4
SAGE
0.8933
6
GCN
actual #6 vs paper #250 244 places worse
0.8868
best reported
actual #6 vs paper #250 244 places worse
paper 0.7564 ±0.0021
64
GraphSAINT
actual #64 vs paper #184 120 places worse
0.8383 ±0.0014
best reported
actual #64 vs paper #184 120 places worse
paper 0.7908 ±0.0024
paper 0.7908 ±0.0024 Δ at pub 0.0475
77
SIGN
actual #77 vs paper #225 148 places worse
0.8310 ±0.0080
best reported
actual #77 vs paper #225 148 places worse
paper 0.7760 ±0.0013
83
GraphSAGE
actual #83 vs paper #199 116 places worse
0.8287
best reported
actual #83 vs paper #199 116 places worse
paper 0.7870 ±0.0036
94
SGC
actual #94 vs paper #314 220 places worse
0.8220 ±0.0090
best reported
actual #94 vs paper #314 220 places worse
paper 0.7246 ±0.0027
paper 0.7246 ±0.0027 Δ at pub 0.0344 Δ today 0.0974
156
Neighbor2Seq+Conv
0.7967 ±0.0016
This paper
169
Neighbor2Seq+Attn
0.7935 ±0.0017
This paper
190
ClusterGCN
0.7897 ±0.0033
This paper
313
Node2Vec
0.7249 ±0.0010
This paper
351
MLP
actual #351 vs paper #370 19 places worse
0.6605 ±0.0020
best reported
actual #351 vs paper #370 19 places worse
paper 0.6106 ±0.0008
paper 0.6106 ±0.0008 Δ at pub 0.2312 Δ today 0.0499
#ModelAccuracy ↑Paper
1
SGC
actual #1 vs paper #76 75 places worse
0.7910
best reported
actual #1 vs paper #76 75 places worse
paper 0.6329 ±0.0019
paper 0.6329 ±0.0019 Δ at pub 0.0281 Δ today 0.1581
2
GAMLP
0.7036 ±0.0002
3
GLEM-GNN
0.7036 ±0.0002
35
SAGE
0.6644 ±0.0017
54
Neighbor2Seq+Conv
0.6562 ±0.0032
This paper
62
Neighbor2Seq+Attn
0.6504 ±0.0024
This paper
88
Node2Vec
actual #88 vs paper #88
0.5810
best reported
actual #88 vs paper #88
paper 0.5560 ±0.0023
95
MLP
actual #95 vs paper #96 1 place worse
0.4960 ±0.0030
best reported
actual #95 vs paper #96 1 place worse
paper 0.4724 ±0.0031
SOTA performance (not significantly different from the best) Proposed method (this paper) Baseline / comparison (this paper) Global context (other papers) Best at pub — reported result worse than the best known when published (best reported in green) Best today — OK at publication; today's leaderboard best is elsewhere