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 条
  • [21] Revisiting Adversarial Attacks on Graph Neural Networks for Graph Classification
    Wang, Xin
    Chang, Heng
    Xie, Beini
    Bian, Tian
    Zhou, Shiji
    Wang, Daixin
    Zhang, Zhiqiang
    Zhu, Wenwu
    IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2024, 36 (05) : 2166 - 2178
  • [22] Superpixel Image Classification with Graph Attention Networks
    Avelar, Pedro H. C.
    Tavares, Anderson R.
    da Silveira, Thiago L. T.
    Jung, Cliudio R.
    Lamb, Luis C.
    2020 33RD SIBGRAPI CONFERENCE ON GRAPHICS, PATTERNS AND IMAGES (SIBGRAPI 2020), 2020, : 203 - 209
  • [23] Hierarchical Graph Convolutional Networks for Image Classification
    Batisteli, João Pedro Oliveira
    Guimarães, Silvio Jamil Ferzoli
    do Patrocínio Júnior, Zenilton Kleber Gonçalves
    Lecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics), 2023, 14196 LNAI : 63 - 76
  • [24] GNNGLY: Graph Neural Networks for Glycan Classification
    Alkuhlani, Alhasan
    Gad, Walaa
    Roushdy, Mohamed
    Salem, Abdel-Badeeh M.
    IEEE ACCESS, 2023, 11 : 51838 - 51847
  • [25] Graph Convolutional Networks for Hyperspectral Image Classification
    Hong, Danfeng
    Gao, Lianru
    Yao, Jing
    Zhang, Bing
    Plaza, Antonio
    Chanussot, Jocelyn
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2021, 59 (07): : 5966 - 5978
  • [26] A comparison of graph neural networks for malware classification
    Malhotra, Vrinda
    Potika, Katerina
    Stamp, Mark
    JOURNAL OF COMPUTER VIROLOGY AND HACKING TECHNIQUES, 2024, 20 (01) : 53 - 69
  • [27] Recurrent Graph Neural Networks for Text Classification
    Wei, Xinde
    Huang, Hai
    Ma, Longxuan
    Yang, Ze
    Xu, Liutong
    PROCEEDINGS OF 2020 IEEE 11TH INTERNATIONAL CONFERENCE ON SOFTWARE ENGINEERING AND SERVICE SCIENCE (ICSESS 2020), 2020, : 91 - 97
  • [28] A comparison of graph neural networks for malware classification
    Vrinda Malhotra
    Katerina Potika
    Mark Stamp
    Journal of Computer Virology and Hacking Techniques, 2024, 20 : 53 - 69
  • [29] On Calibration of Graph Neural Networks for Node Classification
    Liu, Tong
    Liu, Yushan
    Hildebrandt, Marcel
    Joblin, Mitchell
    Li, Hang
    Tresp, Volker
    2022 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2022,
  • [30] Graph Neural Networks for IceCube Signal Classification
    Choma, Nicholas
    Monti, Federico
    Gerhardt, Lisa
    Palczewski, Tomasz
    Ronaghi, Zahra
    Prabhat
    Bhimji, Wahid
    Bronstein, Michael M.
    Klein, Spencer R.
    Bruna, Joan
    2018 17TH IEEE INTERNATIONAL CONFERENCE ON MACHINE LEARNING AND APPLICATIONS (ICMLA), 2018, : 386 - 391