Meta-Learners for Partially-Identified Treatment Effects Across Multiple Environments
Jonas Schweisthal, Dennis Frauen, Mihaela van der Schaar, Stefan Feuerriegel
Abstract
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.
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 893e7040-c2fa-481e-afc0-3b79e22da417Cited by top-tier papers4
- Orthogonal Survival Learners for Estimating Heterogeneous Treatment Effects from Time-to-Event DataDennis Frauen, Maresa Schröder, Konstantin Hess, Stefan FeuerriegelNeurIPS 2025 · 13 citations
- Constructing Confidence Intervals for Average Treatment Effects from Multiple DatasetsYuxin Wang, Maresa Schröder, Dennis Frauen, Jonas Schweisthal et al.ICLR 2025
- Learning Representations of Instruments for Partial Identification of Treatment EffectsJonas Schweisthal, Dennis Frauen, Maresa Schröder, Konstantin Hess et al.ICML 2025
- Differentially private learners for heterogeneous treatment effectsMaresa Schröder, Valentyn Melnychuk, Stefan FeuerriegelICLR 2025
Builds on14
- On Inductive Biases for Heterogeneous Treatment Effect EstimationAlicia Curth, Mihaela van der SchaarNeurIPS 2021 · 114 citations
- Learning Deep Features in Instrumental Variable RegressionLiyuan Xu, Yutian Chen, Siddarth Srinivasan, Nando de Freitas et al.ICLR 2021 · 85 citations
- Quantifying Ignorance in Individual-Level Causal-Effect Estimates under Hidden ConfoundingAndrew Jesson, Sören Mindermann, Yarin Gal, Uri ShalitICML 2021 · 66 citations
- A Class of Algorithms for General Instrumental Variable ModelsNiki Kilbertus, Matt J. Kusner, Ricardo SilvaNeurIPS 2020 · 41 citations
- B-Learner: Quasi-Oracle Bounds on Heterogeneous Causal Effects Under Hidden ConfoundingMiruna Oprescu, Jacob Dorn, Marah Ghoummaid, Andrew Jesson et al.ICML 2023 · 39 citations
Related papers
- Comparison of meta-learners for estimating multi-valued treatment heterogeneous effectsNaoufal Acharki, Ramiro Lugo, Antoine Bertoncello, Josselin GarnierICML 2023 · 18 citations
- A Non-parametric Direct Learning Approach to Heterogeneous Treatment Effect Estimation under Unmeasured ConfoundingXinhai Zhang, Xingye QiaoNeurIPS 2024 · 1 citation
- Estimating individual treatment effects under unobserved confounding using binary instrumentsDennis Frauen, Stefan FeuerriegelICLR 2023 · 3 citations
- Valid Causal Inference with (Some) Invalid InstrumentsJason S. Hartford, Victor Veitch, Dhanya Sridhar, Kevin Leyton-BrownICML 2021 · 30 citations
- A Meta-learner for Heterogeneous Effects in Difference-in-DifferencesHui Lan, Haoge Chang, Eleanor Wiske Dillon, Vasilis SyrgkanisICML 2025
