mlsanityData to 7 Oct 2026

Sources of Gain: Decomposing Performance in Conditional Average Dose Response Estimation

arXiv 2406.08206 ↗First posted 12 Jun 2024cs.LGcited by 2

Not on the map. Why?

Abstract (arXiv)

Estimating conditional average dose responses (CADR) is an important but challenging problem. Estimators must correctly model the potentially complex relationships between covariates, interventions, doses, and outcomes. In recent years, the machine learning community has shown great interest in developing tailored CADR estimators that target specific challenges. Their performance is typically evaluated against other methods on (semi-) synthetic benchmark datasets. Our paper analyses this practice and shows that using popular benchmark datasets without further analysis is insufficient to judge model performance. Established benchmarks entail multiple challenges, whose impacts must be disentangled. Therefore, we propose a novel decomposition scheme that allows the evaluation of the impact of five distinct components contributing to CADR estimator performance. We apply this scheme to eight popular CADR estimators on four widely-used benchmark datasets, running nearly 1,500 individual experiments. Our results reveal that most established benchmarks are challenging for reasons different from their creators' claims. Notably, confounding, the key challenge tackled by most estimators, is not an issue in any of the considered datasets. We discuss the major implications of our findings and present directions for future research.

Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 23 of its 79 references that are in the map. First posted on arXiv 12 Jun 2024.

Builds on

23 of 79 references.

References · 23

  1. 2016XGBoost: A Scalable Tree Boosting Systemcited by 5,120
  2. 2015Estimation and Inference of Heterogeneous Treatment Effects using Random Forestscited by 597
  3. 2019Generalizing from a Few Examples: A Survey on Few-Shot Learningcited by 462
  4. 2017Causal Effect Inference with Deep Latent-Variable Modelscited by 408
  5. 2016Learning Representations for Counterfactual Inferencecited by 327
  6. 2019Adapting Neural Networks for the Estimation of Treatment Effectscited by 285
  7. 2022OpenOOD: Benchmarking Generalized Out-of-Distribution Detectioncited by 267
  8. 2019Learning Counterfactual Representations for Estimating Individual Dose-Response Curvescited by 90
  9. 2021OoD-Bench: Quantifying and Understanding Two Dimensions of Out-of-Distribution Generalizationcited by 86
  10. 2020Estimating the Effects of Continuous-valued Interventions using Generative Adversarial Networkscited by 78
  11. 2020Generalization Bounds and Representation Learning for Estimation of Potential Outcomes and Causal Effectscited by 72
  12. 2021VCNet and Functional Targeted Regularization For Learning Causal Effects of Continuous Treatmentscited by 59
11 more
  1. 2020k-Nearest Neighbour Classifiers: 2nd Edition (with Python examples)cited by 56
  2. 2020Kernel Methods for Causal Functions: Dose, Heterogeneous, and Incremental Response Curvescited by 36
  3. 2023In Search of Insights, Not Magic Bullets: Towards Demystification of the Model Selection Dilemma in Heterogeneous Treatment Effect Estimationcited by 29
  4. 2018A comparison of methods for model selection when estimating individual treatment effectscited by 28
  5. 2022Exploring Transformer Backbones for Heterogeneous Treatment Effect Estimationcited by 25
  6. 2023Reliable Off-Policy Learning for Dosage Combinationscited by 15
  7. 2020Entropy Balancing for Continuous Treatmentscited by 12
  8. 2021Machine learning on small size samples: A synthetic knowledge synthesiscited by 11
  9. 2023Adversarially Balanced Representation for Continuous Treatment Effect Estimationcited by 10
  10. 2024Continuous Treatment Effect Estimation Using Gradient Interpolation and Kernel Smoothingcited by 7
  11. 2024Contrastive Balancing Representation Learning for Heterogeneous Dose-Response Curves Estimationcited by 6

Built on by

Cited by year

Citing papers per year: 2024: 1; 2025: 1.

2 papers.