Skip to main page content
U.S. flag

An official website of the United States government

Dot gov

The .gov means it’s official.
Federal government websites often end in .gov or .mil. Before sharing sensitive information, make sure you’re on a federal government site.

Https

The site is secure.
The https:// ensures that you are connecting to the official website and that any information you provide is encrypted and transmitted securely.

Access keys NCBI Homepage MyNCBI Homepage Main Content Main Navigation
. 2024 Jan 11;19(1):e0291373.
doi: 10.1371/journal.pone.0291373. eCollection 2024.

Utilizing machine learning for survival analysis to identify risk factors for COVID-19 intensive care unit admission: A retrospective cohort study from the United Arab Emirates

Aamna AlShehhi  1   2 , Taleb M Almansoori  3 , Ahmed R Alsuwaidi  4 , Hiba Alblooshi  5
Affiliations

Utilizing machine learning for survival analysis to identify risk factors for COVID-19 intensive care unit admission: A retrospective cohort study from the United Arab Emirates

Aamna AlShehhi et al. PLoS One. .

Abstract

Background: The current situation of the unprecedented COVID-19 pandemic leverages Artificial Intelligence (AI) as an innovative tool for addressing the evolving clinical challenges. An example is utilizing Machine Learning (ML) models-a subfield of AI that take advantage of observational data/Electronic Health Records (EHRs) to support clinical decision-making for COVID-19 cases. This study aimed to evaluate the clinical characteristics and risk factors for COVID-19 patients in the United Arab Emirates utilizing EHRs and ML for survival analysis models.

Methods: We tested various ML models for survival analysis in this work we trained those models using a different subset of features extracted by several feature selection methods. Finally, the best model was evaluated and interpreted using goodness-of-fit based on calibration curves,Partial Dependence Plots and concordance index.

Results: The risk of severe disease increases with elevated levels of C-reactive protein, ferritin, lactate dehydrogenase, Modified Early Warning Score, respiratory rate and troponin. The risk also increases with hypokalemia, oxygen desaturation and lower estimated glomerular filtration rate and hypocalcemia and lymphopenia.

Conclusion: Analyzing clinical data using AI models can provide vital information for clinician to measure the risk of morbidity and mortality of COVID-19 patients. Further validation is crucial to implement the model in real clinical settings.

PubMed Disclaimer

Conflict of interest statement

The authors have declared that no competing interests exist.

Figures

Fig 1
Fig 1. Flow diagram of the cohort.
Inclusion and exclusion criteria for determining patients consider in this study. SEHA extracted 1,800 COVID-19 patients from March 1, 2020 to April 20, 2020. After applying the inclusion and exclusion criteria, our study included 1787 patients.
Fig 2
Fig 2. Models hyperparameters tuning: Heatmap of mean C-index (5-fold cross validation) of the best combination of the hyperparameters and feature selections methods; also we tune the threshold for selecting the highest importance features.
Fig 3
Fig 3
Calibration curves and time-dependent ROC curve of the gradient boosting machine (GBM) model: Calibration curves of predicted compared with observed ICU access after 2 Days (a), 3 Days (b) and 5 Days (c) of hospital administration. Time-dependent ROC curve for the ICU admission predicting after after 2 Days (d),3 Days (e) and 5 Days (f) of hospital administration.
Fig 4
Fig 4. Partial dependence plots inferred from gradient boosting machine (GBM) model using random forest minimal depth features subset: The lines present the change in the risk to access the ICU across selected variable of interest whilst holding other variables constant.
See this image and copyright information in PMC

References

    1. Dyusupova A., Faizova R., Yurkovskaya O., Belyaeva T., Terekhova T., Khismetova A., et al. Clinical characteristics and risk factors for disease severity and mortality of COVID-19 patients with diabetes mellitus in Kazakhstan: A nationwide study. Heliyon. 7, e06561 (2021,3), https://www.sciencedirect.com/science/article/pii/S2405844021006642 - PMC - PubMed
    1. Nemati M., Ansary J. & Nemati N. Machine-Learning Approaches in COVID-19 Survival Analysis and Discharge-Time Likelihood Prediction Using Clinical Data. Patterns. 1, 100074 (2020,8), https://www.sciencedirect.com/science/article/pii/S2666389920300945 - PMC - PubMed
    1. Williamson E., Walker A., Bhaskaran K., Bacon S., Bates C., Morton C., et al. Factors associated with COVID-19-related death using OpenSAFELY. Nature. 584, 430–436 (2020,8) doi: 10.1038/s41586-020-2521-4 - DOI - PMC - PubMed
    1. Schwab P., Mehrjou A., Parbhoo S., Celi L., Hetzel J., Hofer M., et al. Real-time prediction of COVID-19 related mortality using electronic health records. Nature Communications. 12, 1058 (2021,2) doi: 10.1038/s41467-020-20816-7 - DOI - PMC - PubMed
    1. Soltan A., Kouchaki S., Zhu T., Kiyasseh D., Taylor T., Hussain Z., et al. Rapid triage for COVID-19 using routine clinical data for patients attending hospital: development and prospective validation of an artificial intelligence screening test. The Lancet Digital Health. 3, e78–e87 (2021,2), Publisher: Elsevier doi: 10.1016/S2589-7500(20)30274-0 - DOI - PMC - PubMed