On Heterogeneous Treatment Effects in Heterogeneous Causal Graphs
Richard A. Watson, Hengrui Cai, Xinming An, Samuel A. McLean, Rui Song
Abstract
Heterogeneity and comorbidity are two interwoven challenges associated with various healthcare problems that greatly hampered research on developing effective treatment and understanding of the underlying neurobiological mechanism. Very few studies have been conducted to investigate heterogeneous causal effects (HCEs) in graphical contexts due to the lack of statistical methods. To characterize this heterogeneity, we first conceptualize heterogeneous causal graphs (HCGs) by generalizing the causal graphical model with confounder-based interactions and multiple mediators. Such confounders with an interaction with the treatment are known as moderators. This allows us to flexibly produce HCGs given different moderators and explicitly characterize HCEs from the treatment or potential mediators on the outcome. We establish the theoretical forms of HCEs and derive their properties at the individual level in both linear and nonlinear models. An interactive structural learning is developed to estimate the complex HCGs and HCEs with confidence intervals provided. Our method is empirically justified by extensive simulations and its practical usefulness is illustrated by exploring causality among psychiatric disorders for trauma survivors. Code implementing the proposed algorithm is open-source and publicly available at: https: //github.com/richard-watson/ISL .
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers1
Ask how each one uses itBuilds on3
- Causal Discovery with Reinforcement LearningShengyu Zhu, Ignavier Ng, Zhitang ChenICLR 2020 · 285 citations
- Score Matching Enables Causal Discovery of Nonlinear Additive Noise ModelsPaul Rolland, Volkan Cevher, Matthäus Kleindessner, Chris Russell et al.ICML 2022 · 123 citations
- ANOCE: Analysis of Causal Effects with Multiple Mediators via Constrained Structural LearningHengrui Cai, Rui Song, Wenbin LuICLR 2021 · 3 citations
Related papers
- Visual Analysis of Multi-Outcome Causal GraphsMengjie Fan, Jinlu Yu, Daniel Weiskopf, Nan Cao et al.IEEE VIS 2024 · 6 citations
- Decoding Causal Structure: End-to-End Mediation Pathways InferenceYulong Li, Xiwei Liu, Feilong Tang, Ming Hu et al.NeurIPS 2025 · 3 citations
- Improving Data-driven Heterogeneous Treatment Effect Estimation Under Structure UncertaintyChristopher Tran, Elena ZhelevaKDD 2022 · 4 citations
- Deep Multi-Modal Structural Equations For Causal Effect Estimation With Unstructured ProxiesShachi Deshpande, Kaiwen Wang, Dhruv Sreenivas, Zheng Li et al.NeurIPS 2022 · 15 citations
- Beyond Entity Correlations: Disentangling Event Causal Puzzles in Temporal Knowledge GraphsQian Chen, Jinyu Zhang, Ling ChenICLR 2026
