The Challenges of Hyperparameter Tuning for Accurate Causal Effect Estimation
Not on the map. Why?
Abstract (arXiv)
ML is playing an increasingly crucial role in estimating causal effects of treatments on outcomes from observational data. Many ML methods (`causal estimators') have been proposed for this task. All of these methods, as with any ML approach, require extensive hyperparameter tuning. For non-causal predictive tasks, there is a consensus on the choice of tuning metrics (e.g. mean squared error), making it simple to compare models. However, for causal inference tasks, such a consensus is yet to be reached, making any comparison of causal models difficult. On top of that, there is no ideal metric on which to tune causal estimators, so one must rely on proxies. Furthermore, the fact that model selection in causal inference involves multiple components (causal estimator, ML regressor, hyperparameters, metric), complicates the issue even further. In order to evaluate the importance of each component, we perform an extensive empirical study on their combination. Our experimental setup involves many commonly used causal estimators, regressors (`base learners' henceforth) and metrics applied to four well-known causal inference benchmark datasets. Our results show that hyperparameter tuning increased the probability of reaching state-of-the-art performance in average (65% rightarrow 81%) and individualised (50% rightarrow 57%) effect estimation with only commonly used estimators. We also show that the performance of standard metrics can be inconsistent across different scenarios. Our findings highlight the need for further research to establish whether metrics uniformly capable of state-of-the-art performance in causal model evaluation can be found.
Key facts. Cited by 6 papers in the map (data through 7 Oct 2026). Builds on 17 of its 50 references that are in the map. First posted on arXiv 2 Mar 2023.
Builds on
17 of 50 references.
References · 17
- 2017Meta-learners for Estimating Heterogeneous Treatment Effects using Machine Learningcited by 475
- 2017Causal Effect Inference with Deep Latent-Variable Modelscited by 408
- 2017Quasi-Oracle Estimation of Heterogeneous Treatment Effectscited by 328
- 2016Learning Representations for Counterfactual Inferencecited by 327
- 2016Generalized Random Forestscited by 309
- 2019Adapting Neural Networks for the Estimation of Treatment Effectscited by 285
- 2020A Survey on Causal Inferencecited by 202
- 2019Orthogonal Statistical Learningcited by 150
- 2020RealCause: Realistic Causal Inference Benchmarkingcited by 44
- 2019Using Wasserstein Generative Adversarial Networks for the Design of Monte Carlo Simulationscited by 32
- 2023In Search of Insights, Not Magic Bullets: Towards Demystification of the Model Selection Dilemma in Heterogeneous Treatment Effect Estimationcited by 29
- 2018A comparison of methods for model selection when estimating individual treatment effectscited by 28
5 more
- 2022Empirical Analysis of Model Selection for Heterogeneous Causal Effect Estimationcited by 25
- 2019An Evaluation Toolkit to Guide Model Selection and Cohort Definition in Causal Inferencecited by 18
- 2021Benchpress: A Scalable and Versatile Workflow for Benchmarking Structure Learning Algorithmscited by 11
- 2017Synth-Validation: Selecting the Best Causal Inference Method for a Given Datasetcited by 9
- 2022Undersmoothing Causal Estimators with Generative Treescited by 2
Built on by
Citing papers per year: 2023: 6.
6 papers.
- 2023Dynamic Inter-treatment Information Sharing for Individualized Treatment Effects Estimationcited by 11
- 2023Robustness of Algorithms for Causal Structure Learning to Hyperparameter Choicecited by 4
- 2023Double Machine Learning for Static Panel Models with Fixed Effectscited by 4
- 2023Transfer Learning for Causal Effect Estimationcited by 3
- 2023Learned Causal Method Predictioncited by 3
- 2023Assessing Electricity Service Unfairness with Transfer Counterfactual Learningcited by 1
- 2023Double Machine Learning for Static Panel Models with Fixed Effects13 Dec 2023
- 2023Learned Causal Method Prediction7 Nov 2023
- 2023Robustness of Algorithms for Causal Structure Learning to Hyperparameter Choice27 Oct 2023
- 2023Assessing Electricity Service Unfairness with Transfer Counterfactual Learning5 Oct 2023
- 2023Dynamic Inter-treatment Information Sharing for Individualized Treatment Effects Estimation25 May 2023
- 2023Transfer Learning for Causal Effect Estimation16 May 2023