Bagging and deep learning in optimal individualized treatment rules
Bagging and deep learning in optimal individualized treatment rules
复制标题
DOI:
10.1111/biom.12990
复制
发表时间:
2019-06-01
期刊:
影响因子:
1.9
通讯作者:
Zhu, Ruoqing
中科院分区:
文献类型:
--
作者:
Mi, Xinlei;Zou, Fei;Zhu, Ruoqing
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.