AAAI 2020

SAIL: Self-Augmented Graph Contrastive Learning

Lu Yu, Shichao Pei, Lizhong Ding, Jun Zhou, Longfei Li, Chuxu Zhang, Xiangliang Zhang

Main figure for SAIL: Self-Augmented Graph Contrastive Learning

Abstract

This paper studies learning node representations with graph neural networks (GNNs) for unsupervised scenario. Specifically, we derive a theoretical analysis and provide an empirical demonstration about the non-steady performance of GNNs over different graph datasets, when the supervision signals are not appropriately defined. The performance of GNNs depends on both the node feature smoothness and the locality of graph structure. To smooth the discrepancy of node proximity measured by graph topology and node feature, we proposed SAIL - a novel elf- ugmented graph contrast ve earning framework, with two complementary self-distilling regularization modules, i.e., intra- and inter-graph knowledge distillation. We demonstrate the competitive performance of SAIL on a variety of graph applications. Even with a single GNN layer, SAIL has consistently competitive or even better performance on various benchmark datasets, comparing with state-of-the-art baselines.

Proposed method: SAIL

Benchmark Results

Tested on 3 datasets 6 baseline values reported too low

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
44
GAT
actual #44 vs paper #424 380 places worse
0.8897 ±0.0014
best reported
actual #44 vs paper #424 380 places worse
paper 0.8300 ±0.0070
141
GraphSAGE
actual #141 vs paper #736 595 places worse
0.8708 ±0.0085
best reported
actual #141 vs paper #736 595 places worse
paper 0.7720 ±0.0030
156
MVGRL
actual #156 vs paper #365 209 places worse
0.8680 ±0.0050
best reported
actual #156 vs paper #365 209 places worse
paper 0.8350 ±0.0040
paper 0.8350 ±0.0040 Δ today 0.0330
177
DGI
actual #177 vs paper #486 309 places worse
0.8630 ±0.0020
best reported
actual #177 vs paper #486 309 places worse
paper 0.8230 ±0.0060
paper 0.8230 ±0.0060 Δ today 0.0400
218
GMI
actual #218 vs paper #439 221 places worse
0.8526 ±0.0008
best reported
actual #218 vs paper #439 221 places worse
paper 0.8280 ±0.0070
223
SGC
actual #223 vs paper #606 383 places worse
0.8509 ±0.0648
best reported
actual #223 vs paper #606 383 places worse
paper 0.8100

Difference within reported uncertainty; not counted as underperformance.

paper 0.8100 Δ today 0.0409
247
SAIL
0.8460 ±0.0030
This paper
724
MLP
0.7768
772
CAN
0.7520 ±0.0050
This paper
#ModelAccuracy ↑Paper
1
CNA
0.9575 ±0.0058
2
IS-GIB
0.9390 ±0.0187
3
EERM
0.9112 ±0.0145
35
GAT
actual #35 vs paper #388 353 places worse
0.8080 ±0.0026
best reported
actual #35 vs paper #388 353 places worse
paper 0.7250 ±0.0070
paper 0.7250 ±0.0070 Δ at pub 0.0610 Δ today 0.0830
44
GCN
0.8033 ±0.0088
61
GraphSAGE
actual #61 vs paper #694 633 places worse
0.7923 ±0.0053
best reported
actual #61 vs paper #694 633 places worse
paper 0.6780 ±0.0030
paper 0.6780 ±0.0030 Δ at pub 0.0430 Δ today 0.1143
66
DGI
actual #66 vs paper #474 408 places worse
0.7890 ±0.0020
best reported
actual #66 vs paper #474 408 places worse
paper 0.7180 ±0.0070
paper 0.7180 ±0.0070 Δ today 0.0710
73
GMI
actual #73 vs paper #326 253 places worse
0.7869 ±0.0016
best reported
actual #73 vs paper #326 253 places worse
paper 0.7300 ±0.0030
87
MVGRL
actual #87 vs paper #446 359 places worse
0.7808 ±0.0006
best reported
actual #87 vs paper #446 359 places worse
paper 0.7210 ±0.0060
96
SGC
actual #96 vs paper #463 367 places worse
0.7765 ±0.0012
best reported
actual #96 vs paper #463 367 places worse
paper 0.7190 ±0.0010
221
SAIL
0.7420 ±0.0040
This paper
350
MLP
0.7289 ±0.0010
786
CAN
0.6450 ±0.0020
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
0.9296 ±0.0015
20
GAT
actual #20 vs paper #585 565 places worse
0.9233 ±0.0018
best reported
actual #20 vs paper #585 565 places worse
paper 0.7900 ±0.0030
125
GraphSAGE
actual #125 vs paper #711 586 places worse
0.8864 ±0.0030
best reported
actual #125 vs paper #711 586 places worse
paper 0.7750 ±0.0050
paper 0.7750 ±0.0050 Δ at pub 0.0960 Δ today 0.1114
141
SGC
actual #141 vs paper #602 461 places worse
0.8832 ±0.0055
best reported
actual #141 vs paper #602 461 places worse
paper 0.7890
195
GMI
actual #195 vs paper #439 244 places worse
0.8726 ±0.0016
best reported
actual #195 vs paper #439 244 places worse
paper 0.8010 ±0.0020
196
MVGRL
actual #196 vs paper #474 278 places worse
0.8725 ±0.0012
best reported
actual #196 vs paper #474 278 places worse
paper 0.7980 ±0.0070
204
DGI
actual #204 vs paper #745 541 places worse
0.8708 ±0.0017
best reported
actual #204 vs paper #745 541 places worse
paper 0.7680 ±0.0060
210
MLP
0.8690 ±0.0074
286
SAIL
0.8380 ±0.0010
This paper
862
CAN
0.6480 ±0.0030
This paper
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