mlsanityData to 7 Oct 2026

Estimating Heterogeneous Treatment Effects by Combining Weak Instruments and Observational Data

arXiv 2406.06452 ↗First posted 10 Jun 2024stat.ME cs.LG stat.MLcited by 2

Not on the map. Why?

Abstract (arXiv)

Accurately predicting conditional average treatment effects (CATEs) is crucial in personalized medicine and digital platform analytics. Since the treatments of interest often cannot be directly randomized, observational data is leveraged to learn CATEs, but this approach can incur significant bias from unobserved confounding. One strategy to overcome these limitations is to leverage instrumental variables (IVs) as latent quasi-experiments, such as randomized intent-to-treat assignments or randomized product recommendations. This approach, on the other hand, can suffer from low compliance, textiti.e., IV weakness. Some subgroups may even exhibit zero compliance, meaning we cannot instrument for their CATEs at all. In this paper, we develop a novel approach to combine IV and observational data to enable reliable CATE estimation in the presence of unobserved confounding in the observational data and low compliance in the IV data, including no compliance for some subgroups. We propose a two-stage framework that first learns textitbiased CATEs from the observational data, and then applies a compliance-weighted correction using IV data, effectively leveraging IV strength variability across covariates. We characterize the convergence rates of our method and validate its effectiveness through a simulation study. Additionally, we demonstrate its utility with real data by analyzing the heterogeneous effects of 401(k) plan participation on wealth.

Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 38 of its 57 references that are in the map. First posted on arXiv 10 Jun 2024.

Builds on

38 of 57 references.

References · 38

  1. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
  2. 2017A Unified Approach to Interpreting Model Predictionscited by 6,488
  3. 2015Estimation and Inference of Heterogeneous Treatment Effects using Random Forestscited by 597
  4. 2016Error bounds for approximations with deep ReLU networkscited by 567
  5. 2017Meta-learners for Estimating Heterogeneous Treatment Effects using Machine Learningcited by 475
  6. 2017Causal Effect Inference with Deep Latent-Variable Modelscited by 408
  7. 2017Quasi-Oracle Estimation of Heterogeneous Treatment Effectscited by 328
  8. 2016Learning Representations for Counterfactual Inferencecited by 327
  9. 2017Size-Independent Sample Complexity of Neural Networkscited by 289
  10. 2019Adapting Neural Networks for the Estimation of Treatment Effectscited by 285
  11. 2019Orthogonal Statistical Learningcited by 150
  12. 2019Kernel Instrumental Variable Regressioncited by 118
26 more
  1. 2018The Blessings of Multiple Causescited by 103
  2. 2019Deep Generalized Method of Moments for Instrumental Variable Analysiscited by 88
  3. 2018Hyperparameters and Tuning Strategies for Random Forestcited by 83
  4. 2021Discussion on "Bayesian Regression Tree Models for Causal Inference: Regularization, Confounding, and Heterogeneous Effects" by Hahn, Murray and Carvalhocited by 77
  5. 2019Time Series Deconfounder: Estimating Treatment Effects over Time in the Presence of Hidden Confounderscited by 74
  6. 2018Confounding-Robust Policy Improvementcited by 73
  7. 2020Learning Deep Features in Instrumental Variable Regressioncited by 65
  8. 2018Removing Hidden Confounding by Experimental Groundingcited by 64
  9. 2018Orthogonal Random Forest for Causal Inferencecited by 51
  10. 2022Combining Observational and Randomized Data for Estimating Heterogeneous Treatment Effectscited by 32
  11. 2018Combining multiple observational data sources to estimate causal effectscited by 29
  12. 2019Machine Learning Estimation of Heterogeneous Treatment Effects with Instrumentscited by 29
  13. 2020Combining Observational and Experimental Datasets Using Shrinkage Estimatorscited by 28
  14. 2020Using Experiments to Correct for Selection in Observational Studiescited by 28
  15. 2023B-Learner: Quasi-Oracle Bounds on Heterogeneous Causal Effects Under Hidden Confoundingcited by 28
  16. 2022Long-term Causal Inference Under Persistent Confounding via Data Combinationcited by 27
  17. 2021Sequential Deconfounding for Causal Inference with Unobserved Confounderscited by 24
  18. 2023Sharp Bounds for Generalized Causal Sensitivity Analysiscited by 24
  19. 2016Bounded, efficient and multiply robust estimation of average treatment effects using instrumental variablescited by 23
  20. 2021DoubleML -- An Object-Oriented Implementation of Double Machine Learning in Pythoncited by 23
  21. 2014Instrumental Variables Estimation with Some Invalid Instruments and its Application to Mendelian Randomizationcited by 22
  22. 2018Sharp instruments for classifying compliers and generalizing causal effectscited by 22
  23. 2021Adaptive Combination of Randomized and Observational Datacited by 22
  24. 2021Deconfounding Temporal Autoencoder: Estimating Treatment Effects over Time Using Noisy Proxiescited by 16
  25. 2022On the instrumental variable estimation with many weak and invalid instrumentscited by 2
  26. 2021Improving Inference from Simple Instruments through Compliance Estimationcited by 1

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2025: 1; 2026: 1.

2 papers.