Relational Self-Supervised Learning on Graphs

Namkyeong Lee, Dongmin Hyun, Junseok Lee, Chanyoung Park

Main figure for Relational Self-Supervised Learning on Graphs

Abstract

Over the past few years, graph representation learning (GRL) has been a powerful strategy for analyzing graph-structured data. Recently, GRL methods have shown promising results by adopting self-supervised learning methods developed for learning representations of images. Despite their success, existing GRL methods tend to overlook an inherent distinction between images and graphs, i.e., images are assumed to be independently and identically distributed, whereas graphs exhibit relational information among data instances, i.e., nodes. To fully benefit from the relational information inherent in the graph-structured data, we propose a novel GRL method, called RGRL, that learns from the relational information generated from the graph itself. RGRL learns node representations such that the relationship among nodes is invariant to augmentations, i.e., augmentation-invariant relationship, which allows the node representations to vary as long as the relationship among the nodes is preserved. By considering the relationship among nodes in both global and local perspectives, RGRL overcomes limitations of previous contrastive and non-contrastive methods, and achieves the best of both worlds. Extensive experiments on fourteen benchmark datasets over various downstream tasks demonstrate the superiority of RGRL over state-of-the-art baselines. The source code for RGRL is available at.

Proposed method: RGRL

Benchmark Results

Tested on 8 datasets 13 baseline values reported too low

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

Classic

#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 #292 284 places worse
0.9632 ±0.0008
best reported
actual #8 vs paper #292 284 places worse
paper 0.9242 ±0.0022
160
RGRL
0.9343 ±0.0031
This paper
172
BGRL
actual #172 vs paper #215 43 places worse
0.9337 ±0.0021
best reported
actual #172 vs paper #215 43 places worse
paper 0.9306 ±0.0030

Difference within reported uncertainty; not counted as underperformance.

paper 0.9306 ±0.0030 Δ at pub 0.0031
186
CCA-SSG
actual #186 vs paper #272 86 places worse
0.9325 ±0.0021
best reported
actual #186 vs paper #272 86 places worse
paper 0.9259 ±0.0025
paper 0.9259 ±0.0025 Δ at pub 0.0066
191
GCA
actual #191 vs paper #222 31 places worse
0.9320 ±0.0060
best reported
actual #191 vs paper #222 31 places worse
paper 0.9299 ±0.0027

Difference within reported uncertainty; not counted as underperformance.

paper 0.9299 ±0.0027 Δ at pub 0.0014 Δ today 0.0021
203
GRACE
actual #203 vs paper #279 76 places worse
0.9313 ±0.0017
best reported
actual #203 vs paper #279 76 places worse
paper 0.9252 ±0.0032
paper 0.9252 ±0.0032 Δ at pub 0.0028 Δ today 0.0061
206
MVGRL
actual #206 vs paper #359 153 places worse
0.9310 ±0.0040
best reported
actual #206 vs paper #359 153 places worse
paper 0.9174 ±0.0007
213
DGI
actual #213 vs paper #366 153 places worse
0.9309 ±0.0008
best reported
actual #213 vs paper #366 153 places worse
paper 0.9161 ±0.0022
390
MLP
0.9128 ±0.0063
408
GMI
actual #408 vs paper #417 9 places worse
0.9081 ±0.0015
best reported
actual #408 vs paper #417 9 places worse
paper 0.9068 ±0.0017

Difference within reported uncertainty; not counted as underperformance.

paper 0.9068 ±0.0017 Δ at pub 0.0002 Δ today 0.0013
445
DW + feats.
0.9005 ±0.0008
This paper
459
N2V
0.8967 ±0.0012
This paper
465
DW
0.8944 ±0.0011
This paper
539
Feats.
0.7853
This paper
#ModelAccuracy ↑Paper
1
GraphTARIF
0.9461 ±0.0017
2
GCN
actual #2 vs paper #269 267 places worse
0.9412 ±0.0008
best reported
actual #2 vs paper #269 267 places worse
paper 0.8651 ±0.0054
paper 0.8651 ±0.0054 Δ at pub 0.0134 Δ today 0.0761
3
GAT
0.9398 ±0.0022
8
SAGE
0.9351 ±0.0006
79
BGRL
actual #79 vs paper #149 70 places worse
0.9034 ±0.0019
best reported
actual #79 vs paper #149 70 places worse
paper 0.8923 ±0.0034
paper 0.8923 ±0.0034 Δ at pub 0.0107 Δ today 0.0111
125
RGRL
0.8970 ±0.0044
This paper
139
GRACE
actual #139 vs paper #200 61 places worse
0.8953 ±0.0035
best reported
actual #139 vs paper #200 61 places worse
paper 0.8815 ±0.0043
paper 0.8815 ±0.0043 Δ at pub 0.0135 Δ today 0.0138
159
CCA-SSG
actual #159 vs paper #254 95 places worse
0.8896 ±0.0013
best reported
actual #159 vs paper #254 95 places worse
paper 0.8701 ±0.0041
paper 0.8701 ±0.0041 Δ at pub 0.0114 Δ today 0.0195
160
GCA
actual #160 vs paper #186 26 places worse
0.8894 ±0.0015
best reported
actual #160 vs paper #186 26 places worse
paper 0.8849 ±0.0051

Difference within reported uncertainty; not counted as underperformance.

paper 0.8849 ±0.0051 Δ at pub 0.0045 Δ today 0.0045
179
MVGRL
actual #179 vs paper #235 56 places worse
0.8861 ±0.0064
best reported
actual #179 vs paper #235 56 places worse
paper 0.8752 ±0.0011
210
DGI
actual #210 vs paper #354 144 places worse
0.8798 ±0.0081
best reported
actual #210 vs paper #354 144 places worse
paper 0.8395 ±0.0047
paper 0.8395 ±0.0047 Δ at pub 0.0005 Δ today 0.0403
280
DW + feats.
0.8628 ±0.0007
This paper
297
DW
0.8568 ±0.0006
This paper
320
MLP
0.8463
335
N2V
0.8439 ±0.0008
This paper
342
GMI
actual #342 vs paper #398 56 places worse
0.8422 ±0.0052
best reported
actual #342 vs paper #398 56 places worse
paper 0.8221 ±0.0031
paper 0.8221 ±0.0031 Δ today 0.0201
459
Feats.
0.7381
This paper
#ModelF1 ↑Paper
1
EnGCN
0.9714 ±0.0003
2
SAGN+SLE
0.9710
3
GAMLP(JK)
0.9704 ±0.0001
14
GraphSAGE
0.9660
63
CCA-SSG
actual #63 vs paper #74 11 places worse
0.9511 ±0.0018
best reported
actual #63 vs paper #74 11 places worse
paper 0.9489 ±0.0200

Difference within reported uncertainty; not counted as underperformance.

paper 0.9489 ±0.0200 Δ at pub 0.0018 Δ today 0.0022
67
RGRL
0.9504 ±0.0300
This paper
68
GRACE
actual #68 vs paper #75 7 places worse
0.9484 ±0.0300
actual #68 vs paper #75 7 places worse
paper 0.9484 ±0.0300

Improvement within reported uncertainty; not highlighted.

paper 0.9484 ±0.0300 Δ at pub 0.0012
74
BGRL
actual #74 vs paper #74
0.9490 ±0.0400
actual #74 vs paper #74
paper 0.9490 ±0.0400

Improvement within reported uncertainty; not highlighted.

paper 0.9490 ±0.0400 Δ at pub 0.0068
75
GCA
0.9485 ±0.0600
This paper
#ModelAccuracy ↑Paper
1
GrokFormer
0.9831 ±0.0018
2
PolyFormer
0.9808 ±0.0027
3
SGFormer
0.9796 ±0.0081
9
GCN
actual #9 vs paper #90 81 places worse
0.9774 ±0.0035
best reported
actual #9 vs paper #90 81 places worse
paper 0.9565 ±0.0016
paper 0.9565 ±0.0016 Δ at pub 0.0085 Δ today 0.0209
17
SAGE
0.9739 ±0.0018
58
MLP
0.9597 ±0.0007
69
BGRL
actual #69 vs paper #125 56 places worse
0.9580 ±0.0040
best reported
actual #69 vs paper #125 56 places worse
paper 0.9543 ±0.0009
paper 0.9543 ±0.0009 Δ at pub 0.0027 Δ today 0.0037
106
CCA-SSG
actual #106 vs paper #149 43 places worse
0.9559 ±0.0007
best reported
actual #106 vs paper #149 43 places worse
paper 0.9516 ±0.0010
paper 0.9516 ±0.0010 Δ at pub 0.0043
124
RGRL
0.9546 ±0.0010
This paper
125
MVGRL
actual #125 vs paper #134 9 places worse
0.9544 ±0.0027
best reported
actual #125 vs paper #134 9 places worse
paper 0.9533 ±0.0003

Difference within reported uncertainty; not counted as underperformance.

paper 0.9533 ±0.0003 Δ today 0.0011
164
DW + feats.
0.9490 ±0.0009
This paper
167
DGI
actual #167 vs paper #174 7 places worse
0.9489 ±0.0009
best reported
actual #167 vs paper #174 7 places worse
paper 0.9451 ±0.0052

Difference within reported uncertainty; not counted as underperformance.

paper 0.9451 ±0.0052 Δ today 0.0038
214
Feats.
0.9358
This paper
263
DW
0.9177 ±0.0015
This paper
269
N2V
0.9119 ±0.0004
This paper
#ModelF1 ↑Paper
1
GAMLP(JK)
0.9982 ±0.0001
2
GraphCON-GCN
0.9960
3
GCNII
0.9953 ±0.0001
21
GCN
0.9850
117
BGRL
actual #117 vs paper #134 17 places worse
0.7363 ±0.0016
best reported
actual #117 vs paper #134 17 places worse
paper 0.6889 ±0.0800

Difference within reported uncertainty; not counted as underperformance.

paper 0.6889 ±0.0800 Δ at pub 0.0474 Δ today 0.0474
119
CCA-SSG
actual #119 vs paper #138 19 places worse
0.7334 ±0.0017
best reported
actual #119 vs paper #138 19 places worse
paper 0.6609 ±0.0100
paper 0.6609 ±0.0100 Δ at pub 0.0725 Δ today 0.0725
126
GRACE
actual #126 vs paper #136 10 places worse
0.6971 ±0.0017
best reported
actual #126 vs paper #136 10 places worse
paper 0.6712 ±0.0500

Difference within reported uncertainty; not counted as underperformance.

paper 0.6712 ±0.0500 Δ at pub 0.0259 Δ today 0.0259
133
RGRL
0.6928 ±0.0600
This paper
138
GCA
0.6672 ±0.0800
This paper
173
MLP
0.4620
#ModelAccuracy ↑Paper
1
GCN
actual #1 vs paper #202 201 places worse
0.9302 ±0.0011
best reported
actual #1 vs paper #202 201 places worse
paper 0.7719 ±0.0012
paper 0.7719 ±0.0012 Δ at pub 0.1583
2
RSE + RGCN (70b)
0.8852 ±0.0019
3
LEMP+T
0.8825
9
SAGE
0.8771
89
RGRL
0.8029 ±0.0072
This paper
94
GRACE
actual #94 vs paper #167 73 places worse
0.8014 ±0.0048
best reported
actual #94 vs paper #167 73 places worse
paper 0.7825 ±0.0065
paper 0.7825 ±0.0065 Δ at pub 0.0189 Δ today 0.0189
100
BGRL
actual #100 vs paper #108 8 places worse
0.8000 ±0.0010
best reported
actual #100 vs paper #108 8 places worse
paper 0.7960 ±0.0060

Difference within reported uncertainty; not counted as underperformance.

paper 0.7960 ±0.0060 Δ at pub 0.0040 Δ today 0.0040
130
CCA-SSG
actual #130 vs paper #179 49 places worse
0.7908 ±0.0053
best reported
actual #130 vs paper #179 49 places worse
paper 0.7788 ±0.0041
paper 0.7788 ±0.0041 Δ today 0.0120
151
DGI
actual #151 vs paper #228 77 places worse
0.7865 ±0.0690
best reported
actual #151 vs paper #228 77 places worse
paper 0.7535 ±0.0014

Difference within reported uncertainty; not counted as underperformance.

paper 0.7535 ±0.0014 Δ at pub 0.0005 Δ today 0.0330
162
GCA
actual #162 vs paper #166 4 places worse
0.7840 ±0.0013
best reported
actual #162 vs paper #166 4 places worse
paper 0.7830 ±0.0062

Difference within reported uncertainty; not counted as underperformance.

paper 0.7830 ±0.0062 Δ at pub 0.0010 Δ today 0.0010
190
MVGRL
actual #190 vs paper #191 1 place worse
0.7757 ±0.0046
best reported
actual #190 vs paper #191 1 place worse
paper 0.7752 ±0.0008

Difference within reported uncertainty; not counted as underperformance.

paper 0.7752 ±0.0008 Δ today 0.0005
199
DW + feats.
0.7721 ±0.0003
This paper
234
GMI
actual #234 vs paper #234
0.7490 ±0.0010
best reported
actual #234 vs paper #234
paper 0.7485 ±0.0008

Difference within reported uncertainty; not counted as underperformance.

paper 0.7485 ±0.0008 Δ at pub 0.0005
237
MLP
0.7469 ±0.0085
241
DW
0.7435 ±0.0006
This paper
255
Feats.
0.7198
This paper
259
N2V
0.7179 ±0.0005
This paper

Heterogeneous Graph Benchmarks

#ModelMicro-F1 ↑Paper
1
LMSPS
0.9566 ±0.0020
2
PSHGCN
0.9561 ±0.0012
3
RpHGNN
0.9555 ±0.0029
21
GCN
0.9460 ±0.0031
37
HAN
actual #37 vs paper #167 130 places worse
0.9369
best reported
actual #37 vs paper #167 130 places worse
paper 0.7080
44
DMGI
actual #44 vs paper #149 105 places worse
0.9331
best reported
actual #44 vs paper #149 105 places worse
paper 0.7660
64
HDMI
actual #64 vs paper #126 62 places worse
0.9220 ±0.0050
best reported
actual #64 vs paper #126 62 places worse
paper 0.8110
123
RGRL
0.8180
This paper
130
BGRL
actual #130 vs paper #130
0.8070
actual #130 vs paper #130
paper 0.8070
149
DMGI_attn
0.7700
This paper
176
MLP
0.6447 ±0.0136
#ModelMicro-F1 ↑Paper
1
ELLA
0.7930 ±0.0010
2
MHGCN
0.7820 ±0.0138
3
PSHGCN
0.7446 ±0.0032
14
GCN
0.6813 ±0.0083
18
HAN
actual #18 vs paper #52 34 places worse
0.6769 ±0.0064
best reported
actual #18 vs paper #52 34 places worse
paper 0.6070
best at pub: Heterogeneous Graph Benchmarks (KDD 2021)
paper 0.6070 Δ at pub 0.0393 Δ today 0.0699
30
RGRL
0.6580
This paper
29
HDMI
0.6580
This paper
33
DMGI
0.6480
This paper
41
BGRL
0.6340
This paper
55
DMGI_attn
0.6060
This paper
74
MLP
0.5514
SOTA performance (not significantly different from the best) Proposed method (this paper) 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