Deep-Learning-Based Survival Prediction of Patients in Coronary Care Units

Background. A survival prediction model based on deep learning has higher accuracy than the CPH model in predicting the survival of CCU patients, and it also has a better discrimination ability. We collected information on patients with various diseases in coronary care units (CCUs) from the Medical Information Mart for Intensive Care III (MIMIC-III) database. The purpose of this study was to use this information to construct a neural-network model based on deep learning to predict the survival probabilities of patients with conditions that are common in CCUs. Method. We collected information on patients in the United States with five common diseases in CCUs from 2001 to 2012. We randomly divided the patients into a training cohort and a testing cohort at a ratio of 7 : 3 and applied a survival prediction method based on deep learning to predict their survival probability. We compared our model with the Cox proportional-hazards regression (CPH) model and used the concordance indexes (C-indexes), receiver operating characteristic (ROC) curve, and calibration plots to evaluate the predictive performance of the model. Results. The 3,388 CCU patients included in the study were randomly divided into 2,371 in the training cohort and 1,017 in the testing cohort. The stepwise regression results showed that the important factors affecting patient survival were the type of disease, age, race, anion gap, glucose, neutrophils, white blood cells, potassium, creatine kinase, and blood urea nitrogen (P