mlsanityData to 7 Oct 2026

Meta-Learners for Partially-Identified Treatment Effects Across Multiple Environments

arXiv 2406.02464 ↗First posted 4 Jun 2024cs.LG cs.AI stat.MLcited by 11

Not on the map. Why?

Abstract (arXiv)

Estimating the conditional average treatment effect (CATE) from observational data is relevant for many applications such as personalized medicine. Here, we focus on the widespread setting where the observational data come from multiple environments, such as different hospitals, physicians, or countries. Furthermore, we allow for violations of standard causal assumptions, namely, overlap within the environments and unconfoundedness. To this end, we move away from point identification and focus on partial identification. Specifically, we show that current assumptions from the literature on multiple environments allow us to interpret the environment as an instrumental variable (IV). This allows us to adapt bounds from the IV literature for partial identification of CATE by leveraging treatment assignment mechanisms across environments. Then, we propose different model-agnostic learners (so-called meta-learners) to estimate the bounds that can be used in combination with arbitrary machine learning models. We further demonstrate the effectiveness of our meta-learners across various experiments using both simulated and real-world data. Finally, we discuss the applicability of our meta-learners to partial identification in instrumental variable settings, such as randomized controlled trials with non-compliance.

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

Builds on

38 of 66 references.

References · 38

  1. 2015Estimation and Inference of Heterogeneous Treatment Effects using Random Forestscited by 597
  2. 2015Causal inference using invariant prediction: identification and confidence intervalscited by 533
  3. 2017Quasi-Oracle Estimation of Heterogeneous Treatment Effectscited by 328
  4. 2016Learning Representations for Counterfactual Inferencecited by 327
  5. 2019Adapting Neural Networks for the Estimation of Treatment Effectscited by 285
  6. 2019Orthogonal Statistical Learningcited by 150
  7. 2021Nonparametric Estimation of Heterogeneous Treatment Effects: From Theory to Learning Algorithmscited by 144
  8. 2017Invariant Causal Prediction for Nonlinear Modelscited by 140
  9. 2022Semiparametric doubly robust targeted double machine learning: a reviewcited by 129
  10. 2021On Inductive Biases for Heterogeneous Treatment Effect Estimationcited by 74
  11. 2020Learning Deep Features in Instrumental Variable Regressioncited by 65
  12. 2018Removing Hidden Confounding by Experimental Groundingcited by 64
26 more
  1. 2021Sensitivity Analysis of Individual Treatment Effects: A Robust Conformal Inference Approachcited by 62
  2. 2018Interval Estimation of Individual-Level Causal Effects Under Unobserved Confoundingcited by 51
  3. 2021Quantifying Ignorance in Individual-Level Causal-Effect Estimates under Hidden Confoundingcited by 50
  4. 2021Semiparametric counterfactual density estimationcited by 42
  5. 2021Conformal Sensitivity Analysis for Individual Treatment Effectscited by 40
  6. 2021An Automated Approach to Causal Inference in Discrete Settingscited by 39
  7. 2021Sharp Sensitivity Analysis for Inverse Propensity Weighting via Quantile Balancingcited by 32
  8. 2022Combining Observational and Randomized Data for Estimating Heterogeneous Treatment Effectscited by 32
  9. 2019Machine Learning Estimation of Heterogeneous Treatment Effects with Instrumentscited by 29
  10. 2020A Class of Algorithms for General Instrumental Variable Modelscited by 28
  11. 2020Using Experiments to Correct for Selection in Observational Studiescited by 28
  12. 2021Doubly-Valid/Doubly-Sharp Sensitivity Analysis for Causal Inference with Unmeasured Confoundingcited by 28
  13. 2023B-Learner: Quasi-Oracle Bounds on Heterogeneous Causal Effects Under Hidden Confoundingcited by 28
  14. 2022Long-term Causal Inference Under Persistent Confounding via Data Combinationcited by 27
  15. 2020Invariant Representation Learning for Treatment Effect Estimationcited by 25
  16. 2022Stochastic Causal Programming for Bounding Treatment Effectscited by 25
  17. 2023Sharp Bounds for Generalized Causal Sensitivity Analysiscited by 24
  18. 2022Transfer Learning on Heterogeneous Feature Spaces for Treatment Effects Estimationcited by 21
  19. 2022Estimating individual treatment effects under unobserved confounding using binary instrumentscited by 17
  20. 2022Partial Identification of Treatment Effects with Implicit Generative Modelscited by 17
  21. 2023A Neural Framework for Generalized Causal Sensitivity Analysiscited by 17
  22. 2023Reliable Off-Policy Learning for Dosage Combinationscited by 15
  23. 2022Comparison of meta-learners for estimating multi-valued treatment heterogeneous effectscited by 10
  24. 2023Covariate-assisted bounds on causal effects with instrumental variablescited by 10
  25. 2024Causal Machine Learning for Cost-Effective Allocation of Development Aidcited by 6
  26. 2023A Differential Effect Approach to Partial Identification of Treatment Effectscited by 2

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2024: 3; 2025: 7; 2026: 1.

11 papers.