Invariant-Based Diagnostics for Graph Benchmarks
Abstract
Progress on graph foundation models is hindered by benchmark practices that conflate the contributions of node features and graph structure, making it hard to tell whether a model actually learns from connectivity, or whether it even needs to. We propose addressing this using graph invariants, i.e., permutation-invariant, task-agnostic structural descriptors that serve as a diagnostic framework for graph benchmarks. We show that (i) invariants are more expressive than standard GNNs, (ii) invariants characterize structural heterogeneity within and across benchmark datasets, (iii) invariants predict multi-task performance, and (iv) simple invariant-based models are competitive with, and sometimes exceed, transformer and message-passing baselines across 26 datasets. Our results suggest that expressivity is not the main driver of predictive performance, and that on tasks where structure matters, a non-trainable structural proxy often matches trained message-passing models. We thus posit that invariant baselines should become a standard for evaluating whether structure is required for a task and whether a model picks up on it, serving as a stepping stone towards graph foundation models.
Benchmark Results
Tested on 2 datasets 8 baseline values reported too low
Tables show presented results in global context (best 3 papers + GCN + MLP).
MoleculeNet
| # | Model | ROC-AUC ↑ | Paper |
|---|---|---|---|
| 1 | ProtoMol | 0.9140
±0.0030 | |
| 2 | MMSG | 0.9080
±0.0050 | |
| 3 | MemGNN | 0.9070 | |
| 9 | MolCLR | 0.8900
±0.0030 best reported | |
| 20 | D-MPNN | 0.8780
±0.0320 best reported | best at pub: SE(3)-Invariant Multiparameter Persistent Homology for Chiral-Sensitive Molecular Property Prediction (NeurIPS 2023) |
| 43 | GIN | 0.8646
±0.0081 | |
| 45 | AttentiveFP | 0.8630
±0.0150 best reported | best at pub: Molformer: Motif-based Transformer on 3D Heterogeneous Molecular Graphs (AAAI 2021) |
| 58 | GROVER_large | 0.8580 best reported | best at pub: Molformer: Motif-based Transformer on 3D Heterogeneous Molecular Graphs (AAAI 2021) |
| 61 | Uni-Mol | 0.8570
±0.0020 | This paper |
| 62 | GEM | 0.8560
±0.0110 | This paper |
| 83 | PretrainGNN | 0.8450
±0.0070 | This paper |
| 116 | GROVER_base | 0.8260
±0.0070 | This paper |
| 123 | agg + I | 0.8239
±0.0053 | This paper |
| 149 | GraphMVP | 0.8120
±0.0090 | This paper |
| 200 | N-Gram_XGB | 0.7910
±0.0130 | This paper |
| 217 | N-Gram_RF | 0.7790
±0.0150 | 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 | |
| 26 | D-MPNN | 0.9130
±0.0260 best reported | best at pub: SE(3)-Invariant Multiparameter Persistent Homology for Chiral-Sensitive Molecular Property Prediction (NeurIPS 2023) |
| 29 | GROVER_large | 0.9110 best reported | best at pub: Molformer: Motif-based Transformer on 3D Heterogeneous Molecular Graphs (AAAI 2021) |
| 31 | AttentiveFP | 0.9080
±0.0500 best reported | best at pub: Molformer: Motif-based Transformer on 3D Heterogeneous Molecular Graphs (AAAI 2021) |
| 34 | GIN | 0.8875
±0.0049 | |
| 49 | GraphMVP | 0.8600
±0.0340 best reported | best at pub: KPGT: Knowledge-Guided Pre-training of Graph Transformer for Molecular Property Prediction (KDD 2022) |
| 72 | agg + I | 0.7408 | This paper |
| 78 | MolCLR | 0.7360
±0.0050 best reported | best at pub: How Much Structural Information Large Scale Molecular Language Representations Can Capture? (2021) best today: Improving Molecular Contrastive Learning via Faulty Negative Mitigation and Decomposed Fragment Contrast (2022) Difference within reported uncertainty; not counted as underperformance. |
| 84 | Uni-Mol | 0.7290
±0.0060 | This paper |
| 93 | GEM | 0.7240
±0.0040 | This paper |
| 151 | GROVER_base | 0.7000
±0.0010 | This paper |
| 159 | N-Gram_RF | 0.6970
±0.0060 | This paper |
| 175 | N-Gram_XGB | 0.6910
±0.0080 | This paper |
| 186 | PretrainGNN | 0.6870
±0.0130 | This paper |
