Multi-Domain Riemannian Graph Gluing for Building Graph Foundation Models
Organizations: Beijing University of Posts and Telecommunications, Beijing 100876, China · North China Electric Power University, Beijing 102206, China · University of Illinois Chicago, IL, USA
Abstract
Multi-domain graph pre-training integrates knowledge from diverse domains to enhance performance in the target domains, which is crucial for building graph foundation models. Despite initial success, existing solutions often fall short of answering a fundamental question: how is knowledge integrated or transferred across domains? This theoretical limitation motivates us to rethink the consistency and transferability between model pre-training and domain adaptation. In this paper, we propose a fresh Riemannian geometry perspective, whose core idea is to merge any graph dataset into a unified, smooth Riemannian manifold, enabling a systematic understanding of knowledge integration and transfer. To achieve this, our key contribution is the theoretical establishment of neural manifold gluing, which first characterizes local geometry using an adaptive orthogonal frame and then "glues" the local pieces together into a coherent whole. Building on this theory, we present the GraphGlue framework, which supports batched pre-training with EMA prototyping and provides a transferability measure based on geometric consistence. Extensive experiments demonstrate its superior performance across diverse graph domains. Moreover, we empirically validated GraphGlue's geometric scaling law, showing that larger quantities of datasets improve model transferability by producing a smoother manifold. Codes are available at https://github.com/RiemannGraph/GraphGlue.
Figures & tables
| Model | Node Classification | Link Classification | Graph Classification | |||||||
|---|---|---|---|---|---|---|---|---|---|---|
| Arxiv | Computers | FB15k_237 | PROTEINS | |||||||
| 1-shot | 5-shot | 1-shot | 5-shot | 1-shot | 5-shot | 1-shot | 5-shot | 1-shot | 5-shot | |
| GCN | ||||||||||
| GraphSAGE | ||||||||||
| GIN | ||||||||||
| GCC | ||||||||||
Appendix figures & tables18 assets
Supplementary material from the paper’s appendix.
Appendix
| Notation | Description |
|---|---|
| A smooth manifold | |
| A Riemannian metric tensor | |
| A point in | |
| The tangent space of point p on | |
| The volume element at | |
| A coordinate chart of tangent space |
| Module | Computational complexity | Memory complexity |
|---|---|---|
| -Sparse Perturbation | ||
| Adaptive Orthogonal Frame | ||
| Matrix form of metric tensor | ||
| and | ||
| Riemannian prototypes and | ||
| Riemannian MoE |
| Model | Pretraining | Adaptation (per graph sample) |
|---|---|---|
| PRODIGY | ||
| GFT | ||
| RAGraph | ||
| SAMGPT | ||
| GCOPE | ||
| MDGFM |
| Model | 1 | 2 | 3 | 4 | 5 | 6 |
|---|---|---|---|---|---|---|
| GCOPE | 18.39 | 19.11 | 21.12 | OOM | OOM | OOM |
| MDGFM | 19.71 | 21.67 | 29.35 | OOM | OOM | OOM |
| GraphGlue | 12.53 | 15.07 | 15.73 | 16.87 | 28.67 | 29.21 |
| Domain | Dataset | Task | # Graphs | Avg. #Nodes | Avg. #Edges | # Classes |
|---|---|---|---|---|---|---|
| Citation | PubMed | Node | 1 | 19,717 | 88,648 | 3 |
| Arxiv | Node | 1 | 169,343 | 1,166,243 | 40 | |
| Co-purchase | Computers | Node | 1 | 13,752 | 491,722 | 10 |
| Photo | Node | 1 | 7,650 | 238,162 | 8 | |
| Social Network | Node | 1 | 232,965 | 114,615,892 | 41 | |
| FacebookPagePage | Node | 1 | 22,470 | 342,004 | 4 |
| Variants | Arxiv | Computers | FB15k_237 | PROTEINS | HIV | ||
|---|---|---|---|---|---|---|---|
| 1-shot | w/o EMA | 15.46 1.41 | 30.84 9.50 | 7.43 2.37 | 35.90 17.40 | 58.48 2.56 | 52.62 2.41 |
| w/o L_proto | 9.57 4.56 | 31.24 10.36 | 37.90 9.34 | 43.59 8.13 | 58.49 2.60 | 54.10 2.76 | |
| GRAPHGLUE | 29.73 2.56 | 61.03 7.13 | 68.42 4.68 | 60.89 2.11 | 69.12 4.19 | 58.53 8.20 | |
| 5-shot | w/o EMA | 16.08 2.13 | 34.90 7.77 | 11.53 2.26 | 40.21 12.07 | 59.85 4.53 | 53.87 2.43 |
| w/o L_proto | 15.26 9.91 | 36.63 11.84 | 46.13 14.14 | 64.56 16.42 | 61.64 6.28 | 55.50 2.70 | |
| GRAPHGLUE | 39.98 1.67 | 74.15 2.38 | 84.89 0.68 | 79.52 1.75 | 73.94 2.38 | 62.18 2.50 |
| Method | Amazon-Ratings | Roman-empire | Texas | Wisconsin | |
|---|---|---|---|---|---|
| 1-shot | GCOPE | 28.65 5.82 | 11.44 1.91 | 33.19 6.62 | 31.22 6.85 |
| MDGFM | 29.53 3.45 | 14.51 2.08 | 34.63 10.70 | 35.11 10.53 | |
| GraphGlue | 31.16 3.56 | 16.23 3.00 | 35.16 20.43 | 37.95 10.91 | |
| 5-shot | GCOPE | 30.06 5.11 | 16.00 1.29 | 36.31 10.14 | 38.21 2.96 |
| MDGFM | 30.42 3.80 | 17.15 1.66 | 48.33 6.36 | 47.46 4.86 | |
| GraphGlue | 32.17 2.91 | 18.50 1.07 | 50.88 11.93 | 49.71 8.00 |
| Method | Amazon-Ratings | Roman-empire | Texas | Wisconsin | |
|---|---|---|---|---|---|
| 1-shot | GCOPE | 29.03 4.17 | 13.14 2.36 | 33.82 7.39 | 30.08 5.13 |
| MDGFM | 27.01 2.98 | 14.11 2.14 | 36.02 9.54 | 33.28 8.76 | |
| GraphGlue | 34.12 2.57 | 18.19 2.51 | 38.65 13.88 | 40.17 10.09 | |
| 5-shot | GCOPE | 32.83 3.26 | 16.98 1.40 | 41.33 9.85 | 43.74 3.19 |
| MDGFM | 32.54 3.75 | 16.77 1.92 | 48.10 7.26 | 46.62 4.16 | |
| GraphGlue | 36.26 3.09 | 20.67 1.34 | 52.60 6.07 | 51.49 7.97 |
| Model | Node Classification | Link Classification | Graph Classification | |||
|---|---|---|---|---|---|---|
| Arxiv | Computers | FB15k_237 | PROTEINS | HIV | ||
| GCN | ||||||
| GraphSAGE | ||||||
| GIN | ||||||
| GCC | ||||||
| DGI | ||||||
| Model | Node Classification | Link Classification | Graph Classification | |||
|---|---|---|---|---|---|---|
| Arxiv | Computers | FB15k_237 | PROTEINS | HIV | ||
| GCN | ||||||
| GraphSAGE | ||||||
| GIN | ||||||
| GCC | ||||||
| DGI | ||||||
| Model | Node Classification | Link Classification | Graph Classification | |||
|---|---|---|---|---|---|---|
| Arxiv | Computers | FB15k_237 | PROTEINS | HIV | ||
| GCN | ||||||
| GraphSAGE | ||||||
| GIN | ||||||
| GCC | ||||||
| DGI | ||||||
| Model | Node Classification | Link Classification | Graph Classification | |||
|---|---|---|---|---|---|---|
| Arxiv | Computers | FB15k_237 | PROTEINS | HIV | ||
| GCN | ||||||
| GraphSAGE | ||||||
| GIN | ||||||
| GCC | ||||||
| DGI | ||||||
| Variants | Arxiv | Computers | FB15k_237 | PROTEINS | HIV | ||
|---|---|---|---|---|---|---|---|
| 1-shot | w/o | ||||||
| w/o | |||||||
| GraphGlue | |||||||
| 5-shot | w/o | ||||||
| w/o | |||||||
| GraphGlue |
Graph Foundation Models
Graph Foundation Models are pretrained graph-learning systems designed to transfer across datasets, tasks, and domains with different structures and features.
See all 96 papers