Background: Machine learning models could assist physicians in identifying high-risk COVID-19 patients. This study aimed to predict the Intensive Care Unit (ICU) admission in COVID-19 hospitalized patients by the Artificial Neural Network (ANN) model combined with Elastic Net algorithm. Methods: In this prospective study, the data of 139 COVID-19 patients admitted to Imam Reza Hospital in Tabriz between 20 March and 5 April 2020, were analyzed. The Elastic Net method was used to choose features with high importance. The chosen variables were standardized and ANN was fitted to the data with one hidden layer based on the descending gradient algorithm. To validate the model, the training and test group method with a ratio of 70 to 30 was used. The model’s predictive power was reported by calculating the overall accuracy, sensitivity, specificity, and Area Under the ROC curve (AUC). Results: According to the results of the Elastic Net, the ANN model was constructed based on age, sex, body mass index, diabetes mellitus status, history of heart disease, and vital signs including systolic and diastolic blood pressure, saturated oxygen level, pulse rate, respiration rate, and body temperature. The overall accuracy of this model was 93.15%, sensitivity 80%, specificity 95.8%, and AUC 0.90. Saturated oxygen level, pulse rate, and age were the most important and predictive variables. Conclusion: In the investigated sample of patients admitted with COVID-19, the fitted ANN model had acceptable performance to predict the ICU admission. This finding could be useful for physicians and policy makers.
Read full abstract