QGTC: Accelerating Quantized Graph Neural Networks via GPU Tensor Core
Not on the map. Why?
Abstract (arXiv)
Over the most recent years, quantized graph neural network (QGNN) attracts lots of research and industry attention due to its high robustness and low computation and memory overhead. Unfortunately, the performance gains of QGNN have never been realized on modern GPU platforms. To this end, we propose the first Tensor Core (TC) based computing framework, QGTC, to support any-bitwidth computation for QGNNs on GPUs. We introduce a novel quantized low-bit arithmetic design based on the low-bit data representation and bit-decomposed computation. We craft a novel TC-tailored CUDA kernel design by incorporating 3D-stacked bit compression, zero-tile jumping, and non-zero tile reuse technique to improve the performance systematically. We incorporate an effective bandwidth-optimized subgraph packing strategy to maximize the transferring efficiency between CPU host and GPU device. We integrate QGTC with Pytorch for better programmability and extensibility. Extensive experiments demonstrate that QGTC achieves on average 2.7x speedup compared with the state-of-the-art Deep Graph Library framework across diverse settings.
Key facts. Cited by 18 papers in the map (data through 14 Oct 2026). Builds on 18 of its 42 references that are in the map. First posted on arXiv 18 Nov 2021.
Builds on
18 of 42 references.
References · 18
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 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
- 2018How Powerful are Graph Neural Networks?cited by 4,393
- 2019Fast Graph Representation Learning with PyTorch Geometriccited by 2,434
- 2020Open Graph Benchmark: Datasets for Machine Learning on Graphscited by 2,163
- 2016node2vec: Scalable Feature Learning for Networkscited by 1,918
- 2018Hierarchical Graph Representation Learning with Differentiable Poolingcited by 787
- 2019Cluster-GCN: An Efficient Algorithm for Training Deep and Large Graph Convolutional Networkscited by 541
- 2019Deep Graph Library: A Graph-Centric, Highly-Performant Package for Graph Neural Networkscited by 242
- 2020Degree-Quant: Quantization-Aware Training for Graph Neural Networkscited by 96
6 more
- 2020Binary Graph Neural Networkscited by 34
- 2019GraphACT: Accelerating GCN Training on CPU-FPGA Heterogeneous Platformscited by 31
- 2020SGQuant: Squeezing the Last Bit on Graph Neural Networks with Specialized Quantizationcited by 27
- 2021POLE: Polarized Embedding for Signed Networkscited by 17
- 2021A Broader Picture of Random-walk Based Graph Embeddingcited by 16
- 2020Accelerating Binarized Neural Networks via Bit-Tensor-Cores in Turing GPUscited by 14
Built on by
Citing papers per year: 2021: 1; 2022: 5; 2023: 5; 2024: 3; 2025: 3; 2026: 1.
18 papers.
Sort:
By area:
- 2022SparseTIR: Composable Abstractions for Sparse Compilation in Deep Learningcited by 47
- 2022Distributed Graph Neural Network Training: A Surveycited by 32
- 2022Global Counterfactual Explainer for Graph Neural Networkscited by 30
- 2021TC-GNN: Bridging Sparse GNN Computation and Dense Tensor Cores on GPUscited by 17
- 2023The Evolution of Distributed Systems for Graph Neural Networks and their Origin in Graph Processing and Deep Learning: A Surveycited by 14
- 2023Low-bit Quantization for Deep Graph Neural Networks with Smoothness-aware Message Propagationcited by 5
- 2024DiskGNN: Bridging I/O Efficiency and Model Accuracy for Out-of-Core GNN Trainingcited by 5
- 2023Tango: rethinking quantization for graph neural network training on GPUscited by 4
- 2024Quasar-ViT: Hardware-Oriented Quantization-Aware Architecture Search for Vision Transformerscited by 4
- 2023Boosting Distributed Full-graph GNN Training with Asynchronous One-bit Communicationcited by 3
- 2025Efficient Mixed Precision Quantization in Graph Neural Networkscited by 3
- 2024FastCHGNet: Training one Universal Interatomic Potential to 1.5 Hours with 32 GPUscited by 2
6 more
- 2025FlexQ: Efficient Post-training INT6 Quantization for LLM Serving via Algorithm-System Co-Designcited by 2
- 2023DGC: Training Dynamic Graphs with Spatio-Temporal Non-Uniformity using Graph Partitioning by Chunkscited by 1
- 2025Libra: Unleashing GPU Heterogeneity for High-Performance Sparse Matrix Multiplicationcited by 1
- 2022Faith: An Efficient Framework for Transformer Verification on GPUscited by 0
- 2022AcceRL: Policy Acceleration Framework for Deep Reinforcement Learningcited by 0
- 2026Accelerating GPU Inference of Large Language Models with Moderately Unstructured Sparse Weight Matricescited by 0
- 2026Accelerating GPU Inference of Large Language Models with Moderately Unstructured Sparse Weight Matrices13 Jun 2026
- 2025FlexQ: Efficient Post-training INT6 Quantization for LLM Serving via Algorithm-System Co-Design6 Aug 2025
- 2025Libra: Unleashing GPU Heterogeneity for High-Performance Sparse Matrix Multiplication28 Jun 2025
- 2025Efficient Mixed Precision Quantization in Graph Neural Networks14 May 2025
- 2024FastCHGNet: Training one Universal Interatomic Potential to 1.5 Hours with 32 GPUs30 Dec 2024
- 2024Quasar-ViT: Hardware-Oriented Quantization-Aware Architecture Search for Vision Transformers25 Jul 2024
- 2024DiskGNN: Bridging I/O Efficiency and Model Accuracy for Out-of-Core GNN Training8 May 2024
- 2023DGC: Training Dynamic Graphs with Spatio-Temporal Non-Uniformity using Graph Partitioning by Chunks7 Sep 2023
- 2023Low-bit Quantization for Deep Graph Neural Networks with Smoothness-aware Message Propagation29 Aug 2023
- 2023Tango: rethinking quantization for graph neural network training on GPUs2 Aug 2023
- 2023The Evolution of Distributed Systems for Graph Neural Networks and their Origin in Graph Processing and Deep Learning: A Survey23 May 2023
- 2023Boosting Distributed Full-graph GNN Training with Asynchronous One-bit Communication2 Mar 2023
6 more
- 2022AcceRL: Policy Acceleration Framework for Deep Reinforcement Learning28 Nov 2022
- 2022Distributed Graph Neural Network Training: A Survey1 Nov 2022
- 2022Global Counterfactual Explainer for Graph Neural Networks21 Oct 2022
- 2022Faith: An Efficient Framework for Transformer Verification on GPUs23 Sep 2022
- 2022SparseTIR: Composable Abstractions for Sparse Compilation in Deep Learning11 Jul 2022
- 2021TC-GNN: Bridging Sparse GNN Computation and Dense Tensor Cores on GPUs3 Dec 2021
- 2022SparseTIR: Composable Abstractions for Sparse Compilation in Deep Learningcited by 47
- 2022Distributed Graph Neural Network Training: A Surveycited by 32
- 2022Global Counterfactual Explainer for Graph Neural Networkscited by 30
- 2021TC-GNN: Bridging Sparse GNN Computation and Dense Tensor Cores on GPUscited by 17
- 2023The Evolution of Distributed Systems for Graph Neural Networks and their Origin in Graph Processing and Deep Learning: A Surveycited by 14
- 2023Low-bit Quantization for Deep Graph Neural Networks with Smoothness-aware Message Propagationcited by 5
- 2024DiskGNN: Bridging I/O Efficiency and Model Accuracy for Out-of-Core GNN Trainingcited by 5
- 2023Tango: rethinking quantization for graph neural network training on GPUscited by 4
- 2024Quasar-ViT: Hardware-Oriented Quantization-Aware Architecture Search for Vision Transformerscited by 4
- 2023Boosting Distributed Full-graph GNN Training with Asynchronous One-bit Communicationcited by 3
- 2025Efficient Mixed Precision Quantization in Graph Neural Networkscited by 3
- 2024FastCHGNet: Training one Universal Interatomic Potential to 1.5 Hours with 32 GPUscited by 2
12 most cited of 17 All 17 (JSON)
- 2026Accelerating GPU Inference of Large Language Models with Moderately Unstructured Sparse Weight Matrices13 Jun 2026
- 2025Libra: Unleashing GPU Heterogeneity for High-Performance Sparse Matrix Multiplication28 Jun 2025
- 2025Efficient Mixed Precision Quantization in Graph Neural Networks14 May 2025
- 2024FastCHGNet: Training one Universal Interatomic Potential to 1.5 Hours with 32 GPUs30 Dec 2024
- 2024Quasar-ViT: Hardware-Oriented Quantization-Aware Architecture Search for Vision Transformers25 Jul 2024
- 2024DiskGNN: Bridging I/O Efficiency and Model Accuracy for Out-of-Core GNN Training8 May 2024
- 2023DGC: Training Dynamic Graphs with Spatio-Temporal Non-Uniformity using Graph Partitioning by Chunks7 Sep 2023
- 2023Low-bit Quantization for Deep Graph Neural Networks with Smoothness-aware Message Propagation29 Aug 2023
- 2023Tango: rethinking quantization for graph neural network training on GPUs2 Aug 2023
- 2023The Evolution of Distributed Systems for Graph Neural Networks and their Origin in Graph Processing and Deep Learning: A Survey23 May 2023
- 2023Boosting Distributed Full-graph GNN Training with Asynchronous One-bit Communication2 Mar 2023
- 2022AcceRL: Policy Acceleration Framework for Deep Reinforcement Learning28 Nov 2022
12 newest of 17 All 17 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2021APNN-TC: Accelerating Arbitrary Precision Neural Networks on Ampere GPU Tensor Cores
- 2022BiFeat: Supercharge GNN Training via Graph Feature Quantization
- 2021LW-GCN: A Lightweight FPGA-based Graph Convolutional Network Accelerator
- 2023MEGA: A Memory-Efficient GNN Accelerator Exploiting Degree-Aware Mixed-Precision Quantization
- 2024Efficient Arbitrary Precision Acceleration for Large Language Models on GPU Tensor Cores
- 2023Activation Compression of Graph Neural Networks using Block-wise Quantization with Improved Variance Minimization
- 2023GNNHLS: Evaluating Graph Neural Network Inference via High-Level Synthesis
- 2026NCCLZ: Compression-Enabled GPU Collectives with Decoupled Quantization and Entropy Coding