Abstract:A pooling operation is essential for effective graph-level representation learning, where the node drop pooling has become one mainstream graph pooling technology. However, current node drop pooling methods usually keep the top-k nodes according to their significance scores, which ignore the graph diversity in terms of the node features and the graph structures, thus resulting in suboptimal graph-level representations. To address the aforementioned issue, we propose a novel plug-and-play score scheme and refer to it as MID, which consists of a \textbf{M}ultidimensional score space with two operations, \textit{i.e.}, fl\textbf{I}pscore and \textbf{D}ropscore. Specifically, the multidimensional score space depicts the significance of nodes through multiple criteria; the flipscore encourages the maintenance of dissimilar node features; and the dropscore forces the model to notice diverse graph structures instead of being stuck in significant local structures. To evaluate the effectiveness of our proposed MID, we perform extensive experiments by applying it to a wide variety of recent node drop pooling methods, including TopKPool, SAGPool, GSAPool, and ASAP. Specifically, the proposed MID can efficiently and consistently achieve about 2.8\% average improvements over the above four methods on seventeen real-world graph classification datasets, including four social datasets (IMDB-BINARY, IMDB-MULTI, REDDIT-BINARY, and COLLAB), and thirteen biochemical datasets (D\&D, PROTEINS, NCI1, MUTAG, PTC-MR, NCI109, ENZYMES, MUTAGENICITY, FRANKENSTEIN, HIV, BBBP, TOXCAST, and TOX21). Code is available at~\url{<a class="link-external link-https" href="https://github.com/whuchuang/mid" rel="external noopener nofollow">this https URL</a>}.

C2N-ABDP: Cluster-to-Node Attention-Based Differentiable Pooling.

ABDPool: Attention-based Differentiable Pooling.

ENADPool: The Edge-Node Attention-based Differentiable Pooling for Graph Neural Networks

Hierarchical Graph Representation Learning with Differentiable Pooling

Hierarchical Graph Pooling with Self-Adaptive Cluster Aggregation

Node Information Awareness Pooling for Graph Representation Learning

Multistructure Graph Classification Method with Attention-Based Pooling

Hierarchical Triplet Attention Pooling for Graph Classification.

GSAPool: Gated Structure Aware Pooling for Graph Representation Learning.

Multiscale Representation Learning of Graph Data with Node Affinity.

DIPool: Degree-Induced Pooling for Hierarchical Graph Representation Learning.

Hierarchical Adaptive Pooling by Capturing High-order Dependency for Graph Representation Learning

DMSPool: Dual Multi-Scale Pooling for Graph Representation Learning

Diversified Node Sampling based Hierarchical Transformer Pooling for Graph Representation Learning

Graph Multihead Attention Pooling with Self-Supervised Learning.

Ipool - Information-based Pooling in Hierarchical Graph Neural Networks.

AttPool: Towards Hierarchical Feature Representation in Graph Convolutional Networks Via Attention Mechanism

Seeing All From a Few: Nodes Selection Using Graph Pooling for Graph Clustering

On Exploring Node-feature and Graph-structure Diversities for Node Drop Graph Pooling

Distribution Knowledge Embedding for Graph Pooling

Not All Edges Are Peers: Accurate Structure-Aware Graph Pooling Networks