Related Experiment Video
Updated: Nov 22, 2025

Constructing and Visualizing Models using Mime-based Machine-learning Framework
Published on: July 22, 2025
Development and external evaluation of predictions models for mortality of COVID-19 patients using machine learning
Simin Li1, Yulan Lin2, Tong Zhu1
1Yidu Cloud Technology Inc., 8F, Health Work, No. 9 Building, No. 35 of Huayuan North Road, Haidian District, Beijing, 100089 China.
Insights
The Gradient Boosting Decision Tree (GBDT) model shows superior performance in predicting coronavirus disease 2019 (COVID-19) patient mortality compared to logistic regression models. This finding aids in better patient management and resource allocation during the pandemic.
Area of Science:
- Medical Informatics
- Machine Learning in Medicine
- Epidemiology
Background:
- Coronavirus disease 2019 (COVID-19) poses a significant global health challenge.
- Accurate mortality prediction is crucial for patient management and healthcare resource allocation.
- Existing predictive models require evaluation and comparison for optimal clinical utility.
Purpose of the Study:
- To develop and compare machine learning models for predicting COVID-19 patient mortality.
- To evaluate the performance of Gradient Boosting Decision Tree (GBDT) and logistic regression (LR) models.
- To identify key clinical factors influencing COVID-19 mortality prediction.
Main Methods:
- Clinical data from 2924 COVID-19 patients in Wuhan, China (January 18 - March 29, 2020) were analyzed.
- Gradient Boosting Decision Tree (GBDT), logistic regression (LR), and a simplified LR model (LR-5) were constructed.
- Model performance was assessed using fivefold cross-validation, calculating Area Under Curve (AUC), accuracy, Positive Predictive Value (PPV), and Negative Predictive Value (NPV).
Main Results:
- The GBDT model achieved the highest fivefold AUC (0.941), outperforming LR (0.928) and LR-5 (0.913).
- GBDT demonstrated superior diagnostic accuracy (0.889) and sensitivity (0.899) compared to LR models.
- External validation identified leukomonocyte percentage, urea, age, and SpO2 as significant predictors in a separate cohort.
Conclusions:
- The GBDT model offers enhanced accuracy for predicting COVID-19 mortality compared to traditional LR models.
- Machine learning approaches, particularly GBDT, show promise for improving prognostic assessments in infectious diseases.
- The study highlights the importance of robust model evaluation for clinical decision support systems.
Abstract:
To predict the mortality of patients with coronavirus disease 2019 (COVID-19). We collected clinical data of COVID-19 patients between January 18 and March 29 2020 in Wuhan, China . Gradient boosting decision tree (GBDT), logistic regression (LR) model, and simplified LR were built to predict the mortality of COVID-19. We also evaluated different models by computing area under curve (AUC), accuracy, positive predictive value (PPV), and negative predictive value (NPV) under fivefold cross-validation. A total of 2924 patients were included in our evaluation, with 257 (8.8%) died and 2667 (91.2%) survived during hospitalization. Upon admission, there were 21 (0.7%) mild cases, 2051 (70.1%) moderate case, 779 (26.6%) severe cases, and 73 (2.5%) critically severe cases. The GBDT model exhibited the highest fivefold AUC, which was 0.941, followed by LR (0.928) and LR-5 (0.913). The diagnostic accuracies of GBDT, LR, and LR-5 were 0.889, 0.868, and 0.887, respectively. In particular, the GBDT model demonstrated the highest sensitivity (0.899) and specificity (0.889). The NPV of all three models exceeded 97%, while their PPV values were relatively low, resulting in 0.381 for LR, 0.402 for LR-5, and 0.432 for GBDT. Regarding severe and critically severe cases, the GBDT model also performed the best with a fivefold AUC of 0.918. In the external validation test of the LR-5 model using 72 cases of COVID-19 from Brunei, leukomonocyte (%) turned to show the highest fivefold AUC (0.917), followed by urea (0.867), age (0.826), and SPO2 (0.704). The findings confirm that the mortality prediction performance of the GBDT is better than the LR models in confirmed cases of COVID-19. The performance comparison seems independent of disease severity.
Supplementary Information:
The online version contains supplementary material available at(10.1007/s00521-020-05592-1).
Related Concept Videos
Steps in Outbreak Investigation
Cancer Survival Analysis
Kaplan-Meier Approach
Comparing the Survival Analysis of Two or More Groups

