Search papers, labs, and topics across Lattice.
This paper introduces a novel federated de-confounding framework that treats each client as a distinct causal environment to enhance spatio-temporal forecasting. By leveraging the environmental diversity across clients as complementary observations, the proposed method learns a global prototype codebook to capture shared environmental regimes, addressing the limitations of existing personalized approaches. Experimental results show that this framework significantly outperforms traditional federated baselines, offering transferable and interpretable environmental representations while maintaining communication efficiency.
Exploiting client heterogeneity as complementary observations leads to a breakthrough in spatio-temporal forecasting, outperforming traditional methods.
Federated learning has emerged as a promising paradigm for spatio-temporal forecasting (STF), enabling collaborative model training without sharing raw observations. Existing federated STF methods primarily regard cross-client heterogeneity as an optimization challenge and mitigate it through personalized approaches. However, such heterogeneity fundamentally stems from diverse \emph{environmental conditions}, and these methods capture environment-specific forecasting patterns, hardly generalizing under environmental shifts. Our key insight is that the environmental diversity across federated clients should be exploited, as they provide \emph{complementary observations of the same underlying spatio-temporal system}. Based on this insight, we propose \method, a novel federated de-confounding framework that \textbf{treats clients as distinct causal environments}. \method leverages the client heterogeneity as distributed environmental evidence and learns a global prototype codebook to capture shared environmental regimes. We further derive a theoretical federated de-confounding bound that is linearly controlled by the averaged confounding strength. Extensive experiments demonstrate that \method consistently outperforms federated baselines, while providing transferable, interpretable, and communication-efficient environmental representations.
