DeepAI AI Chat
Log In Sign Up

An Adaptive Kernel Approach to Federated Learning of Heterogeneous Causal Effects

by   Thanh Vinh Vo, et al.

We propose a new causal inference framework to learn causal effects from multiple, decentralized data sources in a federated setting. We introduce an adaptive transfer algorithm that learns the similarities among the data sources by utilizing Random Fourier Features to disentangle the loss function into multiple components, each of which is associated with a data source. The data sources may have different distributions; the causal effects are independently and systematically incorporated. The proposed method estimates the similarities among the sources through transfer coefficients, and hence requiring no prior information about the similarity measures. The heterogeneous causal effects can be estimated with no sharing of the raw training data among the sources, thus minimizing the risk of privacy leak. We also provide minimax lower bounds to assess the quality of the parameters learned from the disparate sources. The proposed method is empirically shown to outperform the baselines on decentralized data sources with dissimilar distributions.


page 1

page 2

page 3

page 4


Federated Learning of Causal Effects from Incomplete Observational Data

Decentralized and incomplete data sources are prevalent in real-world ap...

Adaptive Multi-Source Causal Inference

Data scarcity is a tremendous challenge in causal effect estimation. In ...

Collaborative causal inference with a distributed data-sharing management

Data sharing barriers are paramount challenges arising from multicenter ...

Loss Adapted Plasticity in Deep Neural Networks to Learn from Data with Unreliable Sources

When data is streaming from multiple sources, conventional training meth...

Federated Estimation of Causal Effects from Observational Data

Many modern applications collect data that comes in federated spirit, wi...

Transfer Learning for Estimating Causal Effects using Neural Networks

We develop new algorithms for estimating heterogeneous treatment effects...

Do-search – a tool for causal inference and study design with multiple data sources

Epidemiological evidence is based on multiple data sources including cli...