Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
Wander, a novel graph foundation model, addresses the limited generalization of existing approaches by learning across diverse graph modalities and tasks within a single pretrained checkpoint. It frames graph learning as graph completion via random walks, enabling a unified interface for homogeneous and multi-relational graphs with varying features and schemas. A single Wander checkpoint achieves state-of-the-art performance on node classification, homogeneous link prediction, and knowledge-graph link prediction, demonstrating positive transfer and compositional capabilities.
A single graph foundation model learns to wander across diverse graph structures and tasks, achieving SOTA on node classification, homogeneous link prediction, and KG link prediction.
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.