Towards Graph Foundation Models: A Study on the Generalization of Positional and Structural Encodings
Abstract
Recent advances in integrating positional and structural encodings (PSEs) into graph neural networks (GNNs) have significantly enhanced their performance across various graph learning tasks. However, the general applicability of these encodings and their potential to serve as foundational representations for graphs remain uncertain. This paper investigates the fine-tuning efficiency, scalability with sample size, and generalization capability of learnable PSEs across diverse graph datasets. Specifically, we evaluate their potential as universal pre-trained models that can be easily adapted to new tasks with minimal fine-tuning and limited data. Furthermore, we assess the expressivity of the learned representations, particularly, when used to augment downstream GNNs. We demonstrate through extensive benchmarking and empirical analysis that PSEs generally enhance downstream models. However, some datasets may require specific PSE-augmentations to achieve optimal performance. Nevertheless, our findings highlight their significant potential to become integral components of future graph foundation models. We provide new insights into the strengths and limitations of PSEs, contributing to the broader discourse on foundation models in graph learning.
Benchmark Results
Tested on 7 datasets 3 baseline values reported too low
Tables show presented results in global context (best 3 papers + GCN + MLP).
MoleculeNet
| # | Model | ROC-AUC ↑ | Paper |
|---|---|---|---|
| 1 | HIMP | 0.8740
±0.0050 | |
| 2 | GINE w/ VN | 0.8720
±0.0030 | |
| 3 | NaiveGINe+ K=3 w/ VN | 0.8700
±0.0040 | |
| 14 | GCN | 0.8400
±0.0040 | |
| 71 | GraphLoG | 0.7960
±0.0250 best reported | best at pub: KPGT: Knowledge-Guided Pre-training of Graph Transformer for Molecular Property Prediction (KDD 2022) |
| 132 | GPSE- | 0.7750
±0.0090 | This paper |
| 139 | LapPE | 0.7740
±0.0040 | This paper |
| 182 | RWSE | 0.7600
±0.0070 | This paper |
| 183 | AllPSE | 0.7600
±0.0060 | This paper |
| # | Model | ROC-AUC ↑ | Paper |
|---|---|---|---|
| 1 | ProtoMol | 0.9140
±0.0030 | |
| 2 | MMSG | 0.9080
±0.0050 | |
| 3 | MemGNN | 0.9070 | |
| 43 | GIN | 0.8646
±0.0081 | |
| 98 | GraphLoG | 0.8350
±0.0120 best reported | best at pub: Motif-aware Attribute Masking for Molecular Graph Pre-training (2023) best today: UGMAE: A Unified Framework for Graph Masked Autoencoders (2024) Difference within reported uncertainty; not counted as underperformance. |
| 107 | GPSE | 0.8310
±0.0160 | This paper |
| 144 | RWSE | 0.8140
±0.0220 | This paper |
| 216 | LapPE | 0.7810
±0.0260 | This paper |
| 219 | AllPSE | 0.7790
±0.0280 | This paper |
| # | Model | ROC-AUC ↑ | Paper |
|---|---|---|---|
| 1 | MAGPrompt+ | 0.9432
±0.0238 | |
| 2 | CamS-LLaMA | 0.9420 | |
| 3 | SE(3)-I MPPH | 0.9400
±0.0210 | |
| 34 | GIN | 0.8875
±0.0049 | |
| 54 | GraphLoG | 0.8460
±0.0080 best reported | best at pub: KPGT: Knowledge-Guided Pre-training of Graph Transformer for Molecular Property Prediction (KDD 2022) |
| 214 | LapPE | 0.6770
±0.0280 | This paper |
| 217 | GPSE- | 0.6760
±0.0110 | This paper |
| 240 | AllPSE | 0.6680
±0.0070 | This paper |
| 249 | RWSE | 0.6630
±0.0200 | This paper |
| # | Model | ROC-AUC ↑ | Paper |
|---|---|---|---|
| 1 | SE(3)-I MPPH | 0.9930
±0.0040 | |
| 2 | KA-GCN | 0.9920
±0.0050 | |
| 3 | KA-GAT | 0.9910
±0.0050 | |
| 19 | GCN | 0.9130
±0.0170 | |
| 103 | GraphLoG | 0.7872
±0.0258 best reported | best at pub: Let Invariant Rationale Discovery Inspire Graph Contrastive Learning (ICML 2022) best today: Graph Self-Contrast Representation Learning (2023) Difference within reported uncertainty; not counted as underperformance. |
| 117 | GPSE | 0.7670
±0.0460 | This paper |
| 125 | LapPE | 0.7540
±0.0280 | This paper |
| 146 | RWSE | 0.7340
±0.0400 | This paper |
| 159 | AllPSE | 0.7200
±0.0350 | This paper |
| # | Model | ROC-AUC ↑ | Paper |
|---|---|---|---|
| 1 | KA-GAT | 0.8470
±0.0020 | |
| 2 | KA-GCN | 0.8420
±0.0010 | |
| 3 | GraphKAN | 0.8370
±0.0010 | |
| 66 | GIN | 0.6403
±0.0104 | |
| 85 | AllPSE | 0.6290
±0.0030 | This paper |
| 93 | GPSE | 0.6270
±0.0120 | This paper |
| 123 | GraphLoG | 0.6150
±0.0130 best reported | best at pub: KPGT: Knowledge-Guided Pre-training of Graph Transformer for Molecular Property Prediction (KDD 2022) Difference within reported uncertainty; not counted as underperformance. |
| 151 | LapPE | 0.6070
±0.0090 | This paper |
| 201 | RWSE | 0.5920
±0.0080 | This paper |
| # | Model | ROC-AUC ↑ | Paper |
|---|---|---|---|
| 1 | GeomGCL | 0.7630 | |
| 2 | LiGhT DeepGPT | 0.7570
±0.0110 | |
| 3 | CoMPT | 0.7550 | |
| 14 | GCN | 0.7350
±0.0020 | |
| 64 | GraphLoG | 0.6770
±0.0190 best reported | best at pub: KPGT: Knowledge-Guided Pre-training of Graph Transformer for Molecular Property Prediction (KDD 2022) |
| 89 | GPSE | 0.6600
±0.0060 | This paper |
| 115 | LapPE | 0.6450
±0.0070 | This paper |
| 130 | AllPSE | 0.6410
±0.0040 | This paper |
| 133 | RWSE | 0.6390
±0.0050 | This paper |
GNNBenchmark
| # | Model | Accuracy ↑ | Paper |
|---|---|---|---|
| 1 | PPGN | 1.0000 | |
| 2 | RW-AgentNet | 1.0000 | |
| 3 | AgentNet | 1.0000 | |
| 8 | GCN | 1.0000 | |
| 61 | GPSE | 0.4270
±0.1020 | This paper |
