In Search of Insights, Not Magic Bullets: Towards Demystification of the Model Selection Dilemma in Heterogeneous Treatment Effect Estimation
Alicia Curth, Mihaela van der Schaar
Abstract
Personalized treatment effect estimates are often of interest in high-stakes applications -- thus, before deploying a model estimating such effects in practice, one needs to be sure that the best candidate from the ever-growing machine learning toolbox for this task was chosen. Unfortunately, due to the absence of counterfactual information in practice, it is usually not possible to rely on standard validation metrics for doing so, leading to a well-known model selection dilemma in the treatment effect estimation literature. While some solutions have recently been investigated, systematic understanding of the strengths and weaknesses of different model selection criteria is still lacking. In this paper, instead of attempting to declare a global `winner', we therefore empirically investigate success- and failure modes of different selection criteria. We highlight that there is a complex interplay between selection strategies, candidate estimators and the data used for comparing them, and provide interesting insights into the relative (dis)advantages of different criteria alongside desiderata for the design of further illuminating empirical studies in this context.
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 531d0c8a-efd2-4b9a-a577-554ce3505a9aCited by top-tier papers14
- Empirical Analysis of Model Selection for Heterogeneous Causal Effect EstimationDivyat Mahajan, Ioannis Mitliagkas, Brady Neal, Vasilis SyrgkanisICLR 2024 · 28 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
- Inverse-Variance Weighting for Estimation of Heterogeneous Treatment EffectsAaron FisherICML 2024 · 7 citations
- Improving the Generation and Evaluation of Synthetic Data for Downstream Medical Causal InferenceHarry Amad, Zhaozhi Qian, Dennis Frauen, Julianna Piskorz et al.NeurIPS 2025 · 6 citations
Builds on5
- Learning Disentangled Representations for CounterFactual RegressionNegar Hassanpour, Russell GreinerICLR 2020 · 176 citations
- On Inductive Biases for Heterogeneous Treatment Effect EstimationAlicia Curth, Mihaela van der SchaarNeurIPS 2021 · 114 citations
- Validating Causal Inference MethodsHarsh Parikh, Carlos Varjao, Louise Xu, Eric Tchetgen TchetgenICML 2022 · 36 citations
- Counterfactual Cross-Validation: Stable Model Selection Procedure for Causal Inference ModelsYuta Saito, Shota YasuiICML 2020 · 34 citations
- -Intact-VAE: Identifying and Estimating Causal Effects under Limited OverlapPengzhou Abel Wu, Kenji FukumizuICLR 2022 · 10 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
- Defining Expertise: Applications to Treatment Effect EstimationAlihan Hüyük, Qiyao Wei, Alicia Curth, Mihaela van der SchaarICLR 2024 · 3 citations
- AutoCATE: End-to-End, Automated Treatment Effect EstimationToon Vanderschueren, Tim Verdonck, Mihaela van der Schaar, Wouter VerbekeICML 2025
- Estimating Heterogeneous Treatment Effects: Mutual Information Bounds and Learning AlgorithmsXingzhuo Guo, Yuchen Zhang, Jianmin Wang, Mingsheng LongICML 2023 · 11 citations
- Comparing Targeting Strategies for Maximizing Social Welfare with Limited ResourcesVibhhu Sharma, Bryan WilderICLR 2025
