Asynchronous Neural Networks for Learning in Graphs

Lukas Faber, Roger Wattenhofer

Main figure for Asynchronous Neural Networks for Learning in Graphs

Abstract

This paper studies asynchronous message passing (AMP), a new paradigm for applying neural network based learning to graphs. Existing graph neural networks use the synchronous distributed computing model and aggregate their neighbors in each round, which causes problems such as oversmoothing and limits their expressiveness. On the other hand, AMP is based on the asynchronous model, where nodes react to messages of their neighbors individually. We prove that (i) AMP can simulate synchronous GNNs and that (ii) AMP can theoretically distinguish any pair of graphs. We experimentally validate AMP's expressiveness. Further, we show that AMP might be better suited to propagate messages over large distances in graphs and performs well on several graph classification 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.0390
54
PatchySan
0.9260 ±0.0420
This paper
63
GCN
actual #63 vs paper #280 217 places worse
0.9220 ±0.0440
best reported
actual #63 vs paper #280 217 places worse
paper 0.8890 ±0.0760

Difference within reported uncertainty; not counted as underperformance.

paper 0.8890 ±0.0760 Δ at pub 0.0020 Δ today 0.0330
73
ESAN
0.9200 ±0.0500
This paper
140
PPGN
0.9060 ±0.0870
This paper
158
GraphSAGE
actual #158 vs paper #158
0.9040 ±0.0780
actual #158 vs paper #158
paper 0.9040 ±0.0780

Improvement within reported uncertainty; not highlighted.

paper 0.9040 ±0.0780 Δ at pub 0.0470
170
AMP-RNN
0.9040 ±0.0410
This paper
166
DropGNN
0.9040 ±0.0700
This paper
183
GAT
actual #183 vs paper #446 263 places worse
0.9010 ±0.0580
best reported
actual #183 vs paper #446 263 places worse
paper 0.8510 ±0.0930

Difference within reported uncertainty; not counted as underperformance.

paper 0.8510 ±0.0930 Δ at pub 0.0500 Δ today 0.0500
294
1-2-3 GNN
actual #294 vs paper #417 123 places worse
0.8880 ±0.0700
best reported
actual #294 vs paper #417 123 places worse
paper 0.8610
best at pub: Provably Powerful Graph Networks (NeurIPS 2019)
best today: Agent-based Graph Neural Networks (ICLR 2022)

Difference within reported uncertainty; not counted as underperformance.

paper 0.8610 Δ today 0.0270
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
718
SAGE
0.7490 ±0.0870
SOTA performance (not significantly different from the best) Baseline / comparison (this paper) Global context (other papers)