mlsanityData to 7 Oct 2026

Bounds on Representation-Induced Confounding Bias for Treatment Effect Estimation

arXiv 2311.11321 ↗First posted 19 Nov 2023stat.ML cs.AI cs.LGcited by 20

Not on the map. Why?

Abstract (arXiv)

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.

Key facts. Cited by 20 papers in the map (data through 7 Oct 2026). Builds on 34 of its 67 references that are in the map. First posted on arXiv 19 Nov 2023.

Builds on

34 of 67 references.

References · 34

  1. 2015Estimation and Inference of Heterogeneous Treatment Effects using Random Forestscited by 597
  2. 2019Neural Spline Flowscited by 590
  3. 2017Meta-learners for Estimating Heterogeneous Treatment Effects using Machine Learningcited by 475
  4. 2017Quasi-Oracle Estimation of Heterogeneous Treatment Effectscited by 328
  5. 2016Learning Representations for Counterfactual Inferencecited by 327
  6. 2017Bayesian Inference of Individualized Treatment Effects using Multi-task Gaussian Processescited by 156
  7. 2020Estimating Counterfactual Treatment Outcomes over Time Through Adversarially Balanced Representationscited by 154
  8. 2021Nonparametric Estimation of Heterogeneous Treatment Effects: From Theory to Learning Algorithmscited by 144
  9. 2022Causal Transformer for Estimating Counterfactual Outcomescited by 121
  10. 2017Causal Consistency of Structural Equation Modelscited by 114
  11. 2017Double/Debiased/Neyman Machine Learning of Treatment Effectscited by 76
  12. 2021On Inductive Biases for Heterogeneous Treatment Effect Estimationcited by 74
22 more
  1. 2020Generalization Bounds and Representation Learning for Estimation of Potential Outcomes and Causal Effectscited by 72
  2. 2022Continuous-Time Modeling of Counterfactual Outcomes Using Neural Controlled Differential Equationscited by 59
  3. 2020Identifying Causal-Effect Inference Failure with Uncertainty-Aware Modelscited by 57
  4. 2018Perfect Match: A Simple Method for Learning Representations For Counterfactual Inference With Neural Networkscited by 55
  5. 2018Conditional Density Estimation with Bayesian Normalising Flowscited by 54
  6. 2018Interval Estimation of Individual-Level Causal Effects Under Unobserved Confoundingcited by 51
  7. 2021Quantifying Ignorance in Individual-Level Causal-Effect Estimates under Hidden Confoundingcited by 50
  8. 2020Counterfactual Representation Learning with Balancing Weightscited by 45
  9. 2018Learning Weighted Representations for Generalization Across Designscited by 41
  10. 2022Causal Effect Identification in Cluster DAGscited by 33
  11. 2021Sharp Sensitivity Analysis for Inverse Propensity Weighting via Quantile Balancingcited by 32
  12. 2023Optimal Transport for Treatment Effect Estimationcited by 32
  13. 2019Adversarial Balancing-based Representation Learning for Causal Effect Inference with Observational Datacited by 29
  14. 2023B-Learner: Quasi-Oracle Bounds on Heterogeneous Causal Effects Under Hidden Confoundingcited by 28
  15. 2023Bayesian Neural Controlled Differential Equations for Treatment Effect Estimationcited by 26
  16. 2023Sharp Bounds for Generalized Causal Sensitivity Analysiscited by 24
  17. 2019Noise Regularization for Conditional Density Estimationcited by 23
  18. 2020Counterfactual Predictions under Runtime Confoundingcited by 22
  19. 2017Bayesian Nonparametric Causal Inference: Information Rates and Learning Algorithmscited by 16
  20. 2022Treatment Effect Risk: Bounds and Inferencecited by 12
  21. 2022Sensitivity Analysis for Marginal Structural Modelscited by 10
  22. 2023Data-Driven Allocation of Preventive Care With Application to Diabetes Mellitus Type IIcited by 5

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2024: 9; 2025: 7; 2026: 4.

20 papers.

Sort:

By area:

  1. 2024Model-agnostic meta-learners for estimating heterogeneous treatment effects over timecited by 18
  2. 2024DiffPO: A causal diffusion model for learning distributions of potential outcomescited by 17
  3. 2024IGC-Net for conditional average potential outcome estimation over timecited by 10
  4. 2024Quantifying Aleatoric Uncertainty of the Treatment Effect: A Novel Orthogonal Learnercited by 10
  5. 2024MAVias: Mitigate any Visual Biascited by 5
  6. 2025Orthogonal Representation Learning for Estimating Causal Quantitiescited by 4
  7. 2024Uncertainty-Aware Optimal Treatment Selection for Clinical Time Seriescited by 3
  8. 2024DAG-aware Transformer for Causal Effect Estimationcited by 2
  9. 2024Towards Representation Learning for Weighting Problems in Design-Based Causal Inferencecited by 1
  10. 2025Overlap-Adaptive Regularization for Conditional Average Treatment Effect Estimationcited by 1
  11. 2025DeepBlip: Estimating Conditional Average Treatment Effects Over Timecited by 1
  12. 2025Cluster-Dags as Powerful Background Knowledge For Causal Discoverycited by 1
8 more
  1. 2026CAETC: Causal Autoencoding and Treatment Conditioning for Counterfactual Estimation over Timecited by 1
  2. 2024From Text to Treatment Effects: A Meta-Learning Approach to Handling Text-Based Confoundingcited by 0
  3. 2025Conditional Average Treatment Effect Estimation Under Hidden Confounderscited by 0
  4. 2025Annotation-Assisted Learning of Treatment Policies From Multimodal Electronic Health Recordscited by 0
  5. 2025Estimating Treatment Effects in Networks under Unknown Exposure Mappingscited by 0
  6. 2026Causal Representation Learning with Optimal Compression under Complex Treatmentscited by 0
  7. 2026Causal State-Space Model for Causal Inference: Estimating Longitudinal Individual Treatment Effectscited by 0
  8. 2026Admissible Diffusion for Multimodal Interventional Trajectoriescited by 0