To Learn is to Wander: Learning Across Graphs and Tasks with Random Walks
cs.LG
Submitted: 2026-10-05
Updated: 2026-10-06
Code: https://github.com/louistichelman/wander-gfm
License: http://creativecommons.org/licenses/by/4.0/
The gist: Graph foundation models aim to transfer across graphs, feature spaces, relational schemas, and prediction tasks, yet existing approaches typically generalize only within particular graph modalities
Abstract
Graph foundation models aim to transfer across graphs, feature spaces, relational schemas, and prediction tasks, yet existing approaches typically generalize only within particular graph modalities or tasks. We propose Wander, a graph foundation model designed to operate across these settings within a single pretrained checkpoint. Following the prior-predictive perspective, we formulate graph learning as completion of a partially observed graph. We realize this task-general view through a common interface based on random walks, allowing the same model to operate across homogeneous and multi-relational graphs with varying features, labels, and relational schemas. Wander can increase its structural context at inference time without changing its learned parameters and, under suitable assumptions, universally approximates the corresponding Bayes-optimal predictor on bounded connected graphs. Empirically, a single pretrained checkpoint achieves state-of-the-art or highly competitive results across node classification, homogeneous link prediction, and knowledge-graph link prediction. Moreover, joint pretraining across graph modalities and tasks preserves performance in specialized settings while enabling positive transfer and the composition of separately learned capabilities at inference time.
Sources
- Universal Link Predictor By In-Context Learning on Graphs
- Turning Tabular Foundation Models into Graph Foundation Models
- TabPFN-3: Technical Report
- Bringing Graphs to the Table: Zero-shot Node Classification via Tabular Foundation Models
- TFMLinker: Universal Link Predictor by Graph In-Context Learning with Tabular Foundation Models
- Wiki-CS: A Wikipedia-Based Benchmark for Graph Neural Networks
- GLU Variants Improve Transformer
- Pitfalls of Graph Neural Network Evaluation
- LimiX: Unleashing Structured-Data Modeling Capability for Generalist Intelligence
- A Multi-Task Perspective for Link Prediction with New Relation Types and Nodes
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks