NeurIPS 2021

DropGNN: Random Dropouts Increase the Expressiveness of Graph Neural Networks

P. Papp, Karolis Martinkus, Lukas Faber, Roger Wattenhofer

Main figure for DropGNN: Random Dropouts Increase the Expressiveness of Graph Neural Networks

Abstract

This paper studies Dropout Graph Neural Networks (DropGNNs), a new approach that aims to overcome the limitations of standard GNN frameworks. In DropGNNs, we execute multiple runs of a GNN on the input graph, with some of the nodes randomly and independently dropped in each of these runs. Then, we combine the results of these runs to obtain the final result. We prove that DropGNNs can distinguish various graph neighborhoods that cannot be separated by message passing GNNs. We derive theoretical bounds for the number of runs required to ensure a reliable distribution of dropouts, and we prove several properties regarding the expressive capabilities and limits of DropGNNs. We experimentally validate our theoretical findings on expressiveness. Furthermore, we show that DropGNNs perform competitively on established GNN benchmarks.

Benchmark Results

Tested on 1 dataset

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

TU Dortmund

#ModelAccuracy ↑Paper
1
MSH-GNN
0.9910 ±0.0030
2
ID-GNN+BEC
0.9830 ±0.0360
3
SupCosine
0.9830 ±0.0250
39
GIN
actual #39 vs paper #237 198 places worse
0.9330 ±0.0290
best reported
actual #39 vs paper #237 198 places worse
paper 0.8940 ±0.0560

Difference within reported uncertainty; not counted as underperformance.

paper 0.8940 ±0.0560 Δ at pub 0.0266 Δ today 0.0390
54
PatchySan
actual #54 vs paper #272 218 places worse
0.9260 ±0.0420
best reported
actual #54 vs paper #272 218 places worse
paper 0.8900 ±0.0440

Difference within reported uncertainty; not counted as underperformance.

paper 0.8900 ±0.0440 Δ at pub 0.0360 Δ today 0.0360
63
GCN
0.9220 ±0.0440
140
PPGN
actual #140 vs paper #140
0.9060 ±0.0870
actual #140 vs paper #140
paper 0.9060 ±0.0870

Improvement within reported uncertainty; not highlighted.

paper 0.9060 ±0.0870 Δ at pub 0.0005
166
DropGNN
0.9040 ±0.0700
This paper
403
MLP
0.8660 ±0.0495
425
DGCNN
actual #425 vs paper #425
0.8583 ±0.0116
best reported
actual #425 vs paper #425
paper 0.8580 ±0.0170

Difference within reported uncertainty; not counted as underperformance.

paper 0.8580 ±0.0170 Δ at pub 0.0003 Δ today 0.0003
SOTA performance (not significantly different from the best) Proposed method (this paper) Baseline / comparison (this paper) Global context (other papers)