Development of Machine‐Learning Model to Predict COVID‐19 Mortality: Application of Ensemble Model and Regarding Feature Impacts

Seung Min Baik, Miae Lee, Kyung Sook Hong, Dong Jin Park

Research output: Contribution to journalArticlepeer-review

Abstract

This study was designed to develop machine‐learning models to predict COVID‐19 mortality and identify its key features based on clinical characteristics and laboratory tests. For this, deep‐learning (DL) and machine‐learning (ML) models were developed using receiver operating characteristic (ROC) area under the curve (AUC) and F1 score optimization of 87 parameters. Of the two, the DL model exhibited better performance (AUC 0.8721, accuracy 0.84, and F1 score 0.76). However, we also blended DL with ML, and the ensemble model performed the best (AUC 0.8811, accuracy 0.85, and F1 score 0.77). The DL model is generally unable to extract feature importance; however, we succeeded by using the Shapley Additive exPlanations method for each model. This study demonstrated both the applicability of DL and ML models for classifying COVID‐19 mortality using hospital‐structured data and that the ensemble model had the best predictive ability.

Original languageEnglish
Article number1464
JournalDiagnostics
Volume12
Issue number6
DOIs
StatePublished - Jun 2022

Keywords

  • COVID‐19
  • artificial intelligence
  • ensemble model
  • mortality

Fingerprint

Dive into the research topics of 'Development of Machine‐Learning Model to Predict COVID‐19 Mortality: Application of Ensemble Model and Regarding Feature Impacts'. Together they form a unique fingerprint.

Cite this