WWW 2022

SimGRACE: A Simple Framework for Graph Contrastive Learning without Data Augmentation

Jun Xia, Lirong Wu, Jintao Chen, Bozhen Hu, Stan Z. Li

Main figure for SimGRACE: A Simple Framework for Graph Contrastive Learning without Data Augmentation

Abstract

Graph contrastive learning (GCL) has emerged as a dominant technique for graph representation learning which maximizes the mutual information between paired graph augmentations that share the same semantics. Unfortunately, it is difficult to preserve semantics well during augmentations in view of the diverse nature of graph data. Currently, data augmentations in GCL broadly fall into three unsatisfactory ways. First, the augmentations can be manually picked per dataset by trial-and-errors. Second, the augmentations can be selected via cumbersome search. Third, the augmentations can be obtained with expensive domain knowledge as guidance. All of these limit the efficiency and more general applicability of existing GCL methods. To circumvent these crucial issues, we propose a ple framework for ph ontrastive l arning, SimGRACE for brevity, which does not require data augmentations. Specifically, we take original graph as input and GNN model with its perturbed version as two encoders to obtain two correlated views for contrast. SimGRACE is inspired by the observation that graph data can preserve their semantics well during encoder perturbations while not requiring manual trial-and-errors, cumbersome search or expensive domain knowledge for augmentations selection. Also, we explain why SimGRACE can succeed. Furthermore, we devise adversarial training scheme, dubbed AT-SimGRACE, to enhance the robustness of graph contrastive learning and theoretically explain the reasons. Albeit simple, we show that SimGRACE can yield competitive or better performance compared with state-of-the-art methods in terms of generalizability, transferability and robustness, while enjoying unprecedented degree of flexibility and efficiency. The code is available at: magenta.

Proposed method: SimGRACE

Benchmark Results

Tested on 4 datasets

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
0.9330 ±0.0290
272
SimGRACE
actual #272 vs paper #272
0.8910 ±0.0140
best reported
actual #272 vs paper #272
paper 0.8901 ±0.0131

Difference within reported uncertainty; not counted as underperformance.

paper 0.8901 ±0.0131 Δ today 0.0009
403
MLP
0.8660 ±0.0495
#ModelAccuracy ↑Paper
1
WL-MLP
0.9954
2
GIN
0.9934
3
GCN
0.9727
131
GraphSAGE
0.8200 ±0.0100
141
SimGRACE
actual #141 vs paper #258 117 places worse
0.8180 ±0.0015
best reported
actual #141 vs paper #258 117 places worse
paper 0.7912 ±0.0044
paper 0.7912 ±0.0044 Δ today 0.0268
#ModelAccuracy ↑Paper
1
MSH-GNN
0.9410 ±0.0330
2
GRDL
0.8260 ±0.0120
3
GMN
0.8225
32
GIN
0.7880 ±0.0410
149
MLP
0.7600 ±0.0320
209
SimGRACE
0.7535 ±0.0009
This paper
#ModelAccuracy ↑Paper
1
OEPG
0.9630 ±0.0090
2
MSH-GNN
0.9580 ±0.0090
3
GNN-AK+
0.9480 ±0.0080
13
GIN
0.9240 ±0.0250
44
SimGRACE
0.8951 ±0.0089
This paper
SOTA performance (not significantly different from the best) Proposed method (this paper) Global context (other papers) Best today — OK at publication; today's leaderboard best is elsewhere