about
Deep Graph Infomax (arxiv.org)
3 points by jonbaer on Oct 1, 2018 | hide | past | pdf | discuss on HN

In plain words: It learns node summaries without labels by training a graph network so each node's local patch matches a whole-graph summary, unlike usual approaches that walk randomly through the graph. On node classification tasks it matched or sometimes beat fully supervised training.

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. The learnt patch representations summarize subgraphs centered around nodes of interest, and can thus be reused for downstream node-wise learning tasks. In contrast to most prior approaches to unsupervised learning with GCNs, DGI does not rely on random walk objectives, and is readily applicable to both transductive and inductive learning setups. We demonstrate competitive performance on a variety of node classification benchmarks, which at times even exceeds the performance of supervised learning.

Petar Veličković, William Fedus, William L. Hamilton, Pietro Liò, Yoshua Bengio, R Devon Hjelm
arXiv:1809.10341 · stat.ML, cs.IT, cs.LG, cs.SI · submitted Sep 27, 2018 · updated Dec 21, 2018
abstract · pdf · html · To appear at ICLR 2019. 17 pages, 8 figures

add comment on HN