Graph Self-Supervised Learning with Learnable Structural and Positional Encodings
Abstract
Traditional Graph Self-Supervised Learning (GSSL) struggles to capture complex structural properties well. This limitation stems from two main factors: (1) the inadequacy of conventional Graph Neural Networks (GNNs) in representing sophisticated topological features, and (2) the focus of self-supervised learning solely on final graph representations. To address these issues, we introduce GenHopNet, a GNN framework that integrates a $k$-hop message-passing scheme, enhancing its ability to capture local structural information without explicit substructure extraction. We theoretically demonstrate that GenHopNet surpasses the expressiveness of the classical Weisfeiler-Lehman (WL) test for graph isomorphism. Furthermore, we propose a structural- and positional-aware GSSL framework that incorporates topological information throughout the learning process. This approach enables the learning of representations that are both sensitive to graph topology and invariant to specific structural and feature augmentations. Comprehensive experiments on graph classification datasets, including those designed to test structural sensitivity, show that our method consistently outperforms the existing approaches and maintains computational efficiency. Our work significantly advances GSSL's capability in distinguishing graphs with similar local structures but different global topologies.
Benchmark Results
Tested on 9 datasets 11 baseline values reported too low Non-standard metric on 3 datasets
Tables show presented results in global context (best 3 papers + GCN + MLP).
TU Dortmund
| # | Model | Accuracy ↑ | 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 best reported | best at pub: ProtGNN: Towards Self-Explaining Graph Neural Networks (AAAI 2021) Difference within reported uncertainty; not counted as underperformance. |
| 46 | StructPosGSSL-SA | 0.9300
±0.0530 | This paper |
| 55 | PATCHY-SAN | 0.9260
±0.0420 | This paper |
| 63 | GCN | 0.9220
±0.0440 | |
| 91 | StructPosGSSL-FA | 0.9150
±0.0250 | This paper |
| 109 | GALOPA | 0.9110
±0.0120 | This paper |
| 146 | P-WL | 0.9051
±0.0134 best reported | best at pub: On Structural Expressive Power of Graph Transformers (KDD 2023) Difference within reported uncertainty; not counted as underperformance. |
| 155 | GCS | 0.9045
±0.0081 best reported | best at pub: SpeGCL: Self-supervised Graph Spectrum Contrastive Learning without Positive Samples (2024) Difference within reported uncertainty; not counted as underperformance. |
| 175 | RetGK | 0.9030
±0.0110 | This paper |
| 176 | GA2C | 0.9030
±0.0030 | This paper |
| 199 | MVGRL | 0.8993
±0.0616 best reported | best at pub: Improving Expressivity of GNNs with Subgraph-specific Factor Embedded Normalization (KDD 2023) Difference within reported uncertainty; not counted as underperformance. |
| 204 | G3N | 0.8990
±0.0800 | This paper |
| 285 | GraphCL+HTML | 0.8890
±0.0180 | This paper |
| 301 | CapsGNN | 0.8867
±0.0688 best reported | best at pub: Towards Better Graph Representation Learning with Parameterized Decomposition & Filtering (ICML 2023) Difference within reported uncertainty; not counted as underperformance. |
| 353 | WL-PM | 0.8770
±0.0080 | This paper |
| 354 | GraphCL | 0.8766
±0.0103 best reported | best at pub: Graph Masked Autoencoders with Transformers (2022) Difference within reported uncertainty; not counted as underperformance. |
| 388 | WL-OA | 0.8715
±0.0182 best reported | best at pub: Wasserstein Weisfeiler-Lehman Graph Kernels (NeurIPS 2019) |
| 403 | MLP | 0.8660
±0.0495 |
| # | Model | Accuracy ↑ | Paper |
|---|---|---|---|
| 1 | WL-MLP | 0.9954 | |
| 2 | GIN | 0.9934 best reported | best at pub: 1-WL Expressiveness Is (Almost) All You Need (2022) |
| 3 | GCN | 0.9727 | |
| 7 | WL-PM | 0.8640
±0.0020 | This paper |
| 8 | WL-OA | 0.8610
±0.0020 | This paper |
| 18 | P-WL | 0.8540
±0.0010 | This paper |
| 31 | RetGK | 0.8450
±0.0020 | This paper |
| 79 | StructPosGSSL-SA | 0.8290
±0.0380 | This paper |
| 106 | StructPosGSSL-FA | 0.8250
±0.0350 | This paper |
| 104 | GraphCL | 0.8250
±0.0013 best reported | best at pub: Label-invariant Augmentation for Semi-Supervised Graph Classification (NeurIPS 2022) |
| 118 | G3N | 0.8230
±0.0520 | This paper |
| 131 | GraphSAGE | 0.8200
±0.0100 | |
| 189 | GA2C | 0.8060
±0.0030 | This paper |
| 279 | GraphCL+HTML | 0.7870
±0.0070 | This paper |
| 288 | PATCHY-SAN | 0.7860
±0.0190 | This paper |
| 302 | CapsGNN | 0.7840
±0.0160 best reported | best at pub: Molecular Representation Learning via Heterogeneous Motif Graph Neural Networks (ICML 2022) Difference within reported uncertainty; not counted as underperformance. |
| 328 | GALOPA | 0.7780
±0.0030 | This paper |
| 344 | GCS | 0.7737
±0.0030 best reported | best at pub: SpeGCL: Self-supervised Graph Spectrum Contrastive Learning without Positive Samples (2024) Difference within reported uncertainty; not counted as underperformance. |
| 356 | MVGRL | 0.7700
±0.0080 | This paper |
| # | Model | Accuracy ↑ | Paper |
|---|---|---|---|
| 1 | MSH-GNN | 0.9410
±0.0330 | |
| 2 | GRDL | 0.8260
±0.0120 | |
| 3 | GMN | 0.8225 | |
| 32 | GIN | 0.7880
±0.0410 best reported | best at pub: Permutation-Invariant graph partitioning: How graph neural networks capture structural interactions? (2023) Difference within reported uncertainty; not counted as underperformance. |
| 67 | SAGE | 0.7726
±0.0228 | |
| 89 | GALOPA | 0.7690
±0.0010 | This paper |
| 117 | WL-OA | 0.7640
±0.0040 best reported | best at pub: On Valid Optimal Assignment Kernels and Applications to Graph Classification (NeurIPS 2016) |
| 126 | CapsGNN | 0.7630
±0.0360 best reported | best at pub: Wasserstein Embedding for Graph Learning (ICLR 2020) Difference within reported uncertainty; not counted as underperformance. |
| 139 | RetGK | 0.7620
±0.0050 best reported | best at pub: On Structural Expressive Power of Graph Transformers (KDD 2023) |
| 150 | GA2C | 0.7600
±0.0010 | This paper |
| 149 | MLP | 0.7600
±0.0320 | |
| 167 | StructPosGSSL-SA | 0.7580
±0.0260 | This paper |
| 212 | P-WL | 0.7531
±0.0073 best reported | best at pub: On Structural Expressive Power of Graph Transformers (KDD 2023) |
| 215 | StructPosGSSL-FA | 0.7530
±0.0290 | This paper |
| 237 | PATCHY-SAN | 0.7510
±0.0330 best reported | best at pub: Graph Classification via Reference Distribution Learning: Theory and Practice (NeurIPS 2024) Difference within reported uncertainty; not counted as underperformance. |
| 245 | GCS | 0.7502
±0.0081 best reported | best at pub: SpeGCL: Self-supervised Graph Spectrum Contrastive Learning without Positive Samples (2024) Difference within reported uncertainty; not counted as underperformance. |
| 262 | GraphCL+HTML | 0.7490
±0.0030 | This paper |
| 265 | GraphCL | 0.7489
±0.0065 best reported | best at pub: Graph Masked Autoencoders with Transformers (2022) Difference within reported uncertainty; not counted as underperformance. |
| 336 | WL-PM | 0.7360
±0.0020 | This paper |
| 337 | G3N | 0.7360
±0.0530 | This paper |
| # | Model | Accuracy ↑ | Paper |
|---|---|---|---|
| 1 | WL-MLP | 0.9792 | |
| 2 | MSH-GNN | 0.9640
±0.0070 | |
| 3 | CoCN | 0.8722
±0.0013 | |
| 23 | GCN | 0.8260
±0.0220 | |
| 30 | GIN | 0.8220
±0.0210 best reported | best at pub: Scalable Expressiveness through Preprocessed Graph Perturbations (2024) Difference within reported uncertainty; not counted as underperformance. |
| 50 | WL-PM | 0.8150
±0.0050 | This paper |
| 80 | GraphCL | 0.8108
±0.0017 best reported | best at pub: Label-invariant Augmentation for Semi-Supervised Graph Classification (NeurIPS 2022) |
| 82 | RetGK | 0.8100
±0.0030 | This paper |
| 94 | WL-OA | 0.8070
±0.0010 | This paper |
| 104 | MVGRL | 0.8048
±0.0029 best reported | best at pub: Graph Contrastive Learning with Cohesive Subgraph Awareness (WWW 2024) |
| 140 | CapsGNN | 0.7962
±0.0091 best reported | best at pub: Quaternion Graph Neural Networks (2020) best today: Graph Structure Refinement with Energy-based Contrastive Learning (AAAI 2024) Difference within reported uncertainty; not counted as underperformance. |
| 151 | MLP | 0.7920
±0.0240 | |
| 219 | StructPosGSSL-SA | 0.7650
±0.0360 | This paper |
| 227 | StructPosGSSL-FA | 0.7610
±0.0330 | This paper |
| 275 | PATCHY-SAN | 0.7310
±0.0270 best reported | best at pub: Graph Classification via Reference Distribution Learning: Theory and Practice (NeurIPS 2024) Difference within reported uncertainty; not counted as underperformance. |
OGB
ogbg-molhiv
Global leaderboard →Sorted by Accuracy. Highlighted rows are from other papers.
| # | Model | Accuracy ↑ | Paper |
|---|---|---|---|
| 1 | MPNNs | 0.9800 | |
| 2 | non-convex counterparts | 0.9600 | |
| 3 | StructPosGSSL-FA | 0.7900
±0.0243 | This paper |
| 4 | StructPosGSSL-SA | 0.7880
±0.0227 | This paper |
| 5 | GraphSNN | 0.7851
±0.0170 | This paper |
| 6 | Policy-Learn | 0.7849
±0.0101 | This paper |
| 7 | GSN | 0.7799
±0.0100 | This paper |
| 8 | DS-GNN (EGO+) | 0.7740
±0.0219 | This paper |
| 9 | DSS-GNN (EGO+) | 0.7678
±0.0166 | This paper |
| 10 | GIN+VN | 0.7520
±0.0130 | This paper |
ogbg-molpcba
Global leaderboard →Sorted by Accuracy. Highlighted rows are from other papers.
| # | Model | Accuracy ↑ | Paper |
|---|---|---|---|
| 1 | StructPosGSSL-FA | 0.2745
±0.0195 | This paper |
| 2 | StructPosGSSL-SA | 0.2710
±0.0165 | This paper |
| 3 | GIN+VN | 0.2703
±0.0023 | This paper |
| 5 | GraphSNN | 0.2496
±0.0150 | This paper |
| 9 | GCN | 0.2020
±0.0024 |
| # | Model | Accuracy ↑ | Paper |
|---|---|---|---|
| 1 | A-NLSF | 0.8149
±0.0067 | |
| 2 | differential encoding | 0.8096
±0.0029 | |
| 3 | MPNN + VN + NoPE | 0.8055
±0.0038 | |
| 16 | GatedGCN | 0.7531
±0.0083 | |
| 29 | StructPosGSSL-FA | 0.7090
±0.0186 | This paper |
| 30 | StructPosGSSL-SA | 0.7081
±0.0145 | This paper |
| 31 | GraphSNN | 0.7066
±0.0165 | This paper |
| 34 | GIN+VN | 0.7037
±0.0107 | This paper |
| 33 | GIN | 0.7037
±0.0107 |
MoleculeNet
Tox21
Global leaderboard →Sorted by Accuracy. Highlighted rows are from other papers.
| # | Model | Accuracy ↑ | Paper |
|---|---|---|---|
| 1 | GraphSNN | 0.9224
±0.0059 best reported | best at pub: Union Subgraph Neural Networks (AAAI 2023) |
| 2 | UnionSNN | 0.9176
±0.0085 | |
| 3 | GIN | 0.9174
±0.0095 | |
| 14 | GCN | 0.9010
±0.0077 | |
| 27 | DSS-GNN (EGO+) | 0.7795
±0.0040 | This paper |
| 28 | Policy-Learn | 0.7736
±0.0060 | This paper |
| 33 | DS-GNN (EGO+) | 0.7639
±0.0118 | This paper |
| 42 | StructPosGSSL-FA | 0.6700
±0.0220 | This paper |
| 43 | StructPosGSSL-SA | 0.6650
±0.0260 | This paper |
| 44 | GIN+VN | 0.6618
±0.0068 | This paper |
GNNBenchmark
| # | Model | Accuracy ↑ | Paper |
|---|---|---|---|
| 8 | GCN | 1.0000 best reported | best at pub: GNNBenchmark (JMLR 2023) |
| 1 | PPGN | 1.0000 | |
| 2 | RW-AgentNet | 1.0000 | |
| 3 | AgentNet | 1.0000 | |
| 35 | GraphSAGE | 0.9993
±0.0047 | |
| 40 | GIN | 0.9933
±0.0133 best reported | best at pub: GNNBenchmark (JMLR 2023) |
| 42 | StructPosGSSL-SA | 0.9860
±0.0280 | This paper |
| 44 | StructPosGSSL-FA | 0.9830
±0.0250 | This paper |
| 46 | 3WLGNN | 0.9780
±0.1090 | This paper |
| 48 | 3-GCN | 0.9570
±0.1480 | This paper |
| 71 | GCN-RNI | 0.1600 | This paper |
