Learning a Prior-Data Fitted Network for Graph Foundation Models via Structural Causal Model Priors
Organizations: Beihang University Haidian District, Beijing, China · Nankai University Binhai New Area, Tianjin, China · Tsinghua University Haidian District, Beijing, China
Abstract
Graph Foundation Models (GFMs) have recently emerged as a promising paradigm for learning transferable representations across different graphs and domains. However, existing GFMs face notable limitations, including limited pre-training scale or reliance on aligning with textual information in Large Language Models (LLMs). To overcome these limitations, recent studies have drawn inspiration from prior-data fitted networks (PFNs), which learn from large synthetic datasets and support multi-domain prediction backed by causal learning. However, existing graph PFNs encounter severe difficulties in modeling causal relationships within and across node and edge sets, generating diverse graph structures, and designing suitable architectures that can learn from complex synthetic datasets. To address these challenges, we propose GSCM-PFN (Graph SCM-based Prior-Data Fitted Network), a graph PFN pre-trained on synthetic data generated from an SCM-based prior that jointly models nodes and graph topology. Specifically, we (1) design a joint feature-edge-label structural causal model (SCM) to guide synthetic graph generation and capture causal relationships within and across node and edge sets; (2) introduce prior-guided node-edge dependency modeling to instantiate diverse graph generation strategies, and prove that our SCM-based generation prior is more expressive than existing priors; and (3) propose a feature-permutation-invariant dual-path graph network with theoretical guarantees for learning from diverse synthetic graphs. Experiments on 23 real-world benchmarks show that GSCM-PFN outperforms both conventional GFMs and PFN-based GFMs, without utilizing real-world graph data for pre-training. Code: https://anonymous.4open.science/r/GSCM-PFN-A3A3/
Figures & tables
| Dataset | GCN | GAT | GraphAny | TS-Mean | TS-GAT | RGVT | GraphPFN | NodePFN | GSCM-PFN |
| Actor | 28.55 ±0.68 | 27.30 ±0.22 | 28.60 ±0.21 | 28.09 ±0.93 | 28.80 ±2.12 | 32.28 ±0.68 | 34.46 ±1.18 | 32.82 ±1.03 | 35.72 ±1.18 |
| AirBrazil | 42.31 ±7.98 | 57.69 ±14.75 | 34.61 ±16.09 | 39.23 ±5.70 | 36.92 ±10.03 | 86.16 ±5.76 | 67.69 ±5.76 | 72.31 ±6.62 | 72.31 ±2.88 |
| AirEU | 41.88 ±3.60 | 32.50 ±8.45 | 41.50 ±6.50 | 35.88 ±6.91 | 38.38 ±6.32 | 54.75 ±1.75 | 57.75 ±4.87 | 53.50 ±4.53 | 56.50 ±1.29 |
| AirUS | 46.49 ±1.81 | 48.47 ±4.17 | 43.64 ±1.83 | 42.34 ±2.12 | 41.69 ±2.59 | 54.20 ±1.68 | 61.87 ±1.56 | 59.14 ±2.30 | 62.77 ±1.63 |
| Amazon-Comp | 85.83 ±0.86 | 87.01 ±0.50 | 83.04 ±1.24 | 81.37 ±1.25 | 80.23 ±2.44 | 83.85 ±0.54 | 77.62 ±1.40 | 78.44 ±1.64 | 83.11 ±1.57 |
| Amazon-Photo | 91.88 ±0.79 | 91.86 ±1.07 | 90.60 ±0.82 | 90.18 ±1.30 | 89.66 ±1.23 | 91.98 ±0.44 | 87.89 ±2.17 | 87.88 ±1.32 | 91.63 ±1.21 |
| Method | Accuracy |
|---|---|
| Full GSCM-PFN | 72.58 |
| Prior | |
| w/o prior-guided message passing | 68.00 ( 4.58) |
| w/o Decouple | 71.62 ( 0.96) |
| w/o Decouple | 71.25 ( 1.33) |
| w/o multilayer structure | 71.11 ( 1.47) |
Appendix figures & tables1 asset
Supplementary material from the paper’s appendix.
Appendix
| Dataset | #Nodes | #Edges | #Features | #Classes |
| Actor | 7,600 | 26,659 | 932 | 5 |
| AirBrazil | 131 | 1,003 | 131 | 4 |
| AirEU | 399 | 5,993 | 399 | 4 |
| AirUS | 1,190 | 13,599 | 1,190 | 4 |
| Amazon-Comp | 13,752 | 245,861 | 767 | 10 |
| Amazon-Photo | 7,650 | 119,081 | 745 | 8 |
Graph Foundation Models
Graph Foundation Models are pretrained graph-learning systems designed to transfer across datasets, tasks, and domains with different structures and features.
See all 97 papers