mlsanityData to 14 Oct 2026

Estimating individual treatment effects under unobserved confounding using binary instruments

arXiv 2208.08544 ↗First posted 17 Aug 2022stat.ME cs.LG stat.MLcited by 18

Not on the map. Why?

Abstract (arXiv)

Estimating conditional average treatment effects (CATEs) from observational data is relevant in many fields such as personalized medicine. However, in practice, the treatment assignment is usually confounded by unobserved variables and thus introduces bias. A remedy to remove the bias is the use of instrumental variables (IVs). Such settings are widespread in medicine (e.g., trials where the treatment assignment is used as binary IV). In this paper, we propose a novel, multiply robust machine learning framework, called MRIV, for estimating CATEs using binary IVs and thus yield an unbiased CATE estimator. Different from previous work for binary IVs, our framework estimates the CATE directly via a pseudo outcome regression. (1)~We provide a theoretical analysis where we show that our framework yields multiple robust convergence rates: our CATE estimator achieves fast convergence even if several nuisance estimators converge slowly. (2)~We further show that our framework asymptotically outperforms state-of-the-art plug-in IV methods for CATE estimation, in the sense that it achieves a faster rate of convergence if the CATE is smoother than the individual outcome surfaces. (3)~We build upon our theoretical results and propose a tailored deep neural network architecture called MRIV-Net for CATE estimation using binary IVs. Across various computational experiments, we demonstrate empirically that our MRIV-Net achieves state-of-the-art performance. To the best of our knowledge, our MRIV is the first multiply robust machine learning framework tailored to estimating CATEs in the binary IV setting.

Key facts. Cited by 18 papers in the map (data through 14 Oct 2026). Builds on 26 of its 60 references that are in the map. First posted on arXiv 17 Aug 2022.

Builds on

26 of 60 references.

References · 26

  1. 2014Adam: A Method for Stochastic Optimizationcited by 37,240
  2. 2015Estimation and Inference of Heterogeneous Treatment Effects using Random Forestscited by 600
  3. 2017Meta-learners for Estimating Heterogeneous Treatment Effects using Machine Learningcited by 478
  4. 2017Training GANs with Optimismcited by 217
  5. 2017Bayesian Inference of Individualized Treatment Effects using Multi-task Gaussian Processescited by 158
  6. 2020Estimating Counterfactual Treatment Outcomes over Time Through Adversarially Balanced Representationscited by 155
  7. 2021Nonparametric Estimation of Heterogeneous Treatment Effects: From Theory to Learning Algorithmscited by 144
  8. 2022Semiparametric doubly robust targeted double machine learning: a reviewcited by 133
  9. 2022Causal Transformer for Estimating Counterfactual Outcomescited by 123
  10. 2019Kernel Instrumental Variable Regressioncited by 118
  11. 2018The Blessings of Multiple Causescited by 104
  12. 2019Deep Generalized Method of Moments for Instrumental Variable Analysiscited by 88
14 more
  1. 2019Time Series Deconfounder: Estimating Treatment Effects over Time in the Presence of Hidden Confounderscited by 75
  2. 2020Learning Deep Features in Instrumental Variable Regressioncited by 66
  3. 2020Semiparametric proximal causal inferencecited by 59
  4. 2018Interval Estimation of Individual-Level Causal Effects Under Unobserved Confoundingcited by 51
  5. 2021Quantifying Ignorance in Individual-Level Causal-Effect Estimates under Hidden Confoundingcited by 50
  6. 2021Deep Proxy Causal Learning and its Application to Confounded Bandit Policy Evaluationcited by 40
  7. 2019Machine Learning Estimation of Heterogeneous Treatment Effects with Instrumentscited by 29
  8. 2022Estimating average causal effects from patient trajectoriescited by 29
  9. 2021Sequential Deconfounding for Causal Inference with Unobserved Confounderscited by 24
  10. 2016Bounded, efficient and multiply robust estimation of average treatment effects using instrumental variablescited by 23
  11. 2019A semiparametric instrumental variable approach to optimal treatment regimes under endogeneitycited by 22
  12. 2019Heterogeneous causal effects with imperfect compliance: a Bayesian machine learning approachcited by 6
  13. 2019Double Robustness for Complier Parameters and a Semiparametric Test for Complier Characteristicscited by 4
  14. 2016Robust causal inference with continuous instruments using the local instrumental variable curvecited by 2

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2023: 4; 2024: 4; 2025: 8; 2026: 2.

18 papers.

Sort:

  1. 2024Causal machine learning for predicting treatment outcomescited by 74
  2. 2025Foundation Models for Causal Inference via Prior-Data Fitted Networkscited by 32
  3. 2023Sharp Bounds for Generalized Causal Sensitivity Analysiscited by 24
  4. 2024Model-agnostic meta-learners for estimating heterogeneous treatment effects over timecited by 19
  5. 2023Zero-shot causal learningcited by 17
  6. 2023Partial Counterfactual Identification of Continuous Outcomes with a Curvature Sensitivity Modelcited by 14
  7. 2024Meta-Learners for Partially-Identified Treatment Effects Across Multiple Environmentscited by 11
  8. 2025Orthogonal Survival Learners for Estimating Heterogeneous Treatment Effects from Time-to-Event Datacited by 10
  9. 2023Adaptive Instrument Design for Indirect Experimentscited by 6
  10. 2024Learning Representations of Instruments for Partial Identification of Treatment Effectscited by 4
  11. 2025Improving the Generation and Evaluation of Synthetic Data for Downstream Medical Causal Inferencecited by 4
  12. 2025Nonparametric Heterogeneous Long-term Causal Effect Estimation via Data Combinationcited by 2
6 more
  1. 2025Long-term Causal Inference via Modeling Sequential Latent Confoundingcited by 2
  2. 2025Efficient Adaptive Experimentation with Noncompliancecited by 2
  3. 2025CausalKANs: interpretable treatment effect estimation with Kolmogorov-Arnold networkscited by 2
  4. 2025Assessment of the conditional exchangeability assumption in causal machine learning models: a simulation studycited by 0
  5. 2026Treatment Effect Estimation with Differentiated Networked Effect on Graph Datacited by 0
  6. 2026A Guide to Estimating Conditional Average Treatment Effects in Competing Risks Settingscited by 0