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 条
  • [1] Graph Neural Networks for Colorectal Histopathological Image Classification
    Tepe, Esra
    Bilgin, Gokhan
    2022 MEDICAL TECHNOLOGIES CONGRESS (TIPTEKNO'22), 2022,
  • [2] Advances of Hyperspectral Image Classification Based on Graph Neural Networks
    Wan S.
    Yang J.
    Gong C.
    Tien Tzu Hsueh Pao/Acta Electronica Sinica, 2023, 51 (06): : 1687 - 1709
  • [3] A review of hyperspectral image classification based on graph neural networks
    Xiaofeng Zhao
    Junyi Ma
    Lei Wang
    Zhili Zhang
    Yao Ding
    Xiongwu Xiao
    Artificial Intelligence Review, 58 (6)
  • [4] Geometric Superpixel Representations for Efficient Image Classification with Graph Neural Networks
    Cosma, Radu A.
    Knobel, Lukas
    van der Linden, Putri
    Knigge, David M.
    Bekkers, Erik J.
    2023 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION WORKSHOPS, ICCVW, 2023, : 109 - 118
  • [5] Geometric Superpixel Representations for Efficient Image Classification with Graph Neural Networks
    Cosma, Radu A.
    Knobel, Lukas
    Van Der Linden, Putri
    Knigge, David M.
    Bekkers, Erik J.
    Proceedings - 2023 IEEE/CVF International Conference on Computer Vision Workshops, ICCVW 2023, 2023, : 109 - 118
  • [6] GRAPH NEURAL NETWORKS FOR THE CROSS-DOMAIN HISTOPATHOLOGICAL IMAGE CLASSIFICATION
    Cai, Chang
    Xu, Dou
    Fang, Chaowei
    Yang, Meng
    Li, Zhongyu
    2021 IEEE 18TH INTERNATIONAL SYMPOSIUM ON BIOMEDICAL IMAGING (ISBI), 2021, : 1953 - 1957
  • [7] Semisupervised Graph Neural Networks for Graph Classification
    Xie, Yu
    Liang, Yanfeng
    Gong, Maoguo
    Qin, A. K.
    Ong, Yew-Soon
    He, Tiantian
    IEEE TRANSACTIONS ON CYBERNETICS, 2023, 53 (10) : 6222 - 6235
  • [8] Adaptive gradients and weight projection based on quantized neural networks for efficient image classification
    Wu, Ran
    Liu, Huanyu
    Li, Jun-Bao
    COMPUTER VISION AND IMAGE UNDERSTANDING, 2022, 223
  • [9] Hierarchical Graph Neural Networks with Scale-Aware Readout for Image Classification
    Oliveira Batisteli, Joao Pedro
    Ferzoli Guimaraes, Silvio Jamil
    Goncalves do Patrocinio, Zenilton Kleber
    INTERNATIONAL JOURNAL OF SEMANTIC COMPUTING, 2024, : 713 - 738
  • [10] Contextual Image Classification Through Fine-Tuned Graph Neural Networks
    Campos, Walacy S.
    Souza, Luis G.
    Saito, Priscila T. M.
    Bugatti, Pedro H.
    ARTIFICIAL INTELLIGENCE AND SOFT COMPUTING (ICAISC 2021), PT II, 2021, 12855 : 15 - 24