DeepAI AI Chat
Log In Sign Up

Hierarchical Graph Pooling with Structure Learning

by   Zhen Zhang, et al.
Zhejiang University
Simon Fraser University
Alibaba Group

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 numerous graph related tasks. However, existing GNN models mainly focus on designing graph convolution operations. The graph pooling (or downsampling) operations, that play an important role in learning hierarchical representations, are usually overlooked. In this paper, we propose a novel graph pooling operator, called Hierarchical Graph Pooling with Structure Learning (HGP-SL), which can be integrated into various graph neural network architectures. HGP-SL incorporates graph pooling and structure learning into a unified module to generate hierarchical representations of graphs. More specifically, the graph pooling operation adaptively selects a subset of nodes to form an induced subgraph for the subsequent layers. To preserve the integrity of graph's topological information, we further introduce a structure learning mechanism to learn a refined graph structure for the pooled graph at each layer. By combining HGP-SL operator with graph neural networks, we perform graph level representation learning with focus on graph classification task. Experimental results on six widely used benchmarks demonstrate the effectiveness of our proposed model.


page 1

page 2

page 3

page 4


Graph Convolutional Networks with EigenPooling

Graph neural networks, which generalize deep neural network models to gr...

iPool -- Information-based Pooling in Hierarchical Graph Neural Networks

With the advent of data science, the analysis of network or graph data h...

CommPOOL: An Interpretable Graph Pooling Framework for Hierarchical Graph Representation Learning

Recent years have witnessed the emergence and flourishing of hierarchica...

Distribution Knowledge Embedding for Graph Pooling

Graph-level representation learning is the pivotal step for downstream t...

Hierarchical Graph Capsule Network

Graph Neural Networks (GNNs) draw their strength from explicitly modelin...

NervePool: A Simplicial Pooling Layer

For deep learning problems on graph-structured data, pooling layers are ...

Adversarial Attack on Hierarchical Graph Pooling Neural Networks

Recent years have witnessed the emergence and development of graph neura...

Code Repositories


Hierarchical Graph Pooling with Structure Learning(AAAI-2020)

view repo