Quantized Graph Neural Networks for Image Classification

被引:1
|
作者
Xu, Xinbiao [1 ]
Ma, Liyan [1 ]
Zeng, Tieyong [2 ]
Huang, Qinghua [3 ]
机构
[1] Shanghai Univ, Sch Comp Engn & Sci, Shanghai 200444, Peoples R China
[2] Chinese Univ Hong Kong, Dept Math, Hong Kong 999077, Peoples R China
[3] Shanghai Univ, Sch Commun & Informat Engn, Shanghai 200444, Peoples R China
关键词
graph neural network; model quantization; knowledge distillation; image classification;
D O I
10.3390/math11244927
中图分类号
O1 [数学];
学科分类号
0701 ; 070101 ;
摘要
Researchers have resorted to model quantization to compress and accelerate graph neural networks (GNNs). Nevertheless, several challenges remain: (1) quantization functions overlook outliers in the distribution, leading to increased quantization errors; (2) the reliance on full-precision teacher models results in higher computational and memory overhead. To address these issues, this study introduces a novel framework called quantized graph neural networks for image classification (QGNN-IC), which incorporates a novel quantization function, Pauta quantization (PQ), and two innovative self-distillation methods, attention quantization distillation (AQD) and stochastic quantization distillation (SQD). Specifically, PQ utilizes the statistical characteristics of distribution to effectively eliminate outliers, thereby promoting fine-grained quantization and reducing quantization errors. AQD enhances the semantic information extraction capability by learning from beneficial channels via attention. SQD enhances the quantization robustness through stochastic quantization. AQD and SQD significantly improve the performance of the quantized model with minimal overhead. Extensive experiments show that QGNN-IC not only surpasses existing state-of-the-art quantization methods but also demonstrates robust generalizability.
引用
收藏
页数:16
相关论文
共 50 条
  • [41] Active and Semi-Supervised Graph Neural Networks for Graph Classification
    Xie, Yu
    Lv, Shengze
    Qian, Yuhua
    Wen, Chao
    Liang, Jiye
    IEEE TRANSACTIONS ON BIG DATA, 2022, 8 (04) : 920 - 932
  • [42] Polynomial-based graph convolutional neural networks for graph classification
    Luca Pasa
    Nicolò Navarin
    Alessandro Sperduti
    Machine Learning, 2022, 111 : 1205 - 1237
  • [43] Two-Stage Training of Graph Neural Networks for Graph Classification
    Do, Manh Tuan
    Park, Noseong
    Shin, Kijung
    NEURAL PROCESSING LETTERS, 2023, 55 (03) : 2799 - 2823
  • [44] Revisiting Attention-Based Graph Neural Networks for Graph Classification
    Tao, Ye
    Li, Ying
    Wu, Zhonghai
    PARALLEL PROBLEM SOLVING FROM NATURE - PPSN XVII, PPSN 2022, PT I, 2022, 13398 : 442 - 458
  • [45] Differentially Private Graph Neural Networks for Whole-Graph Classification
    Mueller, Tamara T.
    Paetzold, Johannes C.
    Prabhakar, Chinmay
    Usynin, Dmitrii
    Rueckert, Daniel
    Kaissis, Georgios
    IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2023, 45 (06) : 7308 - 7318
  • [46] Two-Stage Training of Graph Neural Networks for Graph Classification
    Manh Tuan Do
    Noseong Park
    Kijung Shin
    Neural Processing Letters, 2023, 55 : 2799 - 2823
  • [47] Bipartite Graph Coarsening for Text Classification Using Graph Neural Networks
    dos Santos, Nicolas Roque
    Minatel, Diego
    Baria Valejo, Alan Demetrius
    Lopes, Alneu de A.
    PROGRESS IN PATTERN RECOGNITION, IMAGE ANALYSIS, COMPUTER VISION, AND APPLICATIONS, CIARP 2023, PT I, 2024, 14469 : 589 - 604
  • [48] Polynomial-based graph convolutional neural networks for graph classification
    Pasa, Luca
    Navarin, Nicolo
    Sperduti, Alessandro
    MACHINE LEARNING, 2022, 111 (04) : 1205 - 1237
  • [49] Graph alternate learning for robust graph neural networks in node classification
    Baoliang Zhang
    Xiaoxin Guo
    Zhenchuan Tu
    Jia Zhang
    Neural Computing and Applications, 2022, 34 : 8723 - 8735
  • [50] Graph alternate learning for robust graph neural networks in node classification
    Zhang, Baoliang
    Guo, Xiaoxin
    Tu, Zhenchuan
    Zhang, Jia
    NEURAL COMPUTING & APPLICATIONS, 2022, 34 (11): : 8723 - 8735