ICLR 2018
Skip to yearly menu bar Skip to main content


Workshop

Graph Partition Neural Networks for Semi-Supervised Classification

Renjie Liao · Marc Brockschmidt · Danny Tarlow · Alexander Gaunt · Raquel Urtasun · Richard Zemel

East Meeting Level 8 + 15 #20

We present graph partition neural networks (GPNN), an extension of graph neural networks (GNNs) able to handle extremely large graphs. GPNNs alternate between locally propagating information between nodes in small subgraphs and globally propagating information between the subgraphs. To efficiently partition graphs, we experiment with spectral partitioning and also propose a modified multi-seed flood fill for fast processing of large scale graphs. We extensively test our model on a variety of semi-supervised node classification tasks. Experimental results indicate that GPNNs are either superior or comparable to state-of-the-art methods on a wide variety of datasets for graph-based semi-supervised classification. We also show that GPNNs can achieve similar performance as standard GNNs with fewer propagation steps.

Live content is unavailable. Log in and register to view live content