论文标题

巴卡迪:带有未知干预措施的贝叶斯因果发现

BaCaDI: Bayesian Causal Discovery with Unknown Interventions

论文作者

Hägele, Alexander, Rothfuss, Jonas, Lorch, Lars, Somnath, Vignesh Ram, Schölkopf, Bernhard, Krause, Andreas

论文摘要

从实验中推断因果结构是许多领域的核心任务。例如,在生物学中,最近的进步使我们能够在多种干预措施(例如药物或基因敲除)下获得单细胞表达数据。但是,干预措施的目标通常是不确定或未知的,并且观察次数有限。结果,不再可靠地使用标准的因果发现方法。为了填补这一空白,我们提出了一个贝叶斯框架(Bacadi),以发现和推理有关在各种未知的实验或介入条件下生成的数据的因果结构。巴卡迪是完全可区分的,这使我们能够通过有效的基于梯度的变异推断来推断干预目标和因果结构上的复杂关节后部。在有关合成因果发现任务和模拟基因表达数据的实验中,巴卡迪在识别因果结构和干预靶标方面优于相关的方法。

Inferring causal structures from experimentation is a central task in many domains. For example, in biology, recent advances allow us to obtain single-cell expression data under multiple interventions such as drugs or gene knockouts. However, the targets of the interventions are often uncertain or unknown and the number of observations limited. As a result, standard causal discovery methods can no longer be reliably used. To fill this gap, we propose a Bayesian framework (BaCaDI) for discovering and reasoning about the causal structure that underlies data generated under various unknown experimental or interventional conditions. BaCaDI is fully differentiable, which allows us to infer the complex joint posterior over the intervention targets and the causal structure via efficient gradient-based variational inference. In experiments on synthetic causal discovery tasks and simulated gene-expression data, BaCaDI outperforms related methods in identifying causal structures and intervention targets.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源