Causality-inspired Federated Learning for Dynamic Spatio-Temporal Graphs
Yuxuan Liu, Wenchao Xu, Haozhao Wang, Zhiming He, Zhaofeng Shi, Chongyang Xu, Peichao Wang, Boyuan Zhang
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.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on26
- GMAN: A Graph Multi-Attention Network for Traffic PredictionChuanpan Zheng, Xiaoliang Fan, Cheng Wang, Jianzhong QiAAAI 2020 · 1,858 citations
- Exploiting Shared Representations for Personalized Federated LearningLiam Collins, Hamed Hassani, Aryan Mokhtari, Sanjay ShakkottaiICML 2021 · 1,081 citations
- Spatio-Temporal Meta-Graph Learning for Traffic ForecastingRenhe Jiang, Zhaonan Wang, Jiawei Yong, Puneet Jeph et al.AAAI 2023 · 336 citations
- Subgraph Federated Learning with Missing Neighbor GenerationKe Zhang, Carl Yang, Xiaoxiao Li, Lichao Sun et al.NeurIPS 2021 · 320 citations
- Federated Graph Classification over Non-IID GraphsHan Xie, Jing Ma, Li Xiong, Carl YangNeurIPS 2021 · 287 citations
Related papers
- Federated Graph Learning via Structure-Aware Fusion Using a Kalman Framework with Learnable DynamicsBisheng Tang, Xiaojun ChenICML 2026
- Less is More: Federated Graph Learning with Alleviating Topology Heterogeneity from A Causal PerspectiveLele Fu, Bowen Deng, Sheng Huang, Tianchi Liao et al.ICML 2025
- Prior Refinement Is Better: Diffusion-Driven Graph Harmonization for Federated Graph LearningShuman Zhuang, Zhihao Wu, Wei Huang, Luojun Lin et al.AAAI 2026
- S2FGL: Spatial Spectral Federated Graph LearningZihan Tan, Suyuan Huang, Guancheng Wan, Wenke Huang et al.ICML 2025
- FedIGL: Federated Invariant Graph Learning for Non-IID GraphsLingren Wang, Wenxuan Tu, Jiaxin Wang, Xiong Wang et al.NeurIPS 2025 · 2 citations
