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

Learning Graph Foundation Models on Riemannian Graph-of-Graphs

About

Graph foundation models (GFMs), pretrained on massive graph data, have transformed graph machine learning by supporting general-purpose reasoning across diverse graph tasks and domains. Existing GFMs pretrained with fixed-hop subgraph sampling impose a fixed receptive field, causing scale mismatch on diverse tasks, which often require heterogeneous and unknown structural contexts beyond a fixed sampling scale. We propose R-GFM, a Riemannian Graph-of-Graphs (GoG) based foundation model, that treats structural scale as a first-class citizen in modeling. R-GFM constructs a multi-scale GoG over-sampled subgraphs at different hop distances and learns geometry-adaptive representations from Riemannian manifolds. Theoretical analysis shows that R-GFM reduces structural domain generalization error compared to fixed-scale GFMs. Experiments on various datasets demonstrate that R-GFM achieves state-of-the-art performance, with up to a 49% relative improvement on downstream tasks. Our code is available at https://github.com/USTC-DataDarknessLab/R-GFM.

Haokun Liu, Zezhong Ding, Xike Xie• 2026

Related benchmarks

TaskDatasetResultRank
Node ClassificationCiteseer
Accuracy73.98
1037
Node ClassificationCornell
Accuracy47.97
900
Node ClassificationWisconsin
Accuracy47.75
898
Node ClassificationTexas
Accuracy0.5164
859
Node ClassificationCora
Accuracy63.77
609
Node ClassificationPubmed
Accuracy63.39
501
Node ClassificationPhoto
Accuracy73.62
285
Link PredictionCiteseer
AUC90.88
174
Link PredictionPubmed
AUC88.62
173
Graph ClassificationHIV
ROC-AUC0.6648
160
Showing 10 of 29 rows

Other info

Follow for update