Federated Estimation of Causal Effects from Observational Data

Abstract

Many modern applications collect data that comes in federated spirit, with data kept locally and undisclosed. Till date, most insight into the causal inference requires data to be stored in a central repository. We present a novel framework for causal inference with federated data sources. We assess and integrate local causal effects from different private data sources without centralizing them. Then, the treatment effects on subjects from observational data using a non-parametric reformulation of the classical potential outcomes framework is estimated. We model the potential outcomes as a random function distributed by Gaussian processes, whose defining parameters can be efficiently learned from multiple data sources, respecting privacy constraints. We demonstrate the promise and efficiency of the proposed approach through a set of simulated and real-world benchmark examples.

arXiv commentPreprint

References in corpus (7)

• Auto-Encoding Variational Bayes
• Federated Learning for Mobile Keyboard Prediction
• Meta-learners for Estimating Heterogeneous Treatment Effects using Machine Learning
• Causal Effect Inference with Deep Latent-Variable Models
• CausalML: Python Package for Causal Machine Learning
• Orthogonal Random Forest for Causal Inference
• Privacy-Preserving Technology to Help Millions of People: Federated Prediction Model for Stroke Prevention