Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Fed-CausalDiff: Decoupled Synchronization for Federated Do-Simulation and Policy Evaluation

About

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.

Pengfei Li, Mohammad Khalil• 2026

Related benchmarks

TaskDatasetResultRank
Interventional Fidelity EvaluationDKT-Synth (semi-synthetic)
PEHE0.057
7
Factual FidelityDiabetes
TSTR-AUC0.565
5
Factual FidelityDKT-Synth
TSTR AUC61.7
5
Factual FidelityOpenBandit
TSTR AUC50.5
5
Factual FidelityStatics 2011
TSTR AUC0.562
5
Showing 5 of 5 rows

Other info

Follow for update