Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Are Common Substructures Transferable? Riemannian Graph Foundation Model with Neural Vector Bundles

About

Foundation models have sparked a revolution via a pretraining-adaptation paradigm, with recent efforts extending this success to graphs. Unlike other modalities, graphs contain rich structural patterns, yet their structural transferability remains poorly understood. Prior studies consider common substructures in the discrete realm, and we are motivated by a fundamental question: Are common substructures transferable? The underlying theory is largely underexplored. In this work, we shift toward learning transferable structures through the lens of functional behavior. Theoretically, we connect transferable substructures to intrinsic geometry of the representation space. However, characterizing such intrinsic geometry has rarely been touched. Grounded in Riemannian geometry, we develop a graph intrinsic geometry learning framework called Neural Vector Bundle, which enables parsing intrinsic geometry with local coordinates. Building on this, we design GAUGE, a pretrainable neural architecture that constructs the vector bundle, flattening geometrically compatible local coordinates, and a new Dirichlet loss, which also measures the transfer effort. We empirically validate its superior expressiveness in challenging tasks including zero-shot link prediction and graph isomorphism.

Li Sun, Zhenhao Huang, Yiding Wang, Qin Chen, Pietro Lio, Philip S. Yu• 2026

Related benchmarks

TaskDatasetResultRank
Node ClassificationPhoto
Mean Accuracy81.33
447
Node ClassificationRoman-Empire
Accuracy26.43
398
Link PredictionPubmed
AUC64.03
173
Node-level classificationFacebook
Accuracy60.04
65
Node ClassificationPubmed
Mean Accuracy71.63
38
Link PredictionFacebook
AUC93.88
10
Link PredictionRoman-Empire
AUC66.22
10
Graph ClassificationCSL--
8
Graph Isomorphism ClassificationMUTAG
AUC88.59
6
Graph Isomorphism ClassificationZINC12K
MAE0.157
6
Showing 10 of 10 rows

Other info

Follow for update