Skip to main navigation Skip to search Skip to main content

Prior Refinement Is Better: Diffusion-Driven Graph Harmonization for Federated Graph Learning

  • Shuman Zhuang
  • , Zhihao Wu
  • , Wei Huang
  • , Luojun Lin
  • , Jia Li Yin*
  • , Lele Fu*
  • , Hong Ning Dai
  • *Corresponding author for this work

Research output: Chapter in book/report/conference proceedingConference proceedingpeer-review

Abstract

Federated Graph Learning (FGL) has emerged as a compelling paradigm for collaboratively training a global model while preserving the privacy of multi-source graphs. Nonetheless, FGL faces a critical challenge of data heterogeneity, where semantic and structural discrepancies across clients significantly degrade its performance. Although existing methods attempt to calibrate client-specific graph distributions during federated training, they inevitably fall short in aligning the optimization behaviors across clients due to dynamic parameter updates, thereby inducing a bottleneck in generalization improvement. To tackle this challenge, we propose a solution from a new perspective of prior refinement, which seeks to proactively harmonize client graph distributions before the federated training. In particular, we propose a Federated Graph Harmonization (FedGH) framework that exploits the generative strengths of graph diffusion models to perform prior refinement of local graphs. In a nutshell, FedGH designs a conditional diffusion mechanism on each client that synthesizes pseudo-graphs encapsulating both feature and structural priors, thereby facilitating explicit correction of inter-client distributional bias. On the server side, we employ the graph contrastive learning between various clientspecific pseudo-graphs to incorporate the global information, subsequently guiding local data reconstruction. Importantly, model-agnostic FedGH can be seamlessly deployed as a plugand-play module to be easily integrated with existing FGL architectures. Extensive experiments demonstrate that FedGH consistently outperforms state-of-the-art FGL baselines.

Original languageEnglish
Title of host publicationProceedings of the 40th AAAI Conference on Artificial Intelligence, AAAI 2026
EditorsSven Koenig, Chad Jenkins, Matthew E. Taylor
PublisherAAAI press
Pages29241-29250
Number of pages10
ISBN (Electronic)1577359062 , 9781577359067
DOIs
Publication statusPublished - 17 Mar 2026
Event40th AAAI Conference on Artificial Intelligence, AAAI 2026 - Singapore, Singapore
Duration: 20 Jan 202627 Jan 2026
https://aaai.org/conference/aaai/aaai-26/ (Conference website)
https://aaai.org/conference/aaai/aaai-26/program-overview/ (Conference programme)
https://ojs.aaai.org/index.php/AAAI/index (Conference Proceedings )

Publication series

NameProceedings of the AAAI Conference on Artificial Intelligence
Number34
Volume40
ISSN (Print)2159-5399
ISSN (Electronic)2374-3468

Conference

Conference40th AAAI Conference on Artificial Intelligence, AAAI 2026
Abbreviated titleAAAI 2026
Country/TerritorySingapore
CitySingapore
Period20/01/2627/01/26
Internet address

UN SDGs

This output contributes to the following UN Sustainable Development Goals (SDGs)

  1. SDG 9 - Industry, Innovation, and Infrastructure
    SDG 9 Industry, Innovation, and Infrastructure

Fingerprint

Dive into the research topics of 'Prior Refinement Is Better: Diffusion-Driven Graph Harmonization for Federated Graph Learning'. Together they form a unique fingerprint.

Cite this