Google Research

Deep Graph Infomax

  • Petar Veličković
  • Wiliam Fedus
  • William L. Hamilton
  • Pietro Liò
  • Yoshua Bengio
  • R Devon Hjelm
International Conference on Learning Representations (2019)

Abstract

We present Deep Graph Infomax (DGI), a general approach for learning node representations within graph-structured data in an unsupervised manner. DGI relies on maximizing mutual information between patch representations and corresponding high-level summaries of graphs—both derived using established graph convolutional network architectures. In contrast to most prior approaches to graph representation learning, DGI does not rely on random walks, and is readily applicable to both transductive and inductive downstream tasks. We demonstrate competitive performance on a variety of node classification benchmarks, which at times even exceeds the performance of supervised learning.

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