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
| # | Model | F1 ↑ | Paper |
|---|---|---|---|
| 1 | EnGCN | 0.9714
±0.0003 | |
| 2 | SAGN+SLE | 0.9710 | |
| 3 | GAMLP(JK) | 0.9704
±0.0001 | |
| 4 | GraphSAINT | 0.9700 best reported | best at pub: SIGN: Scalable Inception Graph Neural Networks (2020) best today: LMC: Fast Training of GNNs via subgraph-wise sampling with Provable Convergence (ICLR 2023) |
| 8 | SIGN | 0.9680 | This paper |
| 10 | Neighbor2Seq+Conv | 0.9670 | This paper |
| 11 | Neighbor2Seq+Attn | 0.9670 | This paper |
| 14 | GraphSAGE | 0.9660 best reported | best at pub: Accelerating Large Scale Real-Time GNN Inference using Channel Pruning (2021) |
| 25 | VR-GCN | 0.9640
±0.0010 | prev best: Cluster-GCN: An Efficient Algorithm for Training Deep and Large Graph Convolutional Networks (KDD 2019) Improvement within reported uncertainty; not highlighted. |
| 23 | AS-GCN | 0.9640
±0.0010 best reported | best at pub: GraphSAINT: Graph Sampling Based Inductive Learning Method (ICLR 2019) |
| 48 | ClusterGCN | 0.9570 best reported | best at pub: GraphSAINT: Graph Sampling Based Inductive Learning Method (ICLR 2019) best today: PaSca: a Graph Neural Architecture Search System under the Scalable Paradigm (WWW 2022) |
| 58 | GCN | 0.9530
±0.0010 best reported | best at pub: GraphSAINT: Graph Sampling Based Inductive Learning Method (ICLR 2019) best today: GraphMAE: Self-Supervised Masked Graph Autoencoders (KDD 2022) |
| 73 | SGC | 0.9490 | This paper |
| 91 | FastGCN | 0.9370 best reported | best at pub: FastGCN: Fast Learning with Graph Convolutional Networks via Importance Sampling (ICLR 2018) |
OGB
| # | Model | Accuracy ↑ | Paper |
|---|---|---|---|
| 1 | Jacobi | 0.8968 | |
| 2 | GPR | 0.8953 | |
| 3 | LEMP4HG | 0.8939 | |
| 4 | SAGE | 0.8933 | |
| 6 | GCN | 0.8868 best reported | best at pub: Calibrating and Improving Graph Contrastive Learning (TMLR 2021) |
| 64 | GraphSAINT | 0.8383
±0.0014 best reported | best at pub: Node Feature Extraction by Self-Supervised Multi-scale Neighborhood Prediction (ICLR 2021) |
| 77 | SIGN | 0.8310
±0.0080 best reported | best at pub: Scalable and Adaptive Graph Neural Networks with Self-Label-Enhanced Training (2021) best today: PaSca: a Graph Neural Architecture Search System under the Scalable Paradigm (WWW 2022) |
| 83 | GraphSAGE | 0.8287 best reported | |
| 94 | SGC | 0.8220
±0.0090 best reported | best at pub: Evaluating Deep Graph Neural Networks (2021) best today: PaSca: a Graph Neural Architecture Search System under the Scalable Paradigm (WWW 2022) |
| 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 | 0.6605
±0.0020 best reported | best at pub: SCR: Training Graph Neural Networks with Consistency Regularization (2021) best today: Latent Graph Inference using Product Manifolds (ICLR 2022) |
| # | Model | Accuracy ↑ | Paper |
|---|---|---|---|
| 1 | SGC | 0.7910 best reported | best at pub: Node Feature Extraction by Self-Supervised Multi-scale Neighborhood Prediction (ICLR 2021) best today: Unifews: You Need Fewer Operations for Efficient Graph Neural Networks (ICML 2024) |
| 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 | 0.5810 best reported | best at pub: SIGN: Scalable Inception Graph Neural Networks (2020) |
| 95 | MLP | 0.4960
±0.0030 best reported | best at pub: SIGN: Scalable Inception Graph Neural Networks (2020) |
