展开

文章目录

修改历史

修改历史记录

  1. 2023-12-25 11:06:10

[IEEE TCDS] Hierarchical Graph Pooling with Self-Adaptive Cluster Aggregation

2023-10-09 11:03:57 科研 365

简介

ZhiPeng Li, HaiLong Su, XiaoBo Zhu, XiuMei Wei, XueSong Jiang, Valeriya Gribova, Vladimir Fedorovich Filaretov, D.S.Huang, “Hierarchical Graph Pooling with Self-Adaptive Cluster Aggregation,” IEEE TCDS , 2021 

      Graph Neural Network (GNN) introduces deep neural networks into graph structure data. It has achieved advanced performance in many fields such as traffic prediction, recommendation systems, and computer vision, which has received extensive attention from the academic community. Most of the existing research on graph neural networks focuses on graph convolution, while graph pooling is usually ignored. Although there are also some graph pooling methods, most of the current pooling methods are based on top-k node selection. In the top-kbased pooling method, unselected nodes will be directly discarded, which will cause the loss of feature information during the pooling process. In this paper, we propose a novel graph pooling operator, called Hierarchical Graph Pooling with Self-Adaptive Cluster Aggregation (HGP-SACA), which uses a sparse and differentiable method to capture the graph structure. Before using top-k for cluster selection, the unselected clusters and the selected clusters in the neighbor perform an n-hop feature information aggregation. The merged clusters which contain neighborhood clusters are used for top-k selection , which can enhance the function of the unselected clusters. Through extensive theoretical analysis and experimental verification on multiple datasets, our experimental results show that combining the existing GNN architecture with HGP-SACA can achieve state-of-the-art results on multiple graph classification benchmarks, which proves the effectiveness of our proposed model.

full paper

 

 

0条评论