arXiv:2412.14497cs.LGcs.AI2024-12被引 1

分离潜在因素的图自编码器提升因果效应估计精度

Disentangled Graph Autoencoder for Treatment Effect Estimation

  • 将潜在因素分解为工具、混杂、调整和噪声四类,独立建模
  • 在多个网络数据集上显著优于现有最先进方法
  • 适合处理有隐藏混杂变量的观测数据因果推断

从观察数据中估计治疗效应在多个研究领域受到广泛关注。然而,许多常用方法依赖于不可观测混杂因素的无混杂假设,这一假设常因无法观测所有混杂因子而不成立。为解决此问题,近期方法利用辅助网络信息推断潜在混杂因子,放宽了该假设。但这些方法通常将观测变量与网络视为潜在混杂因子的代理,当某些变量影响治疗但不影响结果,或反之,这种混淆会导致估计偏差。为此,我们提出一种新型解耦变分图自编码器用于网络化观察数据的治疗效应估计。图编码器将潜在因子解耦为工具、混杂、调整和噪声因子,并使用希尔伯特-施密特独立性准则强制因子独立。在多个网络数据集上的大量实验表明,该方法优于现有最先进方法。

原文摘要 · Abstract (English)

Treatment effect estimation from observational data has attracted significant attention across various research fields. However, many widely used methods rely on the unconfoundedness assumption, which is often unrealistic due to the inability to observe all confounders, thereby overlooking the influence of latent confounders. To address this limitation, recent approaches have utilized auxiliary network information to infer latent confounders, relaxing this assumption. However, these methods often treat observed variables and networks as proxies only for latent confounders, which can result in inaccuracies when certain variables influence treatment without affecting outcomes, or vice versa. This conflation of distinct latent factors undermines the precision of treatment effect estimation. To overcome this challenge, we propose a novel disentangled variational graph autoencoder for treatment effect estimation on networked observational data. Our graph encoder disentangles latent factors into instrumental, confounding, adjustment, and noisy factors, while enforcing factor independence using the Hilbert-Schmidt Independence Criterion. Extensive experiments on multiple networked datasets demonstrate that our method outperforms state-of-the-art approaches.

因果推断图神经网络解耦学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。