A Comprehensive Study on Large-Scale Graph Training: Benchmarking and Rethinking
Abstract
Large-scale graph training is a notoriously challenging problem for graph neural networks (GNNs). Due to the nature of evolving graph structures into the training process, vanilla GNNs usually fail to scale up, limited by the GPU memory space. Up to now, though numerous scalable GNN architectures have been proposed, we still lack a comprehensive survey and fair benchmark of this reservoir to find the rationale for designing scalable GNNs. To this end, we first systematically formulate the representative methods of large-scale graph training into several branches and further establish a fair and consistent benchmark for them by a greedy hyperparameter searching. In addition, regarding efficiency, we theoretically evaluate the time and space complexity of various branches and empirically compare them w.r.t GPU memory usage, throughput, and convergence. Furthermore, We analyze the pros and cons for various branches of scalable GNNs and then present a new ensembling training manner, named EnGCN, to address the existing issues. Remarkably, our proposed method has achieved new state-of-the-art (SOTA) performance on large-scale datasets. Our code is available at.
Benchmark Results
Tested on 2 datasets 11 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 | This paper |
| 2 | SAGN+SLE | 0.9710 | This paper |
| 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) |
| 6 | SAGN | 0.9690 best reported | best at pub: Scalable and Adaptive Graph Neural Networks with Self-Label-Enhanced Training (2021) |
| 8 | SIGN | 0.9680 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) |
| 9 | GAMLP | 0.9673
±0.0003 | This paper |
| 14 | GraphSAGE | 0.9660 best reported | best at pub: Accelerating Large Scale Real-Time GNN Inference using Channel Pruning (2021) |
| 37 | AdaGCN | 0.9605 | This paper |
| 48 | ClusterGCN | 0.9570 best reported | best at pub: PaSca: a Graph Neural Architecture Search System under the Scalable Paradigm (WWW 2022) Difference within reported uncertainty; not counted as underperformance. |
| 57 | C&S | 0.9533
±0.0008 | This paper |
| 58 | GCN | 0.9530
±0.0010 | |
| 73 | SGC | 0.9490 best reported | best at pub: Deep Learning on Graphs: A Survey (2018) |
| 91 | FastGCN | 0.9370 best reported | best at pub: FastGCN: Fast Learning with Graph Convolutional Networks via Importance Sampling (ICLR 2018) |
| 104 | LADIES | 0.9320
±0.0010 best reported |
OGB
| # | Model | Accuracy ↑ | Paper |
|---|---|---|---|
| 1 | Jacobi | 0.8968 | |
| 2 | GPR | 0.8953 | |
| 3 | LEMP4HG | 0.8939 | |
| 4 | SAGE | 0.8933 | |
| 11 | EnGCN | 0.8799
±0.0004 | This paper |
| 14 | SAGN | 0.8736
±0.0007 best reported | best at pub: Learning on Large-scale Text-attributed Graphs via Variational Inference (ICLR 2022) |
| 22 | GIANT-XRT+SAGN+MCR+C&S | 0.8673
±0.0008 | This paper |
| 41 | C&S | 0.8511
±0.0007 | prev best: Alternately Optimized Graph Neural Networks (ICML 2022) |
| 64 | GraphSAINT | 0.8383
±0.0014 best reported | best at pub: Node Feature Extraction by Self-Supervised Multi-scale Neighborhood Prediction (ICLR 2021) |
| 65 | GAMLP | 0.8376
±0.0019 | prev best: Graph Attention MLP with Reliable Label Utilization (2021) Improvement within reported uncertainty; not highlighted. |
| 77 | SIGN | 0.8310
±0.0080 best reported | best at pub: PaSca: a Graph Neural Architecture Search System under the Scalable Paradigm (WWW 2022) |
| 83 | GraphSAGE | 0.8061
±0.0016 | |
| 94 | SGC | 0.8220
±0.0090 best reported | best at pub: PaSca: a Graph Neural Architecture Search System under the Scalable Paradigm (WWW 2022) |
| 190 | ClusterGCN | 0.7897
±0.0033 best reported | best at pub: SIGN: Scalable Inception Graph Neural Networks (2020) best today: From Node Interaction to Hop Interaction: New Effective and Scalable Graph Learning Paradigm (2022) Difference within reported uncertainty; not counted as underperformance. |
| 238 | AdaGCN | 0.7641 | This paper |
| 262 | LADIES | 0.7531
±0.0056 | This paper |
| 308 | FastGCN | 0.7346
±0.0020 | This paper |
| 351 | MLP | 0.6605
±0.0020 |
