Higher-Order Interaction Goes Neural: A Substructure Assembling Graph Attention Network for Graph Classification

Jianliang Gao,Jun Gao,Xiaoting Ying,Mingming Lu,Jianxin Wang
DOI: https://doi.org/10.1109/tkde.2021.3105544
IF: 9.235
2021-01-01
IEEE Transactions on Knowledge and Data Engineering
Abstract:Graph classification has been widely used for knowledge discovery in numerous practical application scenarios, such as social networks and protein-protein interaction networks. Recently, Graph Neural Networks (GNNs), which generalize deep neural networks to graph-structured data, have drawn considerable attention and achieved state-of-the-art performance in graph classification. However, existing GNN models mainly focus on capturing the information of immediate or first-order neighboring nodes within a single layer. The graph substructure and substructure interaction, which plays an important role in learning graph representations, are usually overlooked. In this paper, we propose a Substructure Assembling Graph Attention Network (SA-GAT) to extract graph features and improve the performance of graph classification. SA-GAT is able to fully explore higher-order substructure information hidden in graphs by a core module called Substructure Interaction Attention (SIA), which takes both the information of neighbors' substructures and the interaction information among them into account during aggregation process. Theoretically, we have also proved that SA-GAT satisfies the graph isomorphism theory of graph neural network design, which is that the network should map isomorphic graphs to the same representation and output the same prediction. Extensive experimental results on multiple real-world graph classification datasets demonstrate that the proposed SA-GAT outperforms the state-of-the-art methods including graph kernels and graph neural networks.
What problem does this paper attempt to address?