Predicting 30-days mortality for MIMIC-III patients with sepsis-3: a machine learning approach using XGboost

• Published in: Journal of translational medicine Vol. 18; no. 1; pp. 462 - 14
• Main Authors: Hou, Nianzong, Li, Mingzhe, He, Lu, Xie, Bing, Wang, Lin, Zhang, Rumin, Yu, Yong, Sun, Xiaodong, Pan, Zhengsheng, Wang, Kai
• Format: Journal Article
• Language: English
• Published: England BioMed Central Ltd 07.12.2020; BioMed Central; BMC
• Subjects: Algorithms; Blood pressure; Care and treatment; Critical care; Critically ill; Health aspects; Hospital Mortality; Hospitals; Humans; Infections; Intensive care; Laboratories; Learning algorithms; Logistic Models; Logistic regression; Machine Learning; Medical prognosis; MIMIC-III; Mortality; Nomograms; Patient outcomes; Patients; Physiology; Prediction models; Prognosis; Public health; Regression analysis; ROC Curve; SAPS-II score; Sepsis; Sepsis - diagnosis; Sepsis-3; Survival; Variables; Vital signs; Xgboost; China; United States > US; Massachusetts; Logistic regression; Xgboost; MIMIC-III; SAPS-II score; Sepsis-3; Machine learning
• ISSN: 1479-5876; 1479-5876
• DOI: 10.1186/s12967-020-02620-5

Sepsis is a significant cause of mortality in-hospital, especially in ICU patients. Early prediction of sepsis is essential, as prompt and appropriate treatment can improve survival outcomes. Machine learning methods are flexible prediction algorithms with potential advantages over conventional regression and scoring system. The aims of this study were to develop a machine learning approach using XGboost to predict the 30-days mortality for MIMIC-III Patients with sepsis-3 and to determine whether such model performs better than traditional prediction models. Using the MIMIC-III v1.4, we identified patients with sepsis-3. The data was split into two groups based on death or survival within 30 days and variables, selected based on clinical significance and availability by stepwise analysis, were displayed and compared between groups. Three predictive models including conventional logistic regression model, SAPS-II score prediction model and XGBoost algorithm model were constructed by R software. Then, the performances of the three models were tested and compared by AUCs of the receiver operating characteristic curves and decision curve analysis. At last, nomogram and clinical impact curve were used to validate the model. A total of 4559 sepsis-3 patients are included in the study, in which, 889 patients were death and 3670 survival within 30 days, respectively. According to the results of AUCs (0.819 [95% CI 0.800-0.838], 0.797 [95% CI 0.781-0.813] and 0.857 [95% CI 0.839-0.876]) and decision curve analysis for the three models, the XGboost model performs best. The risk nomogram and clinical impact curve verify that the XGboost model possesses significant predictive value. Using machine learning technique by XGboost, more significant prediction model can be built. This XGboost model may prove clinically useful and assist clinicians in tailoring precise management and therapy for the patients with sepsis-3.