A Data-Centric Decomposition of Estimator Performance in Continuous Treatment Effect Estimation
Christopher Bockel-Rickermann, Daan Caljon, Toon Vanderschueren, Tim Verdonck, Wouter Verbeke
Abstract
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.
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 5f3a1739-9fe9-4170-8aa9-1b372a33d246Builds on12
- Coresets for Data-efficient Training of Machine Learning ModelsBaharan Mirzasoleiman, Jeff A. Bilmes, Jure LeskovecICML 2020 · 494 citations
- Learning Counterfactual Representations for Estimating Individual Dose-Response CurvesPatrick Schwab, Lorenz Linhardt, Stefan Bauer, Joachim M. Buhmann et al.AAAI 2020 · 159 citations
- Estimating the Effects of Continuous-valued Interventions using Generative Adversarial NetworksIoana Bica, James Jordon, Mihaela van der SchaarNeurIPS 2020 · 137 citations
- VCNet and Functional Targeted Regularization For Learning Causal Effects of Continuous TreatmentsLizhen Nie, Mao Ye, Qiang Liu, Dan NicolaeICLR 2021 · 81 citations
- OoD-Bench: Quantifying and Understanding Two Dimensions of Out-of-Distribution GeneralizationNanyang Ye, Kaican Li, Haoyue Bai, Runpeng Yu et al.CVPR 2022 · 74 citations
Related papers
- Unveiling the Potential of Robustness in Selecting Conditional Average Treatment Effect EstimatorsYiyan Huang, Cheuk Hang Leung, Siyi Wang, Yijun Li et al.NeurIPS 2024 · 2 citations
- Empirical Analysis of Model Selection for Heterogeneous Causal Effect EstimationDivyat Mahajan, Ioannis Mitliagkas, Brady Neal, Vasilis SyrgkanisICLR 2024 · 28 citations
- Treatment Effect Estimation with Disentangled Latent FactorsWeijia Zhang, Lin Liu, Jiuyong LiAAAI 2021 · 115 citations
- SpaCE: The Spatial Confounding EnvironmentMauricio Tec, Ana Trisovic, Michelle Audirac, Sophie Woodward et al.ICLR 2024 · 6 citations
- Meta-Learners for Partially-Identified Treatment Effects Across Multiple EnvironmentsJonas Schweisthal, Dennis Frauen, Mihaela van der Schaar, Stefan FeuerriegelICML 2024 · 10 citations
