Causality-inspired Federated Learning for Dynamic Spatio-Temporal Graphs

Authors

  • Yuxuan Liu School of Information and Communication Engineering, University of Electronic Science and Technology of China, China
  • Wenchao Xu Division of Integrative Systems and Design, Hong Kong University of Science and Technology, China
  • Haozhao Wang School of Computer Science and Technology, Huazhong University of Science and Technology, China
  • Zhiming He School of Information and Communication Engineering, University of Electronic Science and Technology of China, China
  • Zhaofeng Shi School of Information and Communication Engineering, University of Electronic Science and Technology of China, China
  • Chongyang Xu School of Information and Communication Engineering, University of Electronic Science and Technology of China, China
  • Peichao Wang School of Information and Communication Engineering, University of Electronic Science and Technology of China, China
  • Boyuan Zhang School of Information and Communication Engineering, University of Electronic Science and Technology of China, China

DOI:

https://doi.org/10.1609/aaai.v40i28.39569

Abstract

Federated Graph Learning (FGL) has emerged as a powerful paradigm for decentralized training of graph neural networks while preserving data privacy. However, existing FGL methods are predominantly designed for static graphs and rely on parameter averaging or distribution alignment, which implicitly assume that all features are equally transferable across clients, overlooking both the spatial and temporal heterogeneity and the presence of client-specific knowledge in real-world graphs. In this work, we identify that such assumptions create a vicious cycle of spurious representation entanglement, client-specific interference, and negative transfer, degrading generalization performance in Federated Learning over Dynamic Spatio-Temporal Graphs (FSTG). To address this issue, we propose a novel causality-inspired framework named SC-FSGL, which explicitly decouples transferable causal knowledge from client-specific noise through representation-level interventions. Specifically, we introduce a Conditional Separation Module that simulates soft interventions through client conditioned masks, enabling the disentanglement of invariant spatio-temporal causal factors from spurious signals and mitigating representation entanglement caused by client heterogeneity. In addition, we propose a Causal Codebook that clusters causal prototypes and aligns local representations via contrastive learning, promoting cross-client consistency and facilitating knowledge sharing across diverse spatio-temporal patterns. Experiments on five diverse heterogeneity Spatio-Temporal Graph (STG) datasets show that SC-FSGL outperforms state-of-the-art methods.

Published

2026-03-14

How to Cite

Liu, Y., Xu, W., Wang, H., He, Z., Shi, Z., Xu, C., … Zhang, B. (2026). Causality-inspired Federated Learning for Dynamic Spatio-Temporal Graphs. Proceedings of the AAAI Conference on Artificial Intelligence, 40(28), 23926–23934. https://doi.org/10.1609/aaai.v40i28.39569

Issue

Section

AAAI Technical Track on Machine Learning V