Feature pyramid-based graph convolutional neural network for graph classification

被引:6
|
作者
Lu, Mingming [1 ]
Xiao, Zhixiang [1 ]
Li, Haifeng [2 ]
Zhang, Ya [3 ]
Xiong, Neal N. [4 ]
机构
[1] Cent South Univ, Sch Comp Sci & Engn, Changsha 410083, Peoples R China
[2] Cent South Univ, Sch Geosci & Info Phys, Changsha 410083, Peoples R China
[3] State Grid Changsha Power Supply Co, Changsha 410083, Peoples R China
[4] Ross State Univ, Dept Comp Sci & Math, Alpine, TX 79830 USA
基金
中国国家自然科学基金;
关键词
Feature pyramid; Graph classification; Graph neural network; Graph pooling;
D O I
10.1016/j.sysarc.2022.102562
中图分类号
TP3 [计算技术、计算机技术];
学科分类号
0812 ;
摘要
As an important task of Graph Neural Networks (GNN), graph classification has received increasing attention, as it can be widely used in numerous fields, such as protein prediction and community prediction. The GNN models for graph classification usually require aggregating the structure and feature information of an input graph into a hidden representation vector. Such a technique can be generally referred as graph pooling, which intends to maximize the removal of noise while minimize the loss of important features. Therefore, there exists a trade-off between the noise reduction and information maximization that needs to be balanced for better performance. However, the existing pooling-based GNNs for graph classification depends on the manual setting of the graphpooling degree, making it difficult to balance the trade-off. To this end, we propose a Feature Pyramid-based Graph Convolutional Neural network for Graph Classification (FPGCN-GC), which constructs multi-scale hierarchical information fusion to reduce information loss, and achieves an adaptive feature fusion through a learnable weighted residual connection and self-attention mechanism. The superior performance of the proposed method is verified on multiple graph classification datasets, thus illustrating the effectiveness and superiority of FPGCN-GC. In addition, we visualize FPGCN-GC by T-SNE to further analyze the underlying reason for its effectiveness.
引用
收藏
页数:10
相关论文
共 50 条
  • [1] Feature pyramid-based convolutional neural network image inpainting
    Shengbo Wang
    Xiuyou Wang
    Signal, Image and Video Processing, 2024, 18 : 437 - 443
  • [2] Feature pyramid-based convolutional neural network image inpainting
    Wang, Shengbo
    Wang, Xiuyou
    SIGNAL IMAGE AND VIDEO PROCESSING, 2024, 18 (01) : 437 - 443
  • [3] Pyramid Cascaded Convolutional Neural Network with Graph Convolution for Hyperspectral Image Classification
    Pan, Haizhu
    Yan, Hui
    Ge, Haimiao
    Wang, Liguo
    Shi, Cuiping
    REMOTE SENSING, 2024, 16 (16)
  • [4] A deep graph convolutional neural network architecture for graph classification
    Zhou, Yuchen
    Huo, Hongtao
    Hou, Zhiwen
    Bu, Fanliang
    PLOS ONE, 2023, 18 (03):
  • [5] A deep graph convolutional neural network architecture for graph classification
    Zhou, Yuchen
    Huo, Hongtao
    Hou, Zhiwen
    Bu, Fanliang
    PLOS BIOLOGY, 2023, 21 (03)
  • [6] A Convolutional Neural Network and Graph Convolutional Network Based Framework for AD Classification
    Lin, Lan
    Xiong, Min
    Zhang, Ge
    Kang, Wenjie
    Sun, Shen
    Wu, Shuicai
    SENSORS, 2023, 23 (04)
  • [7] Evaluating the potential of pyramid-based fusion coupled with convolutional neural network for satellite image classification
    Achala Shakya
    Mantosh Biswas
    Mahesh Pal
    Arabian Journal of Geosciences, 2022, 15 (8)
  • [8] WTRPNet: An Explainable Graph Feature Convolutional Neural Network for Epileptic EEG Classification
    Xin, Qi
    Hu, Shaohao
    Liu, Shuaiqi
    Zhao, Ling
    Wang, Shuihua
    ACM TRANSACTIONS ON MULTIMEDIA COMPUTING COMMUNICATIONS AND APPLICATIONS, 2021, 17 (03)
  • [9] A novel graph convolutional feature based convolutional neural network for stock trend prediction
    Chen, Wei
    Jiang, Manrui
    Zhang, Wei-Guo
    Chen, Zhensong
    INFORMATION SCIENCES, 2021, 556 : 67 - 94
  • [10] Weighted Feature Fusion of Convolutional Neural Network and Graph Attention Network for Hyperspectral Image Classification
    Dong, Yanni
    Liu, Quanwei
    Du, Bo
    Zhang, Liangpei
    IEEE TRANSACTIONS ON IMAGE PROCESSING, 2022, 31 : 1559 - 1572