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
| # | Model | Accuracy ↑ | 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 |
| # | Model | Accuracy ↑ | 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 |
| # | Model | Accuracy ↑ | 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
| # | Model | Accuracy ↑ | 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 |
