NeurIPS 2022

Augmentations in Hypergraph Contrastive Learning: Fabricated and Generative

Tianxin Wei, Yuning You, Tianlong Chen, Yang Shen, Jingrui He, Zhangyang Wang

Main figure for Augmentations in Hypergraph Contrastive Learning: Fabricated and Generative

Abstract

This paper targets at improving the generalizability of hypergraph neural networks in the low-label regime, through applying the contrastive learning approach from images/graphs (we refer to it as HyperGCL). We focus on the following question: How to construct contrastive views for hypergraphs via augmentations? We provide the solutions in two folds. First, guided by domain knowledge, we fabricate two schemes to augment hyperedges with higher-order relations encoded, and adopt three vertex augmentation strategies from graph-structured data. Second, in search of more effective views in a data-driven manner, we for the first time propose a hypergraph generative model to generate augmented views, and then an end-to-end differentiable pipeline to jointly learn hypergraph augmentations and model parameters. Our technical innovations are reflected in designing both fabricated and generative augmentations of hypergraphs. The experimental findings include: (i) Among fabricated augmentations in HyperGCL, augmenting hyperedges provides the most numerical gains, implying that higher-order information in structures is usually more downstream-relevant; (ii) Generative augmentations do better in preserving higher-order information to further benefit generalizability; (iii) HyperGCL also boosts robustness and fairness in hypergraph representation learning. Codes are released at.

Benchmark Results

Tested on 4 datasets

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
724
MLP
0.7768
796
A6
0.7312 ±0.0148
This paper
797
A4
0.7295 ±0.0119
This paper
798
A2
0.7258 ±0.0109
This paper
801
A1
0.7239 ±0.0134
This paper
803
A3
0.7233 ±0.0123
This paper
829
Con
0.6889 ±0.0180
This paper
832
A0
0.6859 ±0.0133
This paper
834
Self
0.6824 ±0.0112
This paper
839
A5
0.6796 ±0.0099
This paper
840
SetGNN
0.6793 ±0.0127
This paper
#ModelAccuracy ↑Paper
1
CNA
0.9575 ±0.0058
2
IS-GIB
0.9390 ±0.0187
3
EERM
0.9112 ±0.0145
44
GCN
0.8033 ±0.0088
350
MLP
0.7289 ±0.0010
723
A6
0.6694 ±0.0100
This paper
738
A2
0.6640 ±0.0135
This paper
742
A1
0.6628 ±0.0127
This paper
743
A4
0.6622 ±0.0095
This paper
759
A3
0.6579 ±0.0118
This paper
800
SetGNN
0.6353 ±0.0132
This paper
803
A5
0.6321 ±0.0125
This paper
808
Con
0.6282 ±0.0121
This paper
812
Self
0.6249 ±0.0148
This paper
814
A0
0.6225 ±0.0215
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
210
MLP
0.8690 ±0.0074
237
A6
0.8572 ±0.0038
This paper
250
A3
0.8524 ±0.0028
This paper
253
A1
0.8517 ±0.0037
This paper
254
A2
0.8516 ±0.0038
This paper
261
A4
0.8488 ±0.0038
This paper
263
Con
0.8456 ±0.0034
This paper
264
A0
0.8454 ±0.0042
This paper
266
A5
0.8448 ±0.0040
This paper
268
Self
0.8438 ±0.0038
This paper
269
SetGNN
0.8433 ±0.0036
This paper

Hypergraph Benchmarks

#ModelAccuracy ↑Paper
1
HAN (full batch)
0.8404 ±0.0102
2
AllSetTransformer
0.8363 ±0.0147
3
UniGCNII
0.8360 ±0.0114
14
A6
0.7621 ±0.0126
This paper
16
A2
0.7562 ±0.0142
This paper
17
A1
0.7545 ±0.0154
This paper
18
A3
0.7534 ±0.0140
This paper
19
A4
0.7529 ±0.0156
This paper
20
GCN
0.7516 ±0.0065
21
MLP
0.7431 ±0.0189
22
Con
0.7322 ±0.0165
This paper
23
Self
0.7274 ±0.0153
This paper
24
A5
0.7261 ±0.0186
This paper
25
SetGNN
0.7221 ±0.0151
This paper
26
A0
0.7185 ±0.0162
This paper
SOTA performance (not significantly different from the best) Baseline / comparison (this paper) Global context (other papers)