ICLR 2020

Combining Label Propagation and Simple Models out-performs Graph Neural Networks

Qian Huang, Horace He, Abhay Singh, Ser-Nam Lim, Austin R. Benson

Main figure for Combining Label Propagation and Simple Models out-performs Graph Neural Networks

Abstract

Graph Neural Networks (GNNs) are the predominant technique for learning over graphs. However, there is relatively little understanding of why GNNs are successful in practice and whether they are necessary for good performance. Here, we show that for many standard transductive node classification benchmarks, we can exceed or match the performance of state-of-the-art GNNs by combining shallow models that ignore the graph structure with two simple post-processing steps that exploit correlation in the label structure: (i) an ``error correlation'' that spreads residual errors in training data to correct errors in test data and (ii) a ``prediction correlation'' that smooths the predictions on the test data. We call this overall procedure Correct and Smooth (C&S), and the post-processing steps are implemented via simple modifications to standard label propagation techniques from early graph-based semi-supervised learning methods. Our approach exceeds or nearly matches the performance of state-of-the-art GNNs on a wide variety of benchmarks, with just a small fraction of the parameters and orders of magnitude faster runtime. For instance, we exceed the best known GNN performance on the OGB-Products dataset with 137 times fewer parameters and greater than 100 times less training time. The performance of our methods highlights how directly incorporating label information into the learning algorithm (as was done in traditional techniques) yields easy and substantial performance gains. We can also incorporate our techniques into big GNN models, providing modest gains. Our code for the OGB results is at.

Benchmark Results

Tested on 1 dataset

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

OGB

#ModelAccuracy ↑Paper
1
Jacobi
0.8968
2
GPR
0.8953
3
LEMP4HG
0.8939
4
SAGE
0.8933
53
FDiff-scale (MLP)
0.8455
This paper
58
MLP+C&S
actual #58 vs paper #138 80 places worse
0.8418 ±0.0007
best reported
actual #58 vs paper #138 80 places worse
paper 0.8034
paper 0.8034 Δ today 0.0384
86
SoTA
0.8256
This paper
94
SGC
actual #94 vs paper #342 248 places worse
0.8220 ±0.0090
best reported
actual #94 vs paper #342 248 places worse
paper 0.6883
paper 0.6883 Δ today 0.1337
136
Autoscale (Linear)
0.8056
This paper
199
Linear
0.7873
This paper
252
Plain GCN
0.7564
This paper
279
LP
0.7476
This paper
350
Plain Linear
0.6627
This paper
351
MLP
0.6605 ±0.0020
SOTA performance (not significantly different from the best) Baseline / comparison (this paper) Global context (other papers) Best today — OK at publication; today's leaderboard best is elsewhere