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 条
  • [31] An Improved Graph Convolutional Neural Network based on Graph Auto-encoder
    Wang, Dongqi
    Du, Tianqi
    Liu, Zhongwu
    Chen, Dongming
    Ren, Tao
    2024 16TH INTERNATIONAL CONFERENCE ON COMPUTER AND AUTOMATION ENGINEERING, ICCAE 2024, 2024, : 442 - 446
  • [32] Graph contrast learning for recommendation based on relational graph convolutional neural network
    Liu, Xiaoyang
    Feng, Hanwen
    Zhang, Xiaoqin
    Zhou, Xia
    Bouyer, Asgarali
    JOURNAL OF KING SAUD UNIVERSITY-COMPUTER AND INFORMATION SCIENCES, 2024, 36 (08)
  • [33] A convolutional neural network and graph convolutional network-based method for predicting the classification of anatomical therapeutic chemicals
    Zhao, Haochen
    Li, Yaohang
    Wang, Jianxin
    BIOINFORMATICS, 2021, 37 (18) : 2841 - 2847
  • [34] Label Contrastive Coding Based Graph Neural Network for Graph Classification
    Ren, Yuxiang
    Bai, Jiyang
    Zhang, Jiawei
    DATABASE SYSTEMS FOR ADVANCED APPLICATIONS (DASFAA 2021), PT I, 2021, 12681 : 123 - 140
  • [35] Graph convolutional neural network for multi-scale feature learning
    Edwards, Michael
    Xie, Xianghua
    Palmer, Robert, I
    Tam, Gary K. L.
    Alcock, Rob
    Roobottom, Carl
    COMPUTER VISION AND IMAGE UNDERSTANDING, 2020, 194
  • [36] AAGCN: a graph convolutional neural network with adaptive feature and topology learning
    Wang, Bin
    Cai, Bodong
    Sheng, Jinfang
    Jiao, Wenzhe
    SCIENTIFIC REPORTS, 2024, 14 (01):
  • [37] Course Recommendation Based on Graph Convolutional Neural Network
    An Cong Tran
    Duc-Thien Tran
    Nguyen Thai-Nghe
    Tran Thanh Dien
    Hai Thanh Nguyen
    ADVANCES AND TRENDS IN ARTIFICIAL INTELLIGENCE. THEORY AND APPLICATIONS, IEA/AIE 2023, PT I, 2023, 13925 : 235 - 240
  • [38] Malware Classification Based on Graph Convolutional Neural Networks and Static Call Graph Features
    Mester, Attila
    Bodo, Zalan
    ADVANCES AND TRENDS IN ARTIFICIAL INTELLIGENCE: THEORY AND PRACTICES IN ARTIFICIAL INTELLIGENCE, 2022, 13343 : 528 - 539
  • [39] Graph-in-Graph Convolutional Network for Hyperspectral Image Classification
    Jia S.
    Jiang S.
    Zhang S.
    Xu M.
    Jia X.
    IEEE Transactions on Neural Networks and Learning Systems, 2024, 35 (01) : 1157 - 1171
  • [40] A Graph-Based Approach to Automatic Convolutional Neural Network Construction for Image Classification
    Yuan, Gonglin
    Xue, Bing
    Zhang, Mengjie
    2020 35TH INTERNATIONAL CONFERENCE ON IMAGE AND VISION COMPUTING NEW ZEALAND (IVCNZ), 2020,