Deep Gaussian Embedding of Graphs: Unsupervised Inductive Learning via Ranking

By Aleksandar Bojchevski, Stephan Günnemann. 2017

780 citations recorded by the source corpus.

Methods that learn representations of nodes in a graph play a critical role in network analysis since they enable many downstream learning tasks. We propose Graph2Gauss - an approach that can efficiently learn versatile node embeddings on large scale (attributed) graphs that show strong performance on tasks such as link prediction and node classification. Unlike most approaches that represent nodes as point vectors in a low-dimensional continuous space, we embed each node as a Gaussian distribution, allowing us to capture uncertainty about the representation. Furthermore, we propose an unsupervised method that handles inductive learning scenarios and is applicable to different types of graphs: plain/attributed, directed/undirected. By leveraging both the network structure and the associated node attributes, we are able to generalize to unseen nodes without additional training. To learn the embeddings we adopt a personalized ranking formulation w.r.t. the node distances that exploits the natural ordering of the nodes imposed by the network structure. Experiments on real world networks demonstrate the high performance of our approach, outperforming state-of-the-art network embedding methods on several different tasks. Additionally, we demonstrate the benefits of modeling uncertainty - by analyzing it we can estimate neighborhood diversity and detect the intrinsic latent dimensionality of a graph.

Subfields: Artificial Intelligence, Statistical and Nonlinear Physics, Representation & Self-Supervised Learning, Graph Learning

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. William L. Hamilton
  2. Stephan Günnemann
  3. Yoshua Bengio
  4. Jure Leskovec
  5. Philip S. Yu
  6. Jie Tang
  7. Zhiyuan Liu
  8. Aleksandar Bojchevski

In Statistical and Nonlinear Physics

  1. Stephan Günnemann
  2. William L. Hamilton
  3. Bryan Perozzi
  4. Daniel Zügner
  5. Aleksandar Bojchevski
  6. Benedek Rozemberczki
  7. Yoshua Bengio
  8. Jian Tang

In Representation & Self-Supervised Learning

  1. Stephan Günnemann
  2. Jian Tang
  3. William L. Hamilton
  4. Aleksandar Bojchevski
  5. Yoshua Bengio
  6. Bryan Perozzi
  7. Zhiyuan Liu
  8. Peng Cui

In Graph Learning

  1. Stephan Günnemann
  2. William L. Hamilton
  3. Aleksandar Bojchevski
  4. Jure Leskovec
  5. Jian Tang
  6. Peng Cui
  7. Johannes Klicpera
  8. Yoshua Bengio

Citations in the corpus