返回
Graph explicit pooling for graph-level representation learning
DOI:10.1016/j.neunet.2024.106790.png)
摘要
En 中文
Graph pooling has been increasingly recognized as crucial for Graph Neural Networks (GNNs) to facilitate hierarchical graph representation learning. Existing graph pooling methods commonly consist of two stages: selecting top-ranked nodes and discarding the remaining to construct coarsened graph representations. However, this paper highlights two key issues with these methods: (1) The process of selecting nodes to discard frequently employs additional Graph Convolutional Networks or Multilayer Perceptrons, lacking a thorough evaluation of each node's impact on the final graph representation and subsequent prediction tasks. (2) Current graph pooling methods tend to directly discard the noise segment (dropped) of the graph without accounting for the latent information contained within these elements. To address the first issue, we introduce a novel Graph explicit Pool ing (GrePool) method, which selects nodes by explicitly leveraging the relationships between the nodes and final representation vectors crucial for classification. The second issue is addressed using an extended version of GrePool ( i.e. , GrePool+), which applies a uniform loss on the discarded nodes. This addition is designed to augment the training process and improve classification accuracy. Furthermore, we conduct comprehensive experiments across 12 widely used datasets to validate our proposed method's effectiveness, including the Open Graph Benchmark datasets. Our experimental results uniformly demonstrate that GrePool outperforms 14 baseline methods for most datasets. Likewise, implementing GrePool+ enhances GrePool's performance without incurring additional computational costs. The code is available at https: //github.com/LiuChuang0059/GrePool.
Keyword:
Graph pooling
Graph neural networks
Graph classification
Node classification
期刊
IF:
6.3
论文数:
8.2K
被引数:
3.0W
机构
引用论文
CommPOOL: An interpretable graph pooling framework for hierarchical graph representation learningCommPOOL: 用于分层图表示学习的可解释图池框架
NEURAL NETWORKS
IF6.3
Multivariate time-series classification with hierarchical variational graph pooling
NEURAL NETWORKS
IF6.3
Enhancing Graph Neural Networks by a High-quality Aggregation of Beneficial Information
NEURAL NETWORKS
IF6.3

