English

Graph Disentangle Causal Model: Enhancing Causal Inference in Networked Observational Data

Machine Learning 2024-12-06 v1 Information Retrieval

Abstract

Estimating individual treatment effects (ITE) from observational data is a critical task across various domains. However, many existing works on ITE estimation overlook the influence of hidden confounders, which remain unobserved at the individual unit level. To address this limitation, researchers have utilized graph neural networks to aggregate neighbors' features to capture the hidden confounders and mitigate confounding bias by minimizing the discrepancy of confounder representations between the treated and control groups. Despite the success of these approaches, practical scenarios often treat all features as confounders and involve substantial differences in feature distributions between the treated and control groups. Confusing the adjustment and confounder and enforcing strict balance on the confounder representations could potentially undermine the effectiveness of outcome prediction. To mitigate this issue, we propose a novel framework called the \textit{Graph Disentangle Causal model} (GDC) to conduct ITE estimation in the network setting. GDC utilizes a causal disentangle module to separate unit features into adjustment and confounder representations. Then we design a graph aggregation module consisting of three distinct graph aggregators to obtain adjustment, confounder, and counterfactual confounder representations. Finally, a causal constraint module is employed to enforce the disentangled representations as true causal factors. The effectiveness of our proposed method is demonstrated by conducting comprehensive experiments on two networked datasets.

Keywords

Cite

@article{arxiv.2412.03913,
  title  = {Graph Disentangle Causal Model: Enhancing Causal Inference in Networked Observational Data},
  author = {Binbin Hu and Zhicheng An and Zhengwei Wu and Ke Tu and Ziqi Liu and Zhiqiang Zhang and Jun Zhou and Yufei Feng and Jiawei Chen},
  journal= {arXiv preprint arXiv:2412.03913},
  year   = {2024}
}

Comments

Accepted by WSDM 2025

R2 v1 2026-06-28T20:23:50.103Z