Google Research

Graph Partition Neural Networks for Semi-Supervised Classification

  • Alexander Gaunt
  • Danny Tarlow
  • Marc Brockschmidt
  • Raquel Urtasun
  • Renjie Liao
  • Richard Zemel
ICLR (2018) (to appear)


We present graph partition neural networks (GPNN), an extension of graph neu- ral 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 partition and also propose a modified multi- seed flood fill for fast processing large scale graphs. We extensively test our model on a variety of semi-supervised node classification tasks. First, we show that GPNNs can achieve similar performance as standard GNNs with fewer propaga- tion steps. Secondly, experimental results indicate that GPNNs results are compa- rable to a wide selection of baselines on medium-sized datasets and superior on a very large dataset.

Research Areas

Learn more about how we do research

We maintain a portfolio of research projects, providing individuals and teams the freedom to emphasize specific types of work