Graph Coarsening with Neural Networks

by   Chen Cai, et al.

As large-scale graphs become increasingly more prevalent, it poses significant computational challenges to process, extract and analyze large graph data. Graph coarsening is one popular technique to reduce the size of a graph while maintaining essential properties. Despite rich graph coarsening literature, there is only limited exploration of data-driven methods in the field. In this work, we leverage the recent progress of deep learning on graphs for graph coarsening. We first propose a framework for measuring the quality of coarsening algorithm and show that depending on the goal, we need to carefully choose the Laplace operator on the coarse graph and associated projection/lift operators. Motivated by the observation that the current choice of edge weight for the coarse graph may be sub-optimal, we parametrize the weight assignment map with graph neural networks and train it to improve the coarsening quality in an unsupervised way. Through extensive experiments on both synthetic and real networks, we demonstrate that our method significantly improves common graph coarsening methods under various metrics, reduction ratios, graph sizes, and graph types. It generalizes to graphs of larger size (25× of training graphs), is adaptive to different losses (differentiable and non-differentiable), and scales to much larger graphs than previous work.



There are no comments yet.


page 1

page 2

page 3

page 4


Optimization-Based Algebraic Multigrid Coarsening Using Reinforcement Learning

Large sparse linear systems of equations are ubiquitous in science and e...

Graph reduction by local variation

How can we reduce the size of a graph without significantly altering its...

Understanding Pooling in Graph Neural Networks

Inspired by the conventional pooling layers in convolutional neural netw...

Machine Learning on Graphs: A Model and Comprehensive Taxonomy

There has been a surge of recent interest in learning representations fo...

Graph Neural Networks in TensorFlow and Keras with Spektral

In this paper we present Spektral, an open-source Python library for bui...

LSP : Acceleration and Regularization of Graph Neural Networks via Locality Sensitive Pruning of Graphs

Graph Neural Networks (GNNs) have emerged as highly successful tools for...

SparRL: Graph Sparsification via Deep Reinforcement Learning

Graph sparsification concerns data reduction where an edge-reduced graph...
This week in AI

Get the week's most popular data science and artificial intelligence research sent straight to your inbox every Saturday.