Hierarchical Graph Representation Learning withDifferentiable Pooling

06/22/2018
by   Rex Ying, et al.
0

Recently, graph neural networks (GNNs) have revolutionized the field of graph representation learning through effectively learned node embeddings, and achieved state-of-the-art results in tasks such as node classification and link prediction. However, current GNN methods are inherently flat and do not learn hierarchical representations of graphs---a limitation that is especially problematic for the task of graph classification, where the goal is to predict the label associated with an entire graph. Here we propose DiffPool, a differentiable graph pooling module that can generate hierarchical representations of graphs and can be combined with various graph neural network architectures in an end-to-end fashion. DiffPool learns a differentiable soft cluster assignment for nodes at each layer of a deep GNN, mapping nodes to a set of clusters, which then form the coarsened input for the next GNN layer. Our experimental results show that combining existing GNN methods with DiffPool yields an average improvement of 5-10 benchmarks, compared to all existing pooling approaches, achieving a new state-of-the-art on four out of five benchmark data sets.

READ FULL TEXT
research
06/22/2018

Hierarchical Graph Representation Learning with Differentiable Pooling

Recently, graph neural networks (GNNs) have revolutionized the field of ...
research
11/18/2019

ASAP: Adaptive Structure Aware Pooling for Learning Hierarchical Graph Representations

Graph Neural Networks (GNN) have been shown to work effectively for mode...
research
10/24/2019

Hierarchical Representation Learning in Graph Neural Networks with Node Decimation Pooling

In graph neural networks (GNNs), pooling operators compute local summari...
research
02/21/2020

Memory-Based Graph Networks

Graph neural networks (GNNs) are a class of deep models that operate on ...
research
02/26/2019

EvolveGCN: Evolving Graph Convolutional Networks for Dynamic Graphs

Graph representation learning resurges as a trending research subject ow...
research
06/11/2022

Soft-mask: Adaptive Substructure Extractions for Graph Neural Networks

For learning graph representations, not all detailed structures within a...
research
04/25/2022

ClusterGNN: Cluster-based Coarse-to-Fine Graph Neural Network for Efficient Feature Matching

Graph Neural Networks (GNNs) with attention have been successfully appli...

Please sign up or login with your details

Forgot password? Click here to reset