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

Identifiability Guarantees for Causal Disentanglement from Soft Interventions

About

Causal disentanglement aims to uncover a representation of data using latent variables that are interrelated through a causal model. Such a representation is identifiable if the latent model that explains the data is unique. In this paper, we focus on the scenario where unpaired observational and interventional data are available, with each intervention changing the mechanism of a latent variable. When the causal variables are fully observed, statistically consistent algorithms have been developed to identify the causal model under faithfulness assumptions. We here show that identifiability can still be achieved with unobserved causal variables, given a generalized notion of faithfulness. Our results guarantee that we can recover the latent causal model up to an equivalence class and predict the effect of unseen combinations of interventions, in the limit of infinite data. We implement our causal disentanglement framework by developing an autoencoding variational Bayes algorithm and apply it to the problem of predicting combinatorial perturbation effects in genomics.

Jiaqi Zhang, Chandler Squires, Kristjan Greenewald, Akash Srivastava, Karthikeyan Shanmugam, Caroline Uhler• 2023

Related benchmarks

TaskDatasetResultRank
Single-cell perturbation predictionhiPSC temporal CRISPR LOO All genes
RMSE0.129
10
Single-cell perturbation predictionE-MTAB-14065 (LOO mode)
Delta Pearson (DE genes)0.3287
10
Single-cell perturbation predictionhiPSC temporal CRISPR LOO, DE genes
AUC-ROC62.2
9
Perturbation predictionNorman 2019 (Single-gene)
RMSE (10 genes)0.5603
6
Perturbation predictionNorman2019 (double-gene)
RMSE (10 genes)0.6084
6
Single-cell perturbation predictionReplogle2022 (single-gene i.i.d.)
L2 Distance10.862
5
Showing 6 of 6 rows

Other info

Code

Follow for update