TMLR 2024

Towards Graph Foundation Models: A Study on the Generalization of Positional and Structural Encodings

B. J. Franks, Moshe Eliasof, Semih Cantürk, Guy Wolf, C. Schonlieb, Sophie Fellenz, Marius Kloft

Main figure for 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

#ModelROC-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
actual #71 vs paper #276 205 places worse
0.7960 ±0.0250
best reported
actual #71 vs paper #276 205 places worse
paper 0.7340 ±0.0090
paper 0.7340 ±0.0090 Δ at pub 0.0620
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
#ModelROC-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
actual #98 vs paper #115 17 places worse
0.8350 ±0.0120
best reported
actual #98 vs paper #115 17 places worse
paper 0.8260 ±0.0160

Difference within reported uncertainty; not counted as underperformance.

paper 0.8260 ±0.0160 Δ at pub 0.0090 Δ today 0.0090
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
#ModelROC-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
actual #54 vs paper #239 185 places worse
0.8460 ±0.0080
best reported
actual #54 vs paper #239 185 places worse
paper 0.6680 ±0.0210
paper 0.6680 ±0.0210 Δ at pub 0.1780
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
#ModelROC-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
actual #103 vs paper #145 42 places worse
0.7872 ±0.0258
best reported
actual #103 vs paper #145 42 places worse
paper 0.7340 ±0.0390

Difference within reported uncertainty; not counted as underperformance.

paper 0.7340 ±0.0390 Δ at pub 0.0532 Δ today 0.0532
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
#ModelROC-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
actual #123 vs paper #154 31 places worse
0.6150 ±0.0130
best reported
actual #123 vs paper #154 31 places worse
paper 0.6060 ±0.0080

Difference within reported uncertainty; not counted as underperformance.

paper 0.6060 ±0.0080 Δ at pub 0.0090
151
LapPE
0.6070 ±0.0090
This paper
201
RWSE
0.5920 ±0.0080
This paper
#ModelROC-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
actual #64 vs paper #162 98 places worse
0.6770 ±0.0190
best reported
actual #64 vs paper #162 98 places worse
paper 0.6300 ±0.0080
paper 0.6300 ±0.0080 Δ at pub 0.0470
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

#ModelAccuracy ↑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
SOTA performance (not significantly different from the best) Baseline / comparison (this paper) Global context (other papers) Best at pub — reported result worse than the best known when published (best reported in green)