Tree based weighted learning for estimating individualized treatment rules with censored data

Yifan Cui, Ruoqing Zhu, Michael Kosorok

Research output: Contribution to journalArticlepeer-review


Estimating individualized treatment rules is a central task for personalized medicine. [25] and [23] proposed outcome weighted learning to estimate individualized treatment rules directly through maximizing the expected outcome without modeling the response directly. In this paper, we extend the outcome weighted learning to right censored survival data without requiring either inverse probability of censoring weighting or semiparametric modeling of the censoring and failure times as done in [26]. To accomplish this, we take advantage of the tree based approach proposed in [29] to nonparametrically impute the survival time in two different ways. The first approach replaces the reward of each individual by the expected survival time, while in the second approach only the censored observations are imputed by their conditional expected failure times. We establish consistency and convergence rates for both estimators. In simulation studies, our estimators demonstrate improved performance compared to existing methods. We also illustrate the proposed method on a phase III clinical trial of non-small cell lung cancer.

Original languageEnglish (US)
Pages (from-to)3927-3953
Number of pages27
JournalElectronic Journal of Statistics
Issue number2
StatePublished - 2017


  • Consistency
  • Individualized treatment rule
  • Nonparametric estimation
  • Outcome weighted learning
  • Recursively imputed survival trees
  • Right censored data

ASJC Scopus subject areas

  • Statistics and Probability
  • Statistics, Probability and Uncertainty


Dive into the research topics of 'Tree based weighted learning for estimating individualized treatment rules with censored data'. Together they form a unique fingerprint.

Cite this