Bagging and deep learning in optimal individualized treatment rules

Xinlei Mi*, Fei Zou, Ruoqing Zhu

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

13 Scopus citations


An ENsemble Deep Learning Optimal Treatment (EndLot) approach is proposed for personalized medicine problems. The statistical framework of the proposed method is based on the outcome weighted learning (OWL) framework which transforms the optimal decision rule problem into a weighted classification problem. We further employ an ensemble of deep neural networks (DNNs) to learn the optimal decision rule. Utilizing the flexibility of DNNs and the stability of bootstrap aggregation, the proposed method achieves a considerable improvement over existing methods. An R package “ITRlearn” is developed to implement the proposed method. Numerical performance is demonstrated via simulation studies and a real data analysis of the Cancer Cell Line Encyclopedia data.

Original languageEnglish (US)
Pages (from-to)674-684
Number of pages11
Issue number2
StatePublished - 2019
Externally publishedYes


  • Bootstrap aggregating
  • deep neural network
  • high-dimensional data
  • outcome weighted learning
  • personalized medicine

ASJC Scopus subject areas

  • Agricultural and Biological Sciences(all)
  • Applied Mathematics
  • Biochemistry, Genetics and Molecular Biology(all)
  • Immunology and Microbiology(all)
  • Statistics and Probability


Dive into the research topics of 'Bagging and deep learning in optimal individualized treatment rules'. Together they form a unique fingerprint.

Cite this