arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2302.00860stat.MLcs.LGstat.ME

基于扩散模型建模因果机制以处理干预与反事实查询

Modeling Causal Mechanisms with Diffusion Models for Interventional and Counterfactual Queries

University of Pennsylvania(宾夕法尼亚大学);Amazon(亚马逊公司)

Patrick Chao, Patrick Blöbaum, Sapan Patel, Shiva Prasad Kasiviswanathan

AI总结:

针对仅含观测数据与因果图的因果充分场景下的多类因果查询问题,提出基于扩散的因果模型学习因果机制,通过潜在编码实现干预采样与反事实溯因,性能优于现有方法并提供通用理论分析框架。

摘要:

我们研究在仅具备观测数据与因果图的因果充分场景下,解答观测性、干预性与反事实查询的问题。借助扩散模型的最新进展,我们提出基于扩散的因果模型(DCM)用于学习因果机制,该模型可生成独特的潜在编码。这些编码使我们能够在干预下直接采样,并为反事实推理执行溯因操作。扩散模型天然适用于该场景,因为它们可将每个节点编码为潜在表示,作为外生噪声的代理。我们的实证评估表明,在解答因果查询方面,该方法相较现有最优方法有显著提升。此外,我们提供了理论结果,给出了分析通用编码器-解码器模型中反事实估计的方法,这在我们提出的方法之外的场景中也可能有用。

英文摘要:

We consider the problem of answering observational, interventional, and counterfactual queries in a causally sufficient setting where only observational data and the causal graph are available. Utilizing the recent developments in diffusion models, we introduce diffusion-based causal models (DCM) to learn causal mechanisms, that generate unique latent encodings. These encodings enable us to directly sample under interventions and perform abduction for counterfactuals. Diffusion models are a natural fit here, since they can encode each node to a latent representation that acts as a proxy for exogenous noise. Our empirical evaluations demonstrate significant improvements over existing state-of-the-art methods for answering causal queries. Furthermore, we provide theoretical results that offer a methodology for analyzing counterfactual estimation in general encoder-decoder models, which could be useful in settings beyond our proposed approach.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑