mlsanityData to 7 Oct 2026

MQ-GNN: A Multi-Queue Pipelined Architecture for Scalable and Efficient GNN Training

arXiv 2601.04707 ↗First posted 8 Jan 2026cs.LG cs.AI cs.DCcited by 0

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

  1. 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,282
  2. 2017Graph Attention Networkscited by 8,854
  3. 2017Inductive Representation Learning on Large Graphscited by 6,284
  4. 2019A Comprehensive Survey on Graph Neural Networkscited by 2,906
  5. 2019Fast Graph Representation Learning with PyTorch Geometriccited by 2,414
  6. 2020Open Graph Benchmark: Datasets for Machine Learning on Graphscited by 2,148
  7. 2019Graph Neural Networks for Social Recommendationcited by 711
  8. 2019GraphSAINT: Graph Sampling Based Inductive Learning Methodcited by 618
  9. 2018FastGCN: Fast Learning with Graph Convolutional Networks via Importance Samplingcited by 583
  10. 2018Deep Learning on Graphs: A Surveycited by 393
  11. 2018Adaptive Sampling Towards Fast Graph Representation Learningcited by 180
  12. 2019Layer-Dependent Importance Sampling for Training Deep and Large Graph Convolutional Networkscited by 177
14 more
  1. 2020Machine Learning on Graphs: A Model and Comprehensive Taxonomycited by 156
  2. 2021Dorylus: Affordable, Scalable, and Accurate GNN Training with Distributed CPU Servers and Serverless Threadscited by 51
  3. 2022PipeGCN: Efficient Full-Graph Training of Graph Convolutional Networks with Pipelined Feature Communicationcited by 45
  4. 2019Estimating Node Importance in Knowledge Graphs Using Graph Neural Networkscited by 43
  5. 2021Accelerating Training and Inference of Graph Neural Networks with Fast Sampling and Pipeliningcited by 39
  6. 2021Distributed Hybrid CPU and GPU training for Graph Neural Networks on Billion-Scale Graphscited by 33
  7. 2022Distributed Graph Neural Network Training: A Surveycited by 32
  8. 2021Marius: Learning Massive Graph Embeddings on a Single Machinecited by 29
  9. 2021Global Neighbor Sampling for Mixed CPU-GPU Training on Giant Graphscited by 22
  10. 2022Saga: A Platform for Continuous Construction and Serving of Knowledge At Scalecited by 18
  11. 2021PyTorch-Direct: Enabling GPU Centric Data Access for Very Large Graph Neural Network Training with Irregular Accessescited by 12
  12. 2024MSPipe: Efficient Temporal GNN Training via Staleness-Aware Pipelinecited by 12
  13. 2022Calibrate and Debias Layer-wise Sampling for Graph Convolutional Networkscited by 5
  14. 2023GNNPipe: Scaling Deep GNN Training with Pipelined Model Parallelismcited by 4

Built on by

Not cited yet.