mlsanityData to 7 Oct 2026

Doing Great at Estimating CATE? On the Neglected Assumptions in Benchmark Comparisons of Treatment Effect Estimators

arXiv 2107.13346 ↗First posted 28 Jul 2021cs.LG stat.MEcited by 4

Not on the map. Why?

Abstract (arXiv)

The machine learning toolbox for estimation of heterogeneous treatment effects from observational data is expanding rapidly, yet many of its algorithms have been evaluated only on a very limited set of semi-synthetic benchmark datasets. In this paper, we show that even in arguably the simplest setting -- estimation under ignorability assumptions -- the results of such empirical evaluations can be misleading if (i) the assumptions underlying the data-generating mechanisms in benchmark datasets and (ii) their interplay with baseline algorithms are inadequately discussed. We consider two popular machine learning benchmark datasets for evaluation of heterogeneous treatment effect estimators -- the IHDP and ACIC2016 datasets -- in detail. We identify problems with their current use and highlight that the inherent characteristics of the benchmark datasets favor some algorithms over others -- a fact that is rarely acknowledged but of immense relevance for interpretation of empirical results. We close by discussing implications and possible next steps.

Key facts. Cited by 4 papers in the map (data through 7 Oct 2026). Builds on 17 of its 45 references that are in the map. First posted on arXiv 28 Jul 2021.

Builds on

17 of 45 references.

References · 17

  1. 2015Estimation and Inference of Heterogeneous Treatment Effects using Random Forestscited by 597
  2. 2017Meta-learners for Estimating Heterogeneous Treatment Effects using Machine Learningcited by 475
  3. 2017Causal Effect Inference with Deep Latent-Variable Modelscited by 408
  4. 2017Quasi-Oracle Estimation of Heterogeneous Treatment Effectscited by 328
  5. 2016Learning Representations for Counterfactual Inferencecited by 327
  6. 2016Generalized Random Forestscited by 309
  7. 2017Bayesian Inference of Individualized Treatment Effects using Multi-task Gaussian Processescited by 156
  8. 2021Nonparametric Estimation of Heterogeneous Treatment Effects: From Theory to Learning Algorithmscited by 144
  9. 2018Troubling Trends in Machine Learning Scholarshipcited by 107
  10. 2017Automated versus do-it-yourself methods for causal inference: Lessons learned from a data analysis competitioncited by 106
  11. 2019Estimating Treatment Effects with Causal Forests: An Applicationcited by 58
  12. 2020Identifying Causal-Effect Inference Failure with Uncertainty-Aware Modelscited by 57
5 more
  1. 2020Counterfactual Representation Learning with Balancing Weightscited by 45
  2. 2020RealCause: Realistic Causal Inference Benchmarkingcited by 44
  3. 2018Learning Weighted Representations for Generalization Across Designscited by 41
  4. 2019Atlantic Causal Inference Conference (ACIC) Data Analysis Challenge 2017cited by 23
  5. 2019Assessing Treatment Effect Variation in Observational Studies: Results from a Data Challengecited by 12

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2021: 1; 2022: 0; 2023: 1; 2024: 1; 2025: 0; 2026: 1.

4 papers.