Efficient Scaling of Dynamic Graph Neural Networks
Not on the map. Why?
Abstract (arXiv)
We present distributed algorithms for training dynamic Graph Neural Networks (GNN) on large scale graphs spanning multi-node, multi-GPU systems. To the best of our knowledge, this is the first scaling study on dynamic GNN. We devise mechanisms for reducing the GPU memory usage and identify two execution time bottlenecks: CPU-GPU data transfer; and communication volume. Exploiting properties of dynamic graphs, we design a graph difference-based strategy to significantly reduce the transfer time. We develop a simple, but effective data distribution technique under which the communication volume remains fixed and linear in the input size, for any number of GPUs. Our experiments using billion-size graphs on a system of 128 GPUs shows that: (i) the distribution scheme achieves up to 30x speedup on 128 GPUs; (ii) the graph-difference technique reduces the transfer time by a factor of up to 4.1x and the overall execution time by up to 40%
Key facts. Cited by 11 papers in the map (data through 7 Oct 2026). Builds on 19 of its 31 references that are in the map. First posted on arXiv 16 Sep 2021.
Builds on
19 of 31 references.
References · 19
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,282
- 2019A Comprehensive Survey on Graph Neural Networkscited by 2,906
- 2019Fast Graph Representation Learning with PyTorch Geometriccited by 2,414
- 2018Graph Convolutional Neural Networks for Web-Scale Recommender Systemscited by 1,133
- 2016Training Deep Nets with Sublinear Memory Costcited by 825
- 2020Temporal Graph Networks for Deep Learning on Dynamic Graphscited by 521
- 2019EvolveGCN: Evolving Graph Convolutional Networks for Dynamic Graphscited by 443
- 2016Structured Sequence Modeling with Graph Convolutional Recurrent Networkscited by 273
- 2019Representation Learning for Dynamic Graphs: A Surveycited by 260
- 2019Deep Graph Library: A Graph-Centric, Highly-Performant Package for Graph Neural Networkscited by 241
- 2019AliGraph: A Comprehensive Graph Neural Network Platformcited by 152
- 2017Know-Evolve: Deep Temporal Reasoning for Dynamic Knowledge Graphscited by 149
7 more
- 2018Streaming Graph Neural Networkscited by 91
- 2017Dynamic Graph Convolutional Networkscited by 83
- 2020Computing Graph Neural Networks: A Survey from Algorithms to Acceleratorscited by 81
- 2016Memory-Efficient Backpropagation Through Timecited by 78
- 2020Reducing Communication in Graph Neural Network Trainingcited by 50
- 2018Scalable Graph Learning for Anti-Money Laundering: A First Lookcited by 32
- 2018Learning Dynamic Embeddings from Temporal Interactionscited by 10
Built on by
Citing papers per year: 2022: 2; 2023: 5; 2024: 3; 2025: 0; 2026: 1.
11 papers.
- 2023Realistic Synthetic Financial Transactions for Anti-Money Laundering Modelscited by 47
- 2022Parallel and Distributed Graph Neural Networks: An In-Depth Concurrency Analysiscited by 29
- 2024A Comprehensive Survey of Dynamic Graph Neural Networks: Models, Frameworks, Benchmarks, Experiments and Challengescited by 29
- 2024A survey of dynamic graph neural networkscited by 21
- 2022A Comprehensive Survey on Distributed Training of Graph Neural Networkscited by 15
- 2023DistTGL: Distributed Memory-Based Temporal Graph Neural Network Trainingcited by 12
- 2023PiPAD: Pipelined and Parallel Dynamic GNN Training on GPUscited by 4
- 2023DGNN-Booster: A Generic FPGA Accelerator Framework For Dynamic Graph Neural Network Inferencecited by 3
- 2024Input Snapshots Fusion for Scalable Discrete-Time Dynamic Graph Neural Networkscited by 3
- 2023DGC: Training Dynamic Graphs with Spatio-Temporal Non-Uniformity using Graph Partitioning by Chunkscited by 1
- 2026DG-CoLearn: An Efficient Collaborative Learning Framework for Dynamic Graphscited by 0
- 2026DG-CoLearn: An Efficient Collaborative Learning Framework for Dynamic Graphs29 May 2026
- 2024Input Snapshots Fusion for Scalable Discrete-Time Dynamic Graph Neural Networks11 May 2024
- 2024A Comprehensive Survey of Dynamic Graph Neural Networks: Models, Frameworks, Benchmarks, Experiments and Challenges1 May 2024
- 2024A survey of dynamic graph neural networks28 Apr 2024
- 2023DGC: Training Dynamic Graphs with Spatio-Temporal Non-Uniformity using Graph Partitioning by Chunks7 Sep 2023
- 2023DistTGL: Distributed Memory-Based Temporal Graph Neural Network Training14 Jul 2023
- 2023Realistic Synthetic Financial Transactions for Anti-Money Laundering Models22 Jun 2023
- 2023DGNN-Booster: A Generic FPGA Accelerator Framework For Dynamic Graph Neural Network Inference13 Apr 2023
- 2023PiPAD: Pipelined and Parallel Dynamic GNN Training on GPUs1 Jan 2023
- 2022A Comprehensive Survey on Distributed Training of Graph Neural Networks10 Nov 2022
- 2022Parallel and Distributed Graph Neural Networks: An In-Depth Concurrency Analysis19 May 2022