Can Graph Foundation Models Generalize Over Architecture?
Organizations: University of Oxford · AITHYRA
Abstract
Graph foundation models (GFMs) have recently attracted interest due to the promise of graph neural network (GNN) architectures that generalize zero-shot across graphs of arbitrary scales, feature dimensions, and domains. While existing work has demonstrated this ability empirically across diverse real-world benchmarks, these tasks share a crucial hidden limitation: they admit a narrow set of effective GNN architectures. In particular, current domain-agnostic GFMs rely on fixed architectural backbones, implicitly assuming that a single message-passing regime suffices across tasks. In this paper, we argue that architecture adaptivity is a necessary requirement for true GFMs. We show that existing approaches are non-robust to task-dependent architectural attributes and, as a case study, use range as a minimal and measurable axis along which this limitation becomes explicit. With theoretical analysis and controlled synthetic experiments, we demonstrate that fixed-backbone GFMs provably under-reach on tasks whose architectural requirements differ from those seen at training time. To address this issue, we introduce a framework that adapts effective GNN architecture at inference time by discovering and mixing task-specific linear graph operators, enabling zero-shot generalization across tasks with heterogeneous architectural requirements, without retraining. We validate our approach on arbitrary-range synthetic tasks and a suite of real-world benchmarks, demonstrating improved performance and robustness over existing domain-agnostic GFMs.
Figures & tables
| End-to-end | Zero-shot | ||||
| MeanGNN | GAT | GraphAny | TS-Mean | GOBLIN | |
| Actor ∗ | 32.03 0.29 | 32.59 0.83 | 27.54 0.20 | 28.09 0.93 | 25.36 0.14 |
| AirBrazil | 32.31 7.50 | 35.38 4.21 | 33.84 15.65 | 39.23 5.70 | 50.77 9.55 |
| AirEU | 39.12 6.44 | 39.00 4.30 | 41.25 7.25 | 35.88 6.91 | 41.25 3.95 |
| AirUS | 43.93 1.16 | 43.03 2.08 | 43.35 1.62 | 42.34 2.12 | 51.78 4.43 |
| AmzComp | 73.88 0.88 | 70.94 3.40 | 82.79 1.13 | 81.37 1.25 | 80.67 2.43 |
| MeanGNN | GAT | GraphAny | TS-GAT | GOBLIN | |
| Paris | 20.79 0.25 | 20.73 0.28 | 20.09 0.10 | 16.36 1.73 | 24.59 0.00 |
| Shanghai | 22.86 0.13 | 20.80 0.18 | 19.72 0.02 | 19.93 1.45 | 23.53 0.00 |
| LA | 19.34 0.08 | 18.74 0.25 | 17.19 0.02 | 15.85 1.27 | 24.29 0.00 |
| London | 21.58 0.12 | 21.13 0.15 | 17.13 0.01 | 17.60 1.49 | 21.60 0.00 |
| Average | 21.14 0.14 | 20.35 0.21 | 18.53 0.04 | 17.44 1.49 | 23.50 0.00 |
Appendix figures & tables25 assets
Supplementary material from the paper’s appendix.
Appendix
| Hyperparameter | Search space | Chosen | |
| Operator search | Train dataset | — | Cora |
| BO objective | — | trimmed_20 | |
| sample_budget | 20, 25, 30, 35 | 25 | |
| UCB | 2.0, 3.0, 4.0 | 3.0 | |
| basis_size | 3, 4, 5 | 4 | |
| Diversity penalty | 0.0, 0.1, 0.2, 0.3, 0.4, 0.5 | 0.2 |
| Dataset | MeanGNN | GAT |
| Paris | 27.10 0.67 | 46.75 0.50 |
| Shanghai | 28.64 0.36 | 58.50 0.75 |
| LA | 26.89 0.15 | 53.36 0.59 |
| London | 28.13 0.25 | 47.72 0.25 |
| Dataset | #Nodes | #Edges | #Feature | #Classes | #Labeled Nodes | Diam. | Mean APSPD | Train/Val/Test Ratios (%) | Hetero- philic | Source |
| AirBrazil | 131 | 1074 | 131 | 4 | 80 | 4 | 2.17 | 61.1/19.1/19.8 | Ribeiro et al. (2017) | |
| Cornell | 183 | 554 | 1703 | 5 | 87 | 8 | 3.18 | 47.5/32.2/20.2 | ✓ | Pei et al. (2020) |
| Texas | 183 | 558 | 1703 | 5 | 87 | 8 | 3.02 | 47.5/31.7/20.2 | ✓ | Pei et al. (2020) |
| Wisconsin | 251 | 900 | 1703 | 5 | 120 | 8 | 3.25 | 47.8/31.9/20.3 | ✓ | Pei et al. (2020) |
| AirEU | 399 | 5995 | 399 | 4 | 80 | 5 | 2.32 | 20.1/39.8/40.1 | Ribeiro et al. (2017) | |
| AirUS | 1190 | 13599 | 1190 | 4 | 80 | — | — | 6.7/46.6/46.6 | Ribeiro et al. (2017) |
| Model | Dataset | Hparam search | Train / seed | Inference / seed | |
| GOBLIN (zero-shot) | LA | 240,587 | — | s | s |
| London | 568,795 | — | — | s | |
| Paris | 114,127 | — | — | s | |
| Shanghai | 183,917 | — | — | s | |
| GAT | LA | 240,587 | s | s | ms |
| London | 568,795 | s | s | ms |
| Dataset | Mean geodesic distance | Val. Acc. (%) | ||
| AirBrazil | 0.80 | 1.63 | 2.17 | 50.00 |
| Cornell | 0.80 | 3.22 | 3.18 | 71.19 |
| Texas | 0.80 | 2.79 | 3.02 | 74.14 |
| AirEU | 0.82 | 1.29 | 2.32 | 34.59 |
| Squirrel | 0.91 | 2.34 | 3.09 | 42.31 |
| Wisconsin | 0.80 | 3.18 | 3.24 | 62.50 |
| LinearGNN | (131) | |||||
| 0.00 | 1.36 | 0.30 | 0.37 | 0.43 | 0.37 | |
| 1.00 | 1.75 | 0.35 | 0.35 | 0.34 | 0.35 | |
| 1.61 | 1.84 | 0.35 | 0.28 | 0.23 | 0.28 | |
| 0.50 | 1.84 | — | — | — | — | |
| 0.87 | 1.77 | — | — | — | — |
| LinearGNN | (183) | |||||
| 0.00 | 2.98 | 0.30 | 0.34 | 0.35 | 0.37 | |
| 1.00 | 3.42 | 0.35 | 0.38 | 0.36 | 0.34 | |
| 1.48 | 3.23 | 0.35 | 0.29 | 0.30 | 0.29 | |
| 0.50 | 3.09 | — | — | — | — | |
| 0.86 | 3.18 | — | — | — | — |
| LinearGNN | (183) | |||||
| 0.00 | 2.82 | 0.30 | 0.32 | 0.33 | 0.34 | |
| 1.00 | 2.11 | 0.35 | 0.42 | 0.36 | 0.34 | |
| 1.50 | 3.89 | 0.35 | 0.25 | 0.31 | 0.31 | |
| 0.50 | 2.94 | — | — | — | — | |
| 0.87 | 3.02 | — | — | — | — |
| LinearGNN | (251) | |||||
| 0.00 | 2.94 | 0.30 | 0.34 | 0.34 | 0.36 | |
| 1.00 | 3.28 | 0.35 | 0.38 | 0.36 | 0.34 | |
| 1.48 | 3.32 | 0.35 | 0.29 | 0.30 | 0.30 | |
| 0.50 | 3.06 | — | — | — | — | |
| 0.86 | 3.11 | — | — | — | — |
| LinearGNN | (399) | |||||
| 0.00 | 0.36 | 0.29 | 0.40 | 0.42 | 0.48 | |
| 1.00 | 1.68 | 0.35 | 0.29 | 0.33 | 0.30 | |
| 1.71 | 2.11 | 0.36 | 0.31 | 0.25 | 0.22 | |
| 0.50 | 1.90 | — | — | — | — | |
| 0.91 | 2.29 | — | — | — | — |
| LinearGNN | (500) | |||||
| 0.00 | 1.44 | 0.30 | 0.35 | 0.34 | 0.38 | |
| 1.00 | 2.64 | 0.35 | 0.31 | 0.33 | 0.31 | |
| 1.76 | 2.99 | 0.35 | 0.34 | 0.33 | 0.31 | |
| 0.50 | 2.38 | — | — | — | — | |
| 0.93 | 2.59 | — | — | — | — |
| LinearGNN | (500) | |||||
| 0.00 | 0.92 | 0.30 | 0.38 | 0.37 | 0.41 | |
| 1.00 | 2.38 | 0.35 | 0.33 | 0.34 | 0.33 | |
| 1.89 | 2.42 | 0.35 | 0.29 | 0.29 | 0.26 | |
| 0.50 | 2.37 | — | — | — | — | |
| 0.97 | 2.37 | — | — | — | — |
| LinearGNN | (500) | |||||
| 0.00 | 1.13 | 0.30 | 0.31 | 0.30 | 0.32 | |
| 1.00 | 2.79 | 0.35 | 0.40 | 0.35 | 0.34 | |
| 1.70 | 3.33 | 0.35 | 0.29 | 0.35 | 0.34 | |
| 0.50 | 2.47 | — | — | — | — | |
| 0.90 | 2.74 | — | — | — | — |
| LinearGNN | (500) | |||||
| 0.00 | 1.00 | 0.30 | 0.33 | 0.35 | 0.38 | |
| 1.00 | 2.39 | 0.35 | 0.30 | 0.33 | 0.31 | |
| 1.69 | 3.55 | 0.35 | 0.36 | 0.33 | 0.30 | |
| 0.50 | 2.46 | — | — | — | — | |
| 0.92 | 3.11 | — | — | — | — |
| LinearGNN | (500) | |||||
| 0.00 | 26.63 | 0.31 | 0.32 | 0.32 | 0.35 | |
| 1.00 | 30.65 | 0.36 | 0.34 | 0.34 | 0.33 | |
| 1.37 | 32.10 | 0.34 | 0.34 | 0.35 | 0.32 | |
| 0.50 | 34.24 | — | — | — | — | |
| 0.81 | 30.59 | — | — | — | — |
| LinearGNN | (500) | |||||
| 0.00 | 2.02 | 0.30 | 0.33 | 0.32 | 0.35 | |
| 1.00 | 1.98 | 0.35 | 0.36 | 0.35 | 0.34 | |
| 1.82 | 2.39 | 0.34 | 0.32 | 0.32 | 0.30 | |
| 0.50 | 2.05 | — | — | — | — | |
| 0.95 | 2.05 | — | — | — | — |
| LinearGNN | (500) | |||||
| 0.00 | 2.21 | 0.30 | 0.34 | 0.31 | 0.34 | |
| 1.00 | 4.14 | 0.35 | 0.37 | 0.36 | 0.35 | |
| 1.54 | 4.73 | 0.34 | 0.29 | 0.33 | 0.31 | |
| 0.50 | 4.02 | — | — | — | — | |
| 0.87 | 4.41 | — | — | — | — |
| LinearGNN | (500) | |||||
| 0.00 | 6.02 | 0.30 | 0.34 | 0.35 | 0.39 | |
| 1.00 | 5.63 | 0.35 | 0.34 | 0.35 | 0.33 | |
| 1.71 | 5.69 | 0.34 | 0.32 | 0.30 | 0.28 | |
| 0.50 | 5.83 | — | — | — | — | |
| 0.93 | 5.73 | — | — | — | — |
| LinearGNN | (500) | |||||
| 0.00 | 2088.70 | 0.30 | 0.33 | 0.35 | 0.39 | |
| 1.00 | 2212.81 | 0.35 | 0.26 | 0.29 | 0.28 | |
| 1.12 | 2166.40 | 0.35 | 0.40 | 0.36 | 0.34 | |
| 0.50 | 2173.21 | — | — | — | — | |
| 0.75 | 2179.69 | — | — | — | — |
| LinearGNN | (500) | |||||
| 0.00 | 16.10 | 0.31 | 0.32 | 0.31 | 0.35 | |
| 1.00 | 15.98 | 0.35 | 0.35 | 0.35 | 0.34 | |
| 1.39 | 16.06 | 0.34 | 0.33 | 0.34 | 0.32 | |
| 0.50 | 15.99 | — | — | — | — | |
| 0.80 | 15.81 | — | — | — | — |
| LinearGNN | (500) | |||||
| 0.00 | 0.83 | 0.30 | 0.33 | 0.29 | 0.31 | |
| 1.00 | 3.36 | 0.35 | 0.39 | 0.38 | 0.37 | |
| 1.57 | 4.54 | 0.34 | 0.27 | 0.34 | 0.32 | |
| 0.50 | 3.05 | — | — | — | — | |
| 0.88 | 4.10 | — | — | — | — |
| LinearGNN | (200) | |||||
| 0.00 | 3.34 | 0.30 | 0.32 | 0.32 | 0.35 | |
| 1.00 | 3.46 | 0.35 | 0.35 | 0.35 | 0.34 | |
| 1.82 | 4.07 | 0.34 | 0.33 | 0.33 | 0.31 | |
| 0.50 | 3.84 | — | — | — | — | |
| 0.96 | 3.82 | — | — | — | — |