Author/Authors :
Xia, Jing Department of Biomedical Engineering - Zhejiang University - Zheda Road - Hangzhou, China , Pan, Su Department of Biomedical Engineering - Zhejiang University - Zheda Road - Hangzhou, China , Zhu, Min Department of Biomedical Engineering - Zhejiang University - Zheda Road - Hangzhou, China , Cai, Guolong Department of ICU - Zhejiang Hospital - Lingyin Road - Hangzhou, China , Yan, Molei Department of ICU - Zhejiang Hospital - Lingyin Road - Hangzhou, China , Su, Qun Department of ICU - (e First Affiliated Hospital - Zhejiang University - Qingchun Road - Hangzhou, China , Yan, Jing Department of ICU - Zhejiang Hospital - Lingyin Road - Hangzhou, China , Ning, Gangmin Department of Biomedical Engineering - Zhejiang University - Zheda Road - Hangzhou, China
Abstract :
In intensive care unit (ICU), it is essential to predict the mortality of patients and mathematical models aid in improving the
prognosis accuracy. Recently, recurrent neural network (RNN), especially long short-term memory (LSTM) network, showed
advantages in sequential modeling and was promising for clinical prediction. However, ICU data are highly complex due to the
diverse patterns of diseases; therefore, instead of single LSTM model, an ensemble algorithm of LSTM (eLSTM) is proposed,
utilizing the superiority of the ensemble framework to handle the diversity of clinical data. .e eLSTM algorithm was evaluated by
the acknowledged database of ICU admissions Medical Information Mart for Intensive Care III (MIMIC-III). .e investigation in
total of 18415 cases shows that compared with clinical scoring systems SAPS II, SOFA, and APACHE II, random forests
classification algorithm, and the single LSTM classifier, the eLSTM model achieved the superior performance with the largest
value of area under the receiver operating characteristic curve (AUROC) of 0.8451 and the largest area under the precision-recall
curve (AUPRC) of 0.4862. Furthermore, it offered an early prognosis of ICU patients. .e results demonstrate that the eLSTM is
capable of dynamically predicting the mortality of patients in complex clinical situations.
Keywords :
Short-Term , Memory , Unit , ICU