Debiased Causal Tree: Heterogeneous Treatment Effects Estimation with Unmeasured Confounding
Caizhi Tang, Huiyuan Wang, Xinyu Li, Qing Cui, Ya-Lin Zhang, Feng Zhu, Longfei Li, Jun Zhou, Linbo Jiang
Abstract
Unmeasured confounding poses a significant threat to the validity of causal inference. Despite that various ad hoc methods are developed to remove confounding effects, they are subject to certain fairly strong assumptions. In this work, we consider the estimation of conditional causal effects in the presence of unmeasured confounding using observational data and historical controls. Under an interpretable transportability condition, we prove the partial identifiability of conditional average treatment effect on the treated group (CATT). For tree-based models, a new notion, confounding entropy, is proposed to measure the discrepancy introduced by unobserved confounders between the conditional outcome distribution of the treated and control groups. The confounding entropy generalizes conventional confounding bias, and can be estimated effectively using historical controls. We develop a new method, debiased causal tree, whose splitting rule is to minimize the empirical risk regularized by the confounding entropy. Notably, our method integrates current observational data (for empirical risk) and their historical controls (for confounding entropy) harmoniously. We highlight that, debiased causal tree can not only estimate CATT well in the presence of unmeasured confounding, but also is a robust estimator of conditional average treatment effect (CATE) against the imbalance of the treated and control populations when all confounders are observed. An extension of combining multiple debiased causal trees to further reduce biases by gradient boosting is considered. The computational feasibility and statistical power of our method are evidenced by simulations and a study of a credit card balance dataset.
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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext e4beb067-62a1-4117-993b-53c2b8be5120Cited by top-tier papers3
- Backdoor Adjustment via Group Adaptation for Debiased Coupon RecommendationsJunpeng Fang, Gongduo Zhang, Qing Cui, Caizhi Tang et al.AAAI 2024 · 8 citations
- FAST: a Fused and Accurate Shrinkage Tree for Heterogeneous Treatment Effects EstimationJia Gu, Caizhi Tang, Han Yan, Qing Cui et al.NeurIPS 2023 · 1 citation
- Difference-in-Differences Meets Tree-based Methods: Heterogeneous Treatment Effects Estimation with Unmeasured ConfoundingCaizhi Tang, Huiyuan Wang, Xinyu Li, Qing Cui et al.ICML 2023
Builds on1
Related papers
- Quantifying Ignorance in Individual-Level Causal-Effect Estimates under Hidden ConfoundingAndrew Jesson, Sören Mindermann, Yarin Gal, Uri ShalitICML 2021 · 66 citations
- Automating the Selection of Proxy Variables of Unmeasured ConfoundersFeng Xie, Zhengming Chen, Shanshan Luo, Wang Miao et al.ICML 2024 · 5 citations
- A Non-parametric Direct Learning Approach to Heterogeneous Treatment Effect Estimation under Unmeasured ConfoundingXinhai Zhang, Xingye QiaoNeurIPS 2024 · 1 citation
- Causal Effect Estimation with Mixed Latent Confounders and Post-treatment VariablesYaochen Zhu, Jing Ma, Liang Wu, Qi Guo et al.ICLR 2025
- Treatment Effect Estimation with Disentangled Latent FactorsWeijia Zhang, Lin Liu, Jiuyong LiAAAI 2021 · 115 citations
