Quantization-Aware Acceleration of Graph Neural Networks on GPUs
Keywords:
graph neural networks; quantization; GPU acceleration; systems co-design; sustainable artificial intelligence; robustness; fairness; infrastructure governanceAbstract
Graph neural networks have become a central computational model for relational data, spanning recommendation, chemistry, transportation, social analysis, and scientific discovery. Their deployment on graphics processing units has enabled substantial progress, yet graph workloads remain difficult to accelerate because they combine irregular memory access, dynamic neighbor sampling, sparse matrix operations, and dense neural computation. Quantization-aware acceleration offers a powerful but structurally complicated response to these pressures. By reducing numerical precision in representations, activations, messages, and gradients, quantization can lower memory bandwidth demand, increase arithmetic throughput, and make larger graphs tractable on fixed hardware. However, the benefits are not automatic. Quantized graph neural networks introduce new trade-offs among accuracy, robustness, fairness, compiler support, runtime scheduling, multi-GPU communication, and long-term infrastructure sustainability. This paper examines quantization-aware acceleration as a system-level problem rather than a purely algorithmic compression technique. It analyzes how low-precision arithmetic interacts with GPU kernel design, memory hierarchies, graph sampling, distributed training, and deployment governance. It also considers how quantization decisions affect vulnerable populations when graph models are used in high-stakes domains, and how documentation, auditing, and policy frameworks must adapt. The discussion concludes that effective quantization for graph neural networks requires co-design across hardware, compilers, runtime systems, model architectures, and institutional governance. Without such integration, quantization may deliver local efficiency gains while producing hidden costs in accuracy, inequity, and operational fragility.
References
1. Kipf, T. N., & Welling, M. (2017). Semi-supervised classification with graph convolutional networks. In International Conference on Learning Representations.
2. Veličković, P., Cucurull, G., Casanova, A., Romero, A., Liò, P., & Bengio, Y. (2018). Graph attention networks. In International Conference on Learning Representations.
3. Hamilton, W. L., Ying, R., & Leskovec, J. (2017). Inductive representation learning on large graphs. In Advances in Neural Information Processing Systems (pp. 1024–1034).
4. Scarselli, F., Gori, M., Tsoi, A. C., Hagenbuchner, M., & Monfardini, G. (2009). The graph neural network model. IEEE Transactions on Neural Networks, 20(1), 61–80.
5. Wu, Z., Pan, S., Chen, F., Long, G., Zhang, C., & Yu, P. S. (2021). A comprehensive survey on graph neural networks. IEEE Transactions on Neural Networks and Learning Systems, 32(1), 4–24.
6. Zhou, J., Cui, G., Hu, S., Zhang, Z., Yang, C., Liu, Z., Wang, L., Li, C., & Sun, M. (2020). Graph neural networks: A review of methods and applications. AI Open, 1, 57–81.
7. Peng, H., Xie, X., Shivdikar, K., Hasan, M. A., Zhao, J., Huang, S., ... & Ding, C. (2024, April). Maxk-gnn: Extremely fast gpu kernel design for accelerating graph neural networks training. In Proceedings of the 29th ACM International Conference on Architectural Support for Programming Languages and Operating Systems, Volume 2 (pp. 683-698).
8. Dave, S., Baghdadi, R., Nowatzki, T., Avancha, S., Shrivastava, A., & Li, B. (2020). Hardware acceleration of graph neural networks. In 2020 57th ACM/IEEE Design Automation Conference (DAC) (pp. 1–6). IEEE.
9. Wang, Y., Feng, B., Li, G., Xie, S., Li, Y., Ding, Y., & Xie, Y. (2021). GNNAdvisor: An adaptive and efficient runtime system for GNN acceleration on GPUs. In 15th USENIX Symposium on Operating Systems Design and Implementation (OSDI 21) (pp. 515–531).
10. Lin, Z., Li, C., Miao, Y., Liu, Y., & Xu, Y. (2020). PaGraph: Scaling GNN training on large graphs via computation-aware caching. In Proceedings of the 11th ACM Symposium on Cloud Computing (pp. 401–415).
11. Gupta, S., Agrawal, A., Gopalakrishnan, K., & Narayanan, P. (2015). Deep learning with limited numerical precision. In International Conference on Machine Learning (pp. 1737–1746). PMLR.
12. Jacob, B., Kligys, S., Chen, B., Zhu, M., Tang, M., Howard, A., Adam, H., & Kalenichenko, D. (2018). Quantization and training of neural networks for efficient integer-arithmetic-only inference. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (pp. 2704–2713).
13. Krishnamoorthi, R. (2018). Quantizing deep convolutional networks for efficient inference: A whitepaper. arXiv preprint arXiv:1806.08342.
14. Wang, K., Liu, Z., Lin, Y., Lin, J., & Han, S. (2019). HAQ: Hardware-aware automated quantization with mixed precision. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (pp. 8612–8620).
15. Micikevicius, P., Stosic, D., Burgess, N., Cornea, M., Dubey, P., Grisenthwaite, R., Ha, S., Heinecke, A., Judd, P., Kamalu, J., Mellempudi, N., Oberman, S., Park, S., Root, B., & Siu, M. (2022). FP8 formats for deep learning. arXiv preprint arXiv:2209.05433.
16. Dwork, C., & Roth, A. (2014). The algorithmic foundations of differential privacy. Foundations and Trends in Theoretical Computer Science, 9(3–4), 211–407.
17. Gebru, T., Morgenstern, J., Vecchione, B., Vaughan, J. W., Wallach, H., Daumé III, H., & Crawford, K. (2021). Datasheets for datasets. Communications of the ACM, 64(12), 86–92.
18. Kang, J., & Tong, H. (2023). Fairness in graph mining: A survey. IEEE Transactions on Knowledge and Data Engineering, 35(10), 10588–10609.
19. Chen, T., Moreau, T., Jiang, Z., Zheng, L., Yan, E., Cowan, M., Shen, H., Wang, L., Hu, Y., Ceze, L., Guestrin, C., & Krishnamurthy, A. (2018). TVM: An automated end-to-end optimizing compiler for deep learning. In 13th USENIX Symposium on Operating Systems Design and Implementation (OSDI 18) (pp. 578–594).
20. Paszke, A., Gross, S., Massa, F., Lerer, A., Bradbury, J., Chanan, G., Killeen, T., Lin, Z., Gimelshein, N., Antiga, L., Desmaison, A., Kopf, A., Yang, E., DeVito, Z., Raison, M., Tejani, A., Chilamkurthy, S., Steiner, B., Fang, L., Bai, J., & Chintala, S. (2019). PyTorch: An imperative style, high-performance deep learning library. In Advances in Neural Information Processing Systems (pp. 8024–8035).
Downloads
Published
Issue
Section
License
Copyright (c) 2026 Journal of Engineering Systems and Digital Innovation

This work is licensed under a Creative Commons Attribution 4.0 International License.