MQ-GNN: A Multi-Queue Pipelined Architecture for Scalable and Efficient GNN Training
Not on the map. Why?
Abstract (arXiv)
Graph Neural Networks (GNNs) are powerful tools for learning graph-structured data, but their scalability is hindered by inefficient mini-batch generation, data transfer bottlenecks, and costly inter-GPU synchronization. Existing training frameworks fail to overlap these stages, leading to suboptimal resource utilization. This paper proposes MQ-GNN, a multi-queue pipelined framework that maximizes training efficiency by interleaving GNN training stages and optimizing resource utilization. MQ-GNN introduces Ready-to-Update Asynchronous Consistent Model (RaCoM), which enables asynchronous gradient sharing and model updates while ensuring global consistency through adaptive periodic synchronization. Additionally, it employs global neighbor sampling with caching to reduce data transfer overhead and an adaptive queue-sizing strategy to balance computation and memory efficiency. Experiments on four large-scale datasets and ten baseline models demonstrate that MQ-GNN achieves up to boldmath bm4.6,× faster training time and 30% improved GPU utilization while maintaining competitive accuracy. These results establish MQ-GNN as a scalable and efficient solution for multi-GPU GNN training.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 26 of its 42 references that are in the map. First posted on arXiv 8 Jan 2026.
Builds on
26 of 42 references.
References · 26
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,282
- 2017Graph Attention Networkscited by 8,854
- 2017Inductive Representation Learning on Large Graphscited by 6,284
- 2019A Comprehensive Survey on Graph Neural Networkscited by 2,906
- 2019Fast Graph Representation Learning with PyTorch Geometriccited by 2,414
- 2020Open Graph Benchmark: Datasets for Machine Learning on Graphscited by 2,148
- 2019Graph Neural Networks for Social Recommendationcited by 711
- 2019GraphSAINT: Graph Sampling Based Inductive Learning Methodcited by 618
- 2018FastGCN: Fast Learning with Graph Convolutional Networks via Importance Samplingcited by 583
- 2018Deep Learning on Graphs: A Surveycited by 393
- 2018Adaptive Sampling Towards Fast Graph Representation Learningcited by 180
- 2019Layer-Dependent Importance Sampling for Training Deep and Large Graph Convolutional Networkscited by 177
14 more
- 2020Machine Learning on Graphs: A Model and Comprehensive Taxonomycited by 156
- 2021Dorylus: Affordable, Scalable, and Accurate GNN Training with Distributed CPU Servers and Serverless Threadscited by 51
- 2022PipeGCN: Efficient Full-Graph Training of Graph Convolutional Networks with Pipelined Feature Communicationcited by 45
- 2019Estimating Node Importance in Knowledge Graphs Using Graph Neural Networkscited by 43
- 2021Accelerating Training and Inference of Graph Neural Networks with Fast Sampling and Pipeliningcited by 39
- 2021Distributed Hybrid CPU and GPU training for Graph Neural Networks on Billion-Scale Graphscited by 33
- 2022Distributed Graph Neural Network Training: A Surveycited by 32
- 2021Marius: Learning Massive Graph Embeddings on a Single Machinecited by 29
- 2021Global Neighbor Sampling for Mixed CPU-GPU Training on Giant Graphscited by 22
- 2022Saga: A Platform for Continuous Construction and Serving of Knowledge At Scalecited by 18
- 2021PyTorch-Direct: Enabling GPU Centric Data Access for Very Large Graph Neural Network Training with Irregular Accessescited by 12
- 2024MSPipe: Efficient Temporal GNN Training via Staleness-Aware Pipelinecited by 12
- 2022Calibrate and Debias Layer-wise Sampling for Graph Convolutional Networkscited by 5
- 2023GNNPipe: Scaling Deep GNN Training with Pipelined Model Parallelismcited by 4
Built on by
Not cited yet.