English

Learning Graph Foundation Models on Riemannian Graph-of-Graphs

Machine Learning 2026-05-12 v1

Abstract

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.

Keywords

Cite

@article{arxiv.2605.09993,
  title  = {Learning Graph Foundation Models on Riemannian Graph-of-Graphs},
  author = {Haokun Liu and Zezhong Ding and Xike Xie},
  journal= {arXiv preprint arXiv:2605.09993},
  year   = {2026}
}

Comments

This paper has been accepted by ICML 2026

R2 v1 2026-07-22T07:03:14.883Z