Chinese Medical Sciences Journal ›› 2022, Vol. 37 ›› Issue (3): 201-209.doi: 10.24920/004102

• Scientific Data Sharing and Reuse:Original Article • Previous Articles     Next Articles

Comparison of Mortality Predictive Models of Sepsis Patients Based on Machine Learning

Ziyang Wang, Yushan Lan, Zidu Xu, Yaowen Gu, Jiao Li*()   

  1. Institute of Medical Information/Medical Library, Chinese Academy of Medical Science & Peking Union Medical College, Beijing 100020, China
  • Received:2022-04-21 Accepted:2022-08-10 Published:2022-09-30 Online:2022-09-20
  • Contact: Jiao Li

Objective To compare the performance of five machine learning models and SAPS II score in predicting the 30-day mortality amongst patients with sepsis.
Methods The sepsis patient-related data were extracted from the MIMIC-IV database. Clinical features were generated and selected by mutual information and grid search. Logistic regression, Random forest, LightGBM, XGBoost, and other machine learning models were constructed to predict the mortality probability. Five measurements including accuracy, precision, recall, F1 score, and area under curve (AUC) were acquired for model evaluation. An external validation was implemented to avoid conclusion bias.
Results LightGBM outperformed other methods, achieving the highest AUC (0.900), accuracy (0.808), and precision (0.559). All machine learning models performed better than SAPS II score (AUC=0.748). LightGBM achieved 0.883 in AUC in the external data validation.
Conclusions The machine learning models are more effective in predicting the 30-day mortality of patients with sepsis than the traditional SAPS II score.

Key words: MIMIC-IV, sepsis, machine learning, risk prediction


Copyright © 2021 Chinese Academy of Medical Sciences.  京公安备110402430088  京ICP备06002729号-1  Powered by Magtech.

Supervised by National Health Commission of the People's Republic of China

9 Dongdan Santiao, Dongcheng district, Beijing, 100730 China

Tel: 86-10-65105897  Fax:86-10-65133074 


Copyright © 2018 Chinese Academy of Medical Sciences

All right reserved.

京公安备110402430088  京ICP备06002729号-1