Edge Representation Learning with Hypergraphs

by   Jaehyeong Jo, et al.

Graph neural networks have recently achieved remarkable success in representing graph-structured data, with rapid progress in both the node embedding and graph pooling methods. Yet, they mostly focus on capturing information from the nodes considering their connectivity, and not much work has been done in representing the edges, which are essential components of a graph. However, for tasks such as graph reconstruction and generation, as well as graph classification tasks for which the edges are important for discrimination, accurately representing edges of a given graph is crucial to the success of the graph representation learning. To this end, we propose a novel edge representation learning framework based on Dual Hypergraph Transformation (DHT), which transforms the edges of a graph into the nodes of a hypergraph. This dual hypergraph construction allows us to apply message passing techniques for node representations to edges. After obtaining edge representations from the hypergraphs, we then cluster or drop edges to obtain holistic graph-level edge representations. We validate our edge representation learning method with hypergraphs on diverse graph datasets for graph representation and generation performance, on which our method largely outperforms existing graph representation learning methods. Moreover, our edge representation learning and pooling method also largely outperforms state-of-the-art graph pooling methods on graph classification, not only because of its accurate edge representation learning, but also due to its lossless compression of the nodes and removal of irrelevant edges for effective message passing.


Edge but not Least: Cross-View Graph Pooling

Graph neural networks have emerged as a powerful model for graph represe...

Graph Convolutional Networks with Dual Message Passing for Subgraph Isomorphism Counting and Matching

Graph neural networks (GNNs) and message passing neural networks (MPNNs)...

Motif-based Graph Representation Learning with Application to Chemical Molecules

This work considers the task of representation learning on the attribute...

MxPool: Multiplex Pooling for Hierarchical Graph Representation Learning

How to utilize deep learning methods for graph classification tasks has ...

HEAT: Hyperedge Attention Networks

Learning from structured data is a core machine learning task. Commonly,...

A Large-Scale Database for Graph Representation Learning

With the rapid emergence of graph representation learning, the construct...

Message Passing Neural Networks for Hypergraphs

Hypergraph representations are both more efficient and better suited to ...