Search papers, labs, and topics across Lattice.
This paper introduces Fed-CausalDiff, a federated causal diffusion framework designed to enhance interventional inference and policy evaluation in federated learning by addressing the limitations of purely observational methods. By implementing a decoupled synchronization (DSS) approach, the framework allows clients to share a global causal score function while retaining local confounders, effectively managing data heterogeneity. Experimental results across four datasets show that Fed-CausalDiff significantly improves average treatment effect (ATE) and policy-value estimation accuracy, achieving a beneficial balance between communication costs and inference fidelity.
Fed-CausalDiff enables federated learning to perform causal inference with unprecedented accuracy by decoupling global and local data influences.
While federated learning enables collaborative modelling on decentralised data, standard methods merely fit historical observations. This purely observational approach is fundamentally insufficient for interventional inference and policy evaluation, as sequential actions dynamically alter future states. We propose \textbf{Fed-CausalDiff}, a federated causal diffusion framework for do-simulation. The architecture decomposes the evolution of the latent state into a global causal score function and a local confounding score function. This design enables \emph{decoupled synchronisation} (DSS), where clients aggregate only the shared causal mechanism while retaining site-specific confounders locally to handle heterogeneity. Experiments on four datasets demonstrate that Fed-CausalDiff achieves better ATE and policy-value estimation accuracy, offering a favorable trade-off between communication cost and inference fidelity.
