Bounds on Representation-Induced Confounding Bias for Treatment Effect Estimation
Valentyn Melnychuk, Dennis Frauen, Stefan Feuerriegel
Abstract
State-of-the-art methods for conditional average treatment effect (CATE) estimation make widespread use of representation learning. Here, the idea is to reduce the variance of the low-sample CATE estimation by a (potentially constrained) low-dimensional representation. However, low-dimensional representations can lose information about the observed confounders and thus lead to bias, because of which the validity of representation learning for CATE estimation is typically violated. In this paper, we propose a new, representation-agnostic refutation framework for estimating bounds on the representation-induced confounding bias that comes from dimensionality reduction (or other constraints on the representations) in CATE estimation. First, we establish theoretically under which conditions CATE is non-identifiable given low-dimensional (constrained) representations. Second, as our remedy, we propose a neural refutation framework which performs partial identification of CATE or, equivalently, aims at estimating lower and upper bounds of the representation-induced confounding bias. We demonstrate the effectiveness of our bounds in a series of experiments. In sum, our refutation framework is of direct relevance in practice where the validity of CATE estimation is of importance.
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 0c7b05a5-a454-4a75-a3c3-b498c90d418dCited by top-tier papers11
- Bayesian Neural Controlled Differential Equations for Treatment Effect EstimationKonstantin Hess, Valentyn Melnychuk, Dennis Frauen, Stefan FeuerriegelICLR 2024 · 27 citations
- DiffPO: A causal diffusion model for learning distributions of potential outcomesYuchen Ma, Valentyn Melnychuk, Jonas Schweisthal, Stefan FeuerriegelNeurIPS 2024 · 23 citations
- Quantifying Aleatoric Uncertainty of the Treatment Effect: A Novel Orthogonal LearnerValentyn Melnychuk, Stefan Feuerriegel, Mihaela van der SchaarNeurIPS 2024 · 13 citations
- IGC-Net for conditional average potential outcome estimation over timeKonstantin Hess, Dennis Frauen, Valentyn Melnychuk, Stefan FeuerriegelICLR 2026 · 8 citations
- DeepBlip: Estimating Conditional Average Treatment Effects Over TimeHaorui Ma, Dennis Frauen, Stefan FeuerriegelICML 2026 · 3 citations
Builds on15
- Estimating counterfactual treatment outcomes over time through adversarially balanced representationsIoana Bica, Ahmed M. Alaa, James Jordon, Mihaela van der SchaarICLR 2020 · 224 citations
- Learning Disentangled Representations for CounterFactual RegressionNegar Hassanpour, Russell GreinerICLR 2020 · 176 citations
- Causal Transformer for Estimating Counterfactual OutcomesValentyn Melnychuk, Dennis Frauen, Stefan FeuerriegelICML 2022 · 146 citations
- On Inductive Biases for Heterogeneous Treatment Effect EstimationAlicia Curth, Mihaela van der SchaarNeurIPS 2021 · 114 citations
- Identifying Causal-Effect Inference Failure with Uncertainty-Aware ModelsAndrew Jesson, Sören Mindermann, Uri Shalit, Yarin GalNeurIPS 2020 · 85 citations
Related papers
- Quantifying Ignorance in Individual-Level Causal-Effect Estimates under Hidden ConfoundingAndrew Jesson, Sören Mindermann, Yarin Gal, Uri ShalitICML 2021 · 66 citations
- Learning Representations of Instruments for Partial Identification of Treatment EffectsJonas Schweisthal, Dennis Frauen, Maresa Schröder, Konstantin Hess et al.ICML 2025
- 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
- Meta-Learners for Partially-Identified Treatment Effects Across Multiple EnvironmentsJonas Schweisthal, Dennis Frauen, Mihaela van der Schaar, Stefan FeuerriegelICML 2024 · 10 citations
- Adjustment for Confounding using Pre-Trained RepresentationsRickmer Schulte, David Rügamer, Thomas NaglerICML 2025
