ICLR 2018

Predict then Propagate: Graph Neural Networks meet Personalized PageRank

Johannes Klicpera, Aleksandar Bojchevski, Stephan Günnemann

Main figure for Predict then Propagate: Graph Neural Networks meet Personalized PageRank

Abstract

Neural message passing algorithms for semi-supervised classification on graphs have recently achieved great success. However, for classifying a node these methods only consider nodes that are a few propagation steps away and the size of this utilized neighborhood is hard to extend. In this paper, we use the relationship between graph convolutional networks (GCN) and PageRank to derive an improved propagation scheme based on personalized PageRank. We utilize this propagation procedure to construct a simple model, personalized propagation of neural predictions (PPNP), and its fast approximation, APPNP. Our model's training time is on par or faster and its number of parameters on par or lower than previous models. It leverages a large, adjustable neighborhood for classification and can be easily combined with any neural network. We show that this model outperforms several recently proposed methods for semi-supervised classification in the most thorough study done so far for GCN-like models. Our implementation is available online.

Proposed method: APPNP

Benchmark Results

Tested on 1 dataset 1 baseline value reported too low

Tables show presented results in global context (best 3 papers + GCN + MLP).

Classic

PubMed

Global leaderboard →

Default metric is Accuracy. The paper reports 7 values on Accuracy and 1 value on F1 .

#ModelAccuracy ↑ defaultF1 ↑Paper
1
LGTL
0.9518 ± 0.0011
—
2
MixHop (LO)
0.9513 ± 0.0038
—
3
GraphSAGE (LO)
0.9511 ± 0.0036
—
17
APPNP
0.9355 ±0.0060
best reported
0.7966 ± 0.0031
F1 0.7966 ± 0.0031
paper 0.7973 ±0.0031
paper 0.7973 ±0.0031 Δ today 0.1382
18
GCN
0.9296 ±0.0015
best reported
—
F1 —
paper 0.7868 ±0.0038
paper 0.7868 ±0.0038 Δ at pub 0.0032 Δ today 0.1428
20
GAT
0.9233 ±0.0018
best reported
—
F1 —
paper 0.7776 ±0.0044
paper 0.7776 ±0.0044 Δ at pub 0.0124 Δ today 0.1457
116
SAGE
0.8881 ± 0.0015
—
210
MLP
0.8690 ± 0.0074
—
694
JK
0.7788 ± 0.0038
—
F1 —
This paper
706
V. GCN
0.7765 ± 0.0040
—
F1 —
This paper
716
N-GCN
0.7743 ± 0.0042
—
F1 —
This paper
819
Bt. FP
0.7294 ± 0.0100
—
F1 —
This paper
SOTA performance (not significantly different from the best) Baseline / comparison (this paper) Global context (other papers) Best at pub — reported result worse than the best known when published (best reported in green) Best today — OK at publication; today's leaderboard best is elsewhere