Identifying factors related to mortality of hospitalized COVID-19 patients using machine learning methods
- PMID: 39170355
- PMCID: PMC11337730
- DOI: 10.1016/j.heliyon.2024.e35561
Identifying factors related to mortality of hospitalized COVID-19 patients using machine learning methods
Abstract
Background: The COVID-19 pandemic has had a profound impact globally, presenting significant social and economic challenges. This study aims to explore the factors affecting mortality among hospitalized COVID-19 patients and construct a machine learning-based model to predict the risk of mortality.
Methods: The study examined COVID-19 patients admitted to Imam Reza Hospital in Tabriz, Iran, between March 2020 and November 2021. The Elastic Net method was employed to identify and rank features associated with mortality risk. Subsequently, an artificial neural network (ANN) model was developed based on these features to predict mortality risk. The performance of the model was evaluated by receiver operating characteristic (ROC) curve analysis.
Results: The study included 706 patients with 96 features, out of them 26 features were identified as crucial predictors of mortality. The ANN model, utilizing 20 of these features, achieved an area under the ROC curve (AUC) of 98.8 %, effectively stratifying patients by mortality risk.
Conclusion: The developed model offers accurate and precipitous mortality risk predictions for COVID-19 patients, enhancing the responsiveness of healthcare systems to high-risk individuals.
Keywords: Artificial neural network; COVID-19; Elastic net; Feature selection; LASSO; Machine learning; Mortality.
© 2024 The Authors.
Conflict of interest statement
The authors declare that they have no known competing financial interests or personal relationships that could have appeared to influence the work reported in this paper.
Figures



Similar articles
-
The Development and Validation of Simplified Machine Learning Algorithms to Predict Prognosis of Hospitalized Patients With COVID-19: Multicenter, Retrospective Study.J Med Internet Res. 2022 Jan 21;24(1):e31549. doi: 10.2196/31549. J Med Internet Res. 2022. PMID: 34951865 Free PMC article.
-
Machine Learning to Predict Mortality and Critical Events in a Cohort of Patients With COVID-19 in New York City: Model Development and Validation.J Med Internet Res. 2020 Nov 6;22(11):e24018. doi: 10.2196/24018. J Med Internet Res. 2020. PMID: 33027032 Free PMC article.
-
Generalizable machine learning approach for COVID-19 mortality risk prediction using on-admission clinical and laboratory features.Sci Rep. 2023 Feb 10;13(1):2399. doi: 10.1038/s41598-023-28943-z. Sci Rep. 2023. PMID: 36765157 Free PMC article.
-
Identification and validation of prognostic factors in patients with COVID-19: A retrospective study based on artificial intelligence algorithms.J Intensive Med. 2021 May 24;1(2):103-109. doi: 10.1016/j.jointm.2021.04.001. eCollection 2021 Oct. J Intensive Med. 2021. PMID: 36943822 Free PMC article.
-
Artificial intelligence in clinical care amidst COVID-19 pandemic: A systematic review.Comput Struct Biotechnol J. 2021;19:2833-2850. doi: 10.1016/j.csbj.2021.05.010. Epub 2021 May 7. Comput Struct Biotechnol J. 2021. PMID: 34025952 Free PMC article. Review.
References
-
- WHO Coronavirus Disease (COVID-19) Dashboard DlupC, https://covid19.who.int/. Available from: https://covid19.who.int/.
LinkOut - more resources
Full Text Sources