Ni Ruizhi, Wang Yongping, Zhang Xiaolin, Ye Jinhui, Tao Xueqing. GRAPH CLASSIFICATION NETWORK BASED ON INDUCTIVE LEARNING GRAPH CONVOLUTION AND SELF-ATTENTION POOLINGJ. Computer Applications and Software, 2024, 41(10): 177-183. DOI: 10.3969/j.issn.1000-386x.2024.10.027
Citation: Ni Ruizhi, Wang Yongping, Zhang Xiaolin, Ye Jinhui, Tao Xueqing. GRAPH CLASSIFICATION NETWORK BASED ON INDUCTIVE LEARNING GRAPH CONVOLUTION AND SELF-ATTENTION POOLINGJ. Computer Applications and Software, 2024, 41(10): 177-183. DOI: 10.3969/j.issn.1000-386x.2024.10.027

GRAPH CLASSIFICATION NETWORK BASED ON INDUCTIVE LEARNING GRAPH CONVOLUTION AND SELF-ATTENTION POOLING

  • In view of the poor classification performance of graph neural network on large-scale graphs, it is unable to quickly form the embedding of unknown nodes and edges, and it is easy to lose the important features of the graph. A graph classification network model based on inductive learning and self-attention pooling is proposed. On the one hand, the inductive learning method after the improved aggregation function was used to quickly embed the node features of the graph. On the other hand, the self-attention pooling method was used to retain the important features of the graph. A hierarchical framework suitable for extracting large-scale graph information was used for downstream graph classification task. The experimental results show that the accuracy of this method is about 2%~10% higher than that of other graph classification models under the same common data set.
  • loading

Catalog

    Turn off MathJax
    Article Contents

    /

    DownLoad:  Full-Size Img  PowerPoint
    Return
    Return