Automated ML Techniques for Predicting COVID-19 Mortality in the ICU

Stud Health Technol Inform. 2023 Jun 29:305:517-520. doi: 10.3233/SHTI230547.

Abstract

The COVID-19 infection is still a serious threat to public health and healthcare systems. Numerous practical machine learning applications have been investigated in this context to support clinical decision-making, forecast disease severity and admission to the intensive care unit, as well as to predict the demand for hospital beds, equipment, and staff in the future. We retrospectively analyzed demographics, and routine blood biomarkers from consecutive Covid-19 patients admitted to the intensive care unit (ICU) of a public tertiary hospital, during a 17-month period, relative to the outcome, in order to build a prognostic model. We used the Google Vertex AI platform, on the one hand, to evaluate its performance in predicting ICU mortality, and on the other hand to show the ease with which even non-experts can make prognostic models. The model's performance regarding the area under the receiver operating characteristic curve (AUC-ROC) was 0.955. The six highest-ranked predictors of mortality in the prognostic model were age, serum urea, platelets, C-reactive protein, hemoglobin, and SGOT.

Keywords: Artificial intelligence; COVID-19; ICU—intensive care unit; SARS-CoV-2; automated machine learning; machine learning.

MeSH terms

  • Area Under Curve
  • Blood Platelets
  • COVID-19* / diagnosis
  • Humans
  • Intensive Care Units
  • Retrospective Studies