Model selection for estimating treatment effects

Craig A. Rolling, Yuhong Yang

Research output: Contribution to journalArticle

11 Scopus citations

Abstract

Researchers often believe that a treatment's effect on a response may be heterogeneous with respect to certain baseline covariates. This is an important premise of personalized medicine. Several methods for estimating heterogeneous treatment effects have been proposed. However, little attention has been given to the problem of choosing between estimators of treatment effects. Models that best estimate the regression function may not be best for estimating the effect of a treatment; therefore, there is a need for model selection methods that are targeted to treatment effect estimation. We demonstrate an application of the focused information criterion in this setting and develop a treatment effect cross-validation aimed at minimizing treatment effect estimation errors. Theoretically, treatment effect cross-validation has a model selection consistency property when the data splitting ratio is properly chosen. Practically, treatment effect cross-validation has the flexibility to compare different types of models. We illustrate the methods by using simulation studies and data from a clinical trial comparing treatments of patients with human immunodeficiency virus.

Original languageEnglish (US)
Pages (from-to)749-769
Number of pages21
JournalJournal of the Royal Statistical Society. Series B: Statistical Methodology
Volume76
Issue number4
DOIs
StatePublished - Sep 2014

    Fingerprint

Keywords

  • Causal inference
  • Cross-validation
  • Focused information criterion
  • Model selection
  • Model selection consistency
  • Treatment effect estimation

Cite this