mlsanityData to 7 Oct 2026

Heterogeneous Treatment Effect Estimation using machine learning for Healthcare application: tutorial and benchmark

arXiv 2109.12769 ↗First posted 27 Sep 2021cs.LG cs.IR stat.APcited by 4

Not on the map. Why?

Abstract (arXiv)

Developing new drugs for target diseases is a time-consuming and expensive task, drug repurposing has become a popular topic in the drug development field. As much health claim data become available, many studies have been conducted on the data. The real-world data is noisy, sparse, and has many confounding factors. In addition, many studies have shown that drugs effects are heterogeneous among the population. Lots of advanced machine learning models about estimating heterogeneous treatment effects (HTE) have emerged in recent years, and have been applied to in econometrics and machine learning communities. These studies acknowledge medicine and drug development as the main application area, but there has been limited translational research from the HTE methodology to drug development. We aim to introduce the HTE methodology to the healthcare area and provide feasibility consideration when translating the methodology with benchmark experiments on healthcare administrative claim data. Also, we want to use benchmark experiments to show how to interpret and evaluate the model when it is applied to healthcare research. By introducing the recent HTE techniques to a broad readership in biomedical informatics communities, we expect to promote the wide adoption of causal inference using machine learning. We also expect to provide the feasibility of HTE for personalized drug effectiveness.

Key facts. Cited by 4 papers in the map (data through 7 Oct 2026). Builds on 16 of its 64 references that are in the map. First posted on arXiv 27 Sep 2021.

Builds on

16 of 64 references.

References · 16

  1. 2017A Unified Approach to Interpreting Model Predictionscited by 6,488
  2. 2015Estimation and Inference of Heterogeneous Treatment Effects using Random Forestscited by 597
  3. 2017Meta-learners for Estimating Heterogeneous Treatment Effects using Machine Learningcited by 475
  4. 2017Quasi-Oracle Estimation of Heterogeneous Treatment Effectscited by 328
  5. 2016Learning Representations for Counterfactual Inferencecited by 327
  6. 2019Adapting Neural Networks for the Estimation of Treatment Effectscited by 285
  7. 2018Scalable and accurate deep learning for electronic health recordscited by 275
  8. 2015Recursive Partitioning for Heterogeneous Causal Effectscited by 254
  9. 2021Nonparametric Estimation of Heterogeneous Treatment Effects: From Theory to Learning Algorithmscited by 144
  10. 2020Estimating the Effects of Continuous-valued Interventions using Generative Adversarial Networkscited by 78
  11. 2020CausalML: Python Package for Causal Machine Learningcited by 74
  12. 2021On Inductive Biases for Heterogeneous Treatment Effect Estimationcited by 74
4 more
  1. 2019Estimating Treatment Effects with Causal Forests: An Applicationcited by 58
  2. 2016Double/Debiased Machine Learning for Treatment and Causal Parameterscited by 41
  3. 2013On the definition of a confoundercited by 19
  4. 2020Debiased Inverse Propensity Score Weighting for Estimation of Average Treatment Effects with High-Dimensional Confounderscited by 11

Built on by

Cited by year (2026 to 7 Oct)

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

4 papers.