Using decision tree algorithms for estimating ICU admission of COVID-19 patients

Introduction: Coronavirus disease 2019 (COVID-19) outbreak has overwhelmed many healthcare systems worldwide and put them at the edge of collapsing. As intensive care unit (ICU) capacities are limited, deciding on the proper allocation of required resources is crucial. This study aimed to develop an...

Full description

Bibliographic Details
Main Authors: Mostafa Shanbehzadeh, Raoof Nopour, Hadi Kazemi-Arpanahi
Format: Article
Language:English
Published: Elsevier 2022-01-01
Series:Informatics in Medicine Unlocked
Subjects:
Online Access:http://www.sciencedirect.com/science/article/pii/S2352914822000685
_version_ 1818201043008225280
author Mostafa Shanbehzadeh
Raoof Nopour
Hadi Kazemi-Arpanahi
author_facet Mostafa Shanbehzadeh
Raoof Nopour
Hadi Kazemi-Arpanahi
author_sort Mostafa Shanbehzadeh
collection DOAJ
description Introduction: Coronavirus disease 2019 (COVID-19) outbreak has overwhelmed many healthcare systems worldwide and put them at the edge of collapsing. As intensive care unit (ICU) capacities are limited, deciding on the proper allocation of required resources is crucial. This study aimed to develop and compare models for early predicting ICU admission in COVID-19 patients at the point of hospital admission. Materials and methods: Using a single-center registry, we studied the records of 512 COVID-19 patients. First, the most important variables were identified using Chi-square test (at p < 0.01) and logistic regression (with odds ratio at P < 0.05). Second, we trained seven decision tree (DT) algorithms (decision stump (DS), Hoeffding tree (HT), LMT, J-48, random forest (RF), random tree (RT) and REP-Tree) using the selected variables. Finally, the models’ performance was evaluated. Furthermore, we used an external dataset to validate the prediction models. Results: Using the Chi-square test, 20 important variables were identified. Then, 12 variables were selected for model construction using logistic regression. Comparing the DT methods demonstrated that J-48 (F-score of 0.816 and AUC of 0.845) had the best performance. Also, the J-48 (F-score = 80.9% and AUC = 0.822) gained the best performance in generalizability using the external dataset. Conclusions: The study results demonstrated that DT algorithms can be used to predict ICU admission requirements in COVID-19 patients based on the first time of admission data. Implementing such models has the potential to inform clinicians and managers to adopt the best policy and get prepare during the COVID-19 time-sensitive and resource-constrained situation.
first_indexed 2024-12-12T02:47:16Z
format Article
id doaj.art-22b92b5c5c2a4ee8a956413380f0b2e9
institution Directory Open Access Journal
issn 2352-9148
language English
last_indexed 2024-12-12T02:47:16Z
publishDate 2022-01-01
publisher Elsevier
record_format Article
series Informatics in Medicine Unlocked
spelling doaj.art-22b92b5c5c2a4ee8a956413380f0b2e92022-12-22T00:40:58ZengElsevierInformatics in Medicine Unlocked2352-91482022-01-0130100919Using decision tree algorithms for estimating ICU admission of COVID-19 patientsMostafa Shanbehzadeh0Raoof Nopour1Hadi Kazemi-Arpanahi2Department of Health Information Technology, School of Paramedical, Ilam University of Medical Sciences, Ilam, IranDepartment of Health Information Management, Student Research Committee, School of Health Management and Information Sciences Branch, Iran University of Medical Sciences, Tehran, IranDepartment of Health Information Technology, Abadan University of Medical Sciences, Abadan, Iran; Department of Student Research Committee, Abadan University of Medical Sciences, Iran; Corresponding author. Department of Health Information Technology, Abadan University of Medical Sciences, Abadan, Iran.Introduction: Coronavirus disease 2019 (COVID-19) outbreak has overwhelmed many healthcare systems worldwide and put them at the edge of collapsing. As intensive care unit (ICU) capacities are limited, deciding on the proper allocation of required resources is crucial. This study aimed to develop and compare models for early predicting ICU admission in COVID-19 patients at the point of hospital admission. Materials and methods: Using a single-center registry, we studied the records of 512 COVID-19 patients. First, the most important variables were identified using Chi-square test (at p < 0.01) and logistic regression (with odds ratio at P < 0.05). Second, we trained seven decision tree (DT) algorithms (decision stump (DS), Hoeffding tree (HT), LMT, J-48, random forest (RF), random tree (RT) and REP-Tree) using the selected variables. Finally, the models’ performance was evaluated. Furthermore, we used an external dataset to validate the prediction models. Results: Using the Chi-square test, 20 important variables were identified. Then, 12 variables were selected for model construction using logistic regression. Comparing the DT methods demonstrated that J-48 (F-score of 0.816 and AUC of 0.845) had the best performance. Also, the J-48 (F-score = 80.9% and AUC = 0.822) gained the best performance in generalizability using the external dataset. Conclusions: The study results demonstrated that DT algorithms can be used to predict ICU admission requirements in COVID-19 patients based on the first time of admission data. Implementing such models has the potential to inform clinicians and managers to adopt the best policy and get prepare during the COVID-19 time-sensitive and resource-constrained situation.http://www.sciencedirect.com/science/article/pii/S2352914822000685COVID-19CoronavirusMachine learningIntensive care unitDecision tree
spellingShingle Mostafa Shanbehzadeh
Raoof Nopour
Hadi Kazemi-Arpanahi
Using decision tree algorithms for estimating ICU admission of COVID-19 patients
Informatics in Medicine Unlocked
COVID-19
Coronavirus
Machine learning
Intensive care unit
Decision tree
title Using decision tree algorithms for estimating ICU admission of COVID-19 patients
title_full Using decision tree algorithms for estimating ICU admission of COVID-19 patients
title_fullStr Using decision tree algorithms for estimating ICU admission of COVID-19 patients
title_full_unstemmed Using decision tree algorithms for estimating ICU admission of COVID-19 patients
title_short Using decision tree algorithms for estimating ICU admission of COVID-19 patients
title_sort using decision tree algorithms for estimating icu admission of covid 19 patients
topic COVID-19
Coronavirus
Machine learning
Intensive care unit
Decision tree
url http://www.sciencedirect.com/science/article/pii/S2352914822000685
work_keys_str_mv AT mostafashanbehzadeh usingdecisiontreealgorithmsforestimatingicuadmissionofcovid19patients
AT raoofnopour usingdecisiontreealgorithmsforestimatingicuadmissionofcovid19patients
AT hadikazemiarpanahi usingdecisiontreealgorithmsforestimatingicuadmissionofcovid19patients