Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Towards Effective Federated Graph Foundation Model via Mitigating Knowledge Entanglement

About

Recent advances in graph machine learning have shifted to data-centric paradigms, driven by two emerging fields: (1) Federated graph learning (FGL) enables multi-client collaboration but faces challenges from data and task heterogeneity, limiting its practicality; (2) Graph foundation models (GFM) offer strong domain generalization but are usually trained on single machines, missing out on cross-silo data and resources. These paradigms are complementary, and their integration brings notable benefits. Motivated by this, we propose FedGFM, a novel decentralized GFM training paradigm. However, a key challenge is knowledge entanglement, where multi-domain knowledge merges into indistinguishable representations, hindering downstream adaptation. To address this, we present FedGFM+, an enhanced framework with two core modules to reduce knowledge entanglement: (1) AncDAI: A global anchor-based domain-aware initialization strategy. Before pre-training, each client encodes its local graph into domain-specific prototypes that serve as semantic anchors. Synthetic embeddings around these anchors initialize the global model. We theoretically prove these prototypes are distinguishable across domains, providing a strong inductive bias to disentangle domain-specific knowledge. (2) AdaDPP: A local adaptive domain-sensitive prompt pool. Each client learns a lightweight graph prompt capturing domain semantics during pre-training. During fine-tuning, prompts from all clients form a pool from which the GFM selects relevant prompts to augment target graph attributes, improving downstream adaptation. FedGFM+ is evaluated on 8 diverse benchmarks across multiple domains and tasks, outperforming 20 baselines from supervised learning, FGL, and federated GFM variants.

Yinlin Zhu, Xunkai Li, Jishuo Jia, Miao Hu, Di Wu, Meikang Qiu• 2025

Related benchmarks

TaskDatasetResultRank
Node ClassificationPubmed
Accuracy87.17
307
Node ClassificationwikiCS
Accuracy78.69
198
Node ClassificationOgbn-arxiv
Accuracy74.51
191
Graph ClassificationHIV
ROC-AUC0.697
104
Edge classificationFB15K237
Accuracy73.23
17
Edge classificationWN18RR
Accuracy84.31
17
Graph ClassificationPCBA
AUC-ROC74.81
17
Showing 7 of 7 rows

Other info

Follow for update