GNNFlow: A Distributed Framework for Continuous Temporal GNN Learning on Dynamic Graphs
Not on the map. Why?
Abstract (arXiv)
Graph Neural Networks (GNNs) play a crucial role in various fields. However, most existing deep graph learning frameworks assume pre-stored static graphs and do not support training on graph streams. In contrast, many real-world graphs are dynamic and contain time domain information. We introduce GNNFlow, a distributed framework that enables efficient continuous temporal graph representation learning on dynamic graphs on multi-GPU machines. GNNFlow introduces an adaptive time-indexed block-based data structure that effectively balances memory usage with graph update and sampling operation efficiency. It features a hybrid GPU-CPU graph data placement for rapid GPU-based temporal neighborhood sampling and kernel optimizations for enhanced sampling processes. A dynamic GPU cache for node and edge features is developed to maximize cache hit rates through reuse and restoration strategies. GNNFlow supports distributed training across multiple machines with static scheduling to ensure load balance. We implement GNNFlow based on DGL and PyTorch. Our experimental results show that GNNFlow provides up to 21.1x faster continuous learning than existing systems.
Key facts. Cited by 5 papers in the map (data through 14 Oct 2026). Builds on 21 of its 71 references that are in the map. First posted on arXiv 29 Nov 2023.
Builds on
21 of 71 references.
References · 21
- 2017Attention Is All You Needcited by 60,361
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,361
- 2017Graph Attention Networkscited by 8,923
- 2017Inductive Representation Learning on Large Graphscited by 6,320
- 2017Neural Message Passing for Quantum Chemistrycited by 3,487
- 2019Fast Graph Representation Learning with PyTorch Geometriccited by 2,434
- 2018Graph Convolutional Neural Networks for Web-Scale Recommender Systemscited by 1,138
- 2019Graph Neural Networks for Social Recommendationcited by 714
- 2020Inductive Representation Learning on Temporal Graphscited by 462
- 2019Predicting Dynamic Embedding Trajectory in Temporal Interaction Networkscited by 448
- 2019EvolveGCN: Evolving Graph Convolutional Networks for Dynamic Graphscited by 447
- 2019AliGraph: A Comprehensive Graph Neural Network Platformcited by 153
9 more
- 2017Know-Evolve: Deep Temporal Reasoning for Dynamic Knowledge Graphscited by 149
- 2018Streaming Graph Neural Networkscited by 91
- 2020Streaming Graph Neural Networks via Continual Learningcited by 64
- 2021BGL: GPU-Efficient GNN Training by Optimizing Graph Data I/O and Preprocessingcited by 50
- 2020GraphSAIL: Graph Structure Aware Incremental Learning for Recommender Systemscited by 42
- 2020Accelerating Graph Sampling for Graph Machine Learning using GPUscited by 29
- 2020C-SAW: A Framework for Graph Sampling and Random Walk on GPUscited by 13
- 2019Low-Latency Graph Streaming Using Compressed Purely-Functional Treescited by 6
- 2021An O(1) algorithm for implementing the LFU cache eviction schemecited by 4
Built on by
Citing papers per year: 2024: 3; 2025: 1; 2026: 1.
5 papers.
- 2024A Comprehensive Survey of Dynamic Graph Neural Networks: Models, Frameworks, Benchmarks, Experiments and Challengescited by 30
- 2024MSPipe: Efficient Temporal GNN Training via Staleness-Aware Pipelinecited by 12
- 2024Towards Ideal Temporal Graph Neural Networks: Evaluations and Conclusions after 10,000 GPU Hourscited by 2
- 2025GNN-Suite: a Graph Neural Network Benchmarking Framework for Biomedical Informaticscited by 2
- 2026Directly Optimizing Mean Demographic Parity for Nonlinear Regressioncited by 0
- 2026Directly Optimizing Mean Demographic Parity for Nonlinear Regression6 Jul 2026
- 2025GNN-Suite: a Graph Neural Network Benchmarking Framework for Biomedical Informatics15 May 2025
- 2024Towards Ideal Temporal Graph Neural Networks: Evaluations and Conclusions after 10,000 GPU Hours28 Dec 2024
- 2024A Comprehensive Survey of Dynamic Graph Neural Networks: Models, Frameworks, Benchmarks, Experiments and Challenges1 May 2024
- 2024MSPipe: Efficient Temporal GNN Training via Staleness-Aware Pipeline23 Feb 2024