TMLR 2021

Calibrating and Improving Graph Contrastive Learning

Kaili Ma, Haochen Yang, Han Yang, Tatiana Jin, Pengfei Chen, Yongqiang Chen, Barakeel Fanseu Kamhoua, James Cheng

Main figure for Calibrating and Improving Graph Contrastive Learning

Abstract

Graph contrastive learning algorithms have demonstrated remarkable success in various applications such as node classification, link prediction, and graph clustering.

Benchmark Results

Tested on 5 datasets 10 baseline values reported too low

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

Classic

#ModelAccuracy ↑Paper
1
LGTL
0.9518 ±0.0011
2
MixHop (LO)
0.9513 ±0.0038
3
GraphSAGE (LO)
0.9511 ±0.0036
18
GCN
actual #18 vs paper #552 534 places worse
0.9296 ±0.0015
best reported
actual #18 vs paper #552 534 places worse
paper 0.7926 ±0.0038
116
SAGE
0.8881 ±0.0015
195
GMI
actual #195 vs paper #682 487 places worse
0.8726 ±0.0016
best reported
actual #195 vs paper #682 487 places worse
paper 0.7797 ±0.0104
paper 0.7797 ±0.0104 Δ at pub 0.0213 Δ today 0.0929
204
DGI
actual #204 vs paper #743 539 places worse
0.8708 ±0.0017
best reported
actual #204 vs paper #743 539 places worse
paper 0.7689 ±0.0053
210
MLP
0.8690 ±0.0074
251
Node2Vec
actual #251 vs paper #858 607 places worse
0.8520 ±0.0002
best reported
actual #251 vs paper #858 607 places worse
paper 0.6634 ±0.0140
#ModelAccuracy ↑Paper
1
GraphTARIF
0.9703 ±0.0019
2
GAT
0.9669 ±0.0014
3
GAT+T
0.9660 ±0.0007
7
SAGE
0.9643 ±0.0027
8
GCN
actual #8 vs paper #479 471 places worse
0.9632 ±0.0008
best reported
actual #8 vs paper #479 471 places worse
paper 0.8875 ±0.0199
paper 0.8875 ±0.0199 Δ at pub 0.0367 Δ today 0.0757
164
GraphMAE
actual #164 vs paper #415 251 places worse
0.9341 ±0.0010
best reported
actual #164 vs paper #415 251 places worse
paper 0.9071 ±0.0129
paper 0.9071 ±0.0129 Δ today 0.0270
191
GCA
actual #191 vs paper #202 11 places worse
0.9320 ±0.0060
best reported
actual #191 vs paper #202 11 places worse
paper 0.9313 ±0.0032

Difference within reported uncertainty; not counted as underperformance.

paper 0.9313 ±0.0032 Δ today 0.0007
213
DGI
actual #213 vs paper #544 331 places worse
0.9309 ±0.0008
best reported
actual #213 vs paper #544 331 places worse
paper 0.7719 ±0.0260
390
MLP
0.9128 ±0.0063
408
GMI
actual #408 vs paper #497 89 places worse
0.9081 ±0.0015
best reported
actual #408 vs paper #497 89 places worse
paper 0.8708 ±0.0123
457
Node2Vec
actual #457 vs paper #521 64 places worse
0.8970 ±0.0010
best reported
actual #457 vs paper #521 64 places worse
paper 0.8363 ±0.0153
paper 0.8363 ±0.0153 Δ at pub 0.0604 Δ today 0.0607
494
LC
0.8759 ±0.0150
This paper
499
ML
0.8678 ±0.0170
This paper
#ModelAccuracy ↑Paper
1
GraphTARIF
0.9461 ±0.0017
2
GCN
actual #2 vs paper #430 428 places worse
0.9412 ±0.0008
best reported
actual #2 vs paper #430 428 places worse
paper 0.7982 ±0.0204
paper 0.7982 ±0.0204 Δ at pub 0.0803 Δ today 0.1430
3
GAT
0.9398 ±0.0022
8
SAGE
0.9351 ±0.0006
100
GraphMAE
actual #100 vs paper #417 317 places worse
0.9002 ±0.0024
best reported
actual #100 vs paper #417 317 places worse
paper 0.8116 ±0.0110
paper 0.8116 ±0.0110 Δ today 0.0886
160
GCA
actual #160 vs paper #199 39 places worse
0.8894 ±0.0015
best reported
actual #160 vs paper #199 39 places worse
paper 0.8821 ±0.0037
paper 0.8821 ±0.0037 Δ today 0.0073
210
DGI
actual #210 vs paper #479 269 places worse
0.8798 ±0.0081
best reported
actual #210 vs paper #479 269 places worse
paper 0.6492 ±0.0193
paper 0.6492 ±0.0193 Δ at pub 0.1903 Δ today 0.2306
320
MLP
0.8463
330
Node2Vec
actual #330 vs paper #451 121 places worse
0.8440 ±0.0010
best reported
actual #330 vs paper #451 121 places worse
paper 0.7537 ±0.0152
paper 0.7537 ±0.0152 Δ at pub 0.0902 Δ today 0.0903
342
GMI
actual #342 vs paper #432 90 places worse
0.8422 ±0.0052
best reported
actual #342 vs paper #432 90 places worse
paper 0.7927 ±0.0164
paper 0.7927 ±0.0164 Δ at pub 0.0294 Δ today 0.0495
400
ML
0.8211 ±0.0147
This paper
407
LC
0.8198 ±0.0152
This paper

Reddit

Global leaderboard →

Default metric is F1. The paper reports 4 values on F1 and 2 values on Accuracy .

#ModelF1 ↑ defaultAccuracy ↑Paper
1
EnGCN
0.9714 ± 0.0003
—
2
SAGN+SLE
0.9710 ± 0.0000
—
3
GAMLP(JK)
0.9704 ± 0.0001
—
14
GraphSAGE
0.9660
—
58
GCN
0.9648
best reported
0.9402 ± 0.0005
Accuracy 0.9402 ± 0.0005
paper 0.9402 ±0.0005
paper 0.9402 ±0.0005 Δ at pub 0.0246
70
GMI
0.9500
best reported
—
Accuracy —
paper 0.9419 ±0.0004
77
LC
0.9443 ± 0.0003
—
Accuracy —
This paper
78
ML
0.9438 ± 0.0004
—
Accuracy —
This paper
82
DGI
0.9414 ±0.0003
—
Accuracy —
paper 0.9414 ±0.0003

Improvement within reported uncertainty; not highlighted.

paper 0.9414 ±0.0003 Δ at pub 0.0014
145
Node2Vec
0.9326 ±0.0004
0.9326 ± 0.0004
Accuracy 0.9326 ± 0.0004
paper 0.9326 ±0.0004
paper 0.9326 ±0.0004 Δ at pub 0.4969

OGB

#ModelAccuracy ↑Paper
1
Jacobi
0.8968
2
GPR
0.8953
3
LEMP4HG
0.8939
4
SAGE
0.8933
6
GCN
actual #6 vs paper #250 244 places worse
0.8868
best reported
actual #6 vs paper #250 244 places worse
paper 0.7564 ±0.0021
23
LC
0.8645 ±0.0013
This paper
26
DGI
0.8637 ±0.0019
This paper
30
GraphMAE
actual #30 vs paper #324 294 places worse
0.8630 ±0.0007
best reported
actual #30 vs paper #324 294 places worse
paper 0.7156 ±0.0019
paper 0.7156 ±0.0019 Δ today 0.1474
31
ML
0.8628 ±0.0020
This paper
33
GMI
0.8614 ±0.0016
This paper
48
No pretraining
0.8469 ±0.0079
This paper
313
Node2Vec
0.7249 ±0.0010
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 at pub — reported result worse than the best known when published (best reported in green) Best at pub — reported value improved over the best known when published Best today — OK at publication; today's leaderboard best is elsewhere