Self-supervised Learning on Graphs: Deep Insights and New Directions

Wei Jin, Tyler Derr, Haochen Liu, Yiqi Wang, Suhang Wang, Zitao Liu, Jiliang Tang

Main figure for Self-supervised Learning on Graphs: Deep Insights and New Directions

Abstract

The success of deep learning notoriously requires larger amounts of costly annotated data. This has led to the development of self-supervised learning (SSL) that aims to alleviate this limitation by creating domain specific pretext tasks on unlabeled data. Simultaneously, there are increasing interests in generalizing deep learning to the graph domain in the form of graph neural networks (GNNs). GNNs can naturally utilize unlabeled nodes through the simple neighborhood aggregation that is unable to thoroughly make use of unlabeled nodes. Thus, we seek to harness SSL for GNNs to fully exploit the unlabeled data. Different from data instances in the image and text domains, nodes in graphs present unique structure information and they are inherently linked indicating not independent and identically distributed (or i.i.d.). Such complexity is a double-edged sword for SSL on graphs. On the one hand, it determines that it is challenging to adopt solutions from the image and text domains to graphs and dedicated efforts are desired. On the other hand, it provides rich information that enables us to build SSL from a variety of perspectives. Thus, in this paper, we first deepen our understandings on when, why, and which strategies of SSL work with GNNs by empirically studying numerous basic SSL pretext tasks on graphs. Inspired by deep insights from the empirical studies, we propose a new direction SelfTask to build advanced pretext tasks that are able to achieve state-of-the-art performance on various real-world datasets. The specific experimental settings to reproduce our results can be found in.

Benchmark Results

Tested on 4 datasets 1 baseline value reported too low Non-standard metric on 1 dataset

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

Classic

#ModelAccuracy ↑Paper
1
GraphSAGE+CNA
0.9418 ±0.0033
2
GCN+CNA
0.9366 ±0.0048
3
TransformerConv
0.9350 ±0.0058
21
GCN
0.9004
470
3GC
0.8249
This paper
482
2GC
0.8239
This paper
624
1GC+1Linear
0.8075
This paper
631
2GC+1Linear
0.8055
This paper
724
MLP
0.7768
#ModelAccuracy ↑Paper
1
CNA
0.9575 ±0.0058
2
IS-GIB
0.9390 ±0.0187
3
EERM
0.9112 ±0.0145
44
GCN
actual #44 vs paper #503 459 places worse
0.7153 ±0.0027
actual #44 vs paper #503 459 places worse
paper 0.7153 ±0.0027

Improvement within reported uncertainty; not highlighted.

paper 0.7153 ±0.0027 Δ at pub 0.0033 Δ today 0.0880
61
GraphSAGE
0.7923 ±0.0053
300
Self-training
actual #300 vs paper #437 137 places worse
0.7324 ±0.0044
best reported
actual #300 vs paper #437 137 places worse
paper 0.7221 ±0.0053
paper 0.7221 ±0.0053 Δ today 0.0103
325
SelfTask-CorrectedLabel-ICA
0.7304 ±0.0035
This paper
350
MLP
0.7289 ±0.0010
384
SelfTask-ContextLabel-ICA
0.7259 ±0.0045
This paper
403
SelfTask-EnsembleLabel
0.7248 ±0.0028
This paper
432
SelfTask-CorrectedLabel-LP
0.7230 ±0.0053
This paper
465
PairwiseNodeDistance
0.7190
This paper
466
SelTask-ContextLabel-LP
0.7190 ±0.0020
This paper
480
M3S
0.7172 ±0.0036
This paper
493
PairwiseAttrSim
0.7167
This paper
494
SelfTask-Distance2Labeled
0.7164 ±0.0028
This paper
499
NodeProperty
0.7160
This paper
504
EdgeMask
0.7151
This paper
516
Distance2Cluster
0.7144
This paper
534
GCN-DroppedGraph
0.7129
This paper
588
AttributeMask
0.7057
This paper
600
GCN-PCA
0.7038
This paper
#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 #551 533 places worse
0.9296 ±0.0015
best reported
actual #18 vs paper #551 533 places worse
paper 0.7928 ±0.0044
paper 0.7928 ±0.0044 Δ at pub 0.0022 Δ today 0.1368
116
SAGE
0.8881 ±0.0015
210
MLP
0.8690 ±0.0074
321
SelTask-ContextLabel-ICA
0.8231 ±0.0014
This paper
356
SelfTask-EnsembleLabel
0.8131 ±0.0016
This paper
418
Self-training
actual #418 vs paper #541 123 places worse
0.8032 ±0.0018
best reported
actual #418 vs paper #541 123 places worse
paper 0.7935 ±0.0062
paper 0.7935 ±0.0062 Δ today 0.0097
437
SelfTask-CorrectedLabel-LP
0.8016 ±0.0030
This paper
445
SelTask-ContextLabel-LP
0.8007 ±0.0018
This paper
447
PairwiseNodeDistance
0.8005
This paper
469
Distance2Cluster
0.7988
This paper
510
M3S
actual #510 vs paper #510
0.7950 ±0.0076
actual #510 vs paper #510
paper 0.7950 ±0.0076

Improvement within reported uncertainty; not highlighted.

paper 0.7950 ±0.0076 Δ at pub 0.0022
520
PairwiseAttrSim
0.7945
This paper
523
NodeProperty
0.7944
This paper
552
GCN-DroppedGraph
0.7928
This paper
605
EdgeMask
0.7890
This paper
609
AttributeMask
0.7888
This paper
612
GCN-PCA
0.7883
This paper

Reddit

Global leaderboard →
Non-default metric. This paper reports Accuracy; the dataset default is F1. Values below use Accuracy only.

Sorted by Accuracy. Highlighted rows are from other papers.

#ModelAccuracy ↑Paper
1
GAP-EDP
0.9870 ±0.0003
2
NE-ASGCN
0.9758 ±0.0027
3
HSGT
0.9730 ±0.0024
21
GraphSAGE
0.9653 ±0.0011
26
GCN
actual #26 vs paper #72 46 places worse
0.9648
best reported
actual #26 vs paper #72 46 places worse
paper 0.9499 ±0.0004
paper 0.9499 ±0.0004 Δ at pub 0.0149
54
SelfTask-CorrectedLabel-LP
0.9533 ±0.0002
This paper
55
SelTask-ContextLabel-LP
0.9532 ±0.0002
This paper
57
SelfTask-EnsembleLabel
0.9529 ±0.0002
This paper
59
SelfTask-CorrectedLabel-ICA
0.9525 ±0.0009
This paper
62
SelfTask-ContextLabel-ICA
0.9521 ±0.0008
This paper
63
M3S
0.9518 ±0.0002
This paper
72
Self-training
0.9500 ±0.0003
This paper
147
MLP
0.8240 ±0.0010
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 today — OK at publication; today's leaderboard best is elsewhere