A descriptive study of random forest algorithm for predicting COVID-19 patients outcome

Abstract

Background The outbreak of coronavirus disease 2019 (COVID-19) that occurred in Wuhan, China, has become a global public health threat. It is necessary to identify indicators that can be used as optimal predictors for clinical outcomes of COVID-19 patients. Methods The clinical information from 126 patients diagnosed with COVID-19 were collected from Wuhan Fourth Hospital. Specific clinical characteristics, laboratory findings, treatments and clinical outcomes were analyzed from patients hospitalized for treatment from 1 February to 15 March 2020, and subsequently died or were discharged. A random forest (RF) algorithm was used to predict the prognoses of COVID-19 patients and identify the optimal diagnostic predictors for patients’ clinical prognoses. Results Seven of the 126 patients were excluded for losing endpoints, 103 of the remaining 119 patients were discharged (alive) and 16 died in the hospital. A synthetic minority over-sampling technique (SMOTE) was used to correct the imbalanced distribution of clinical patients. Recursive feature elimination (RFE) was used to select the optimal subset for analysis. Eleven clinical parameters, Myo, CD8, age, LDH, LMR, CD45, Th/Ts, dyspnea, NLR, D-Dimer and CK were chosen with AUC approximately 0.9905. The RF algorithm was built to predict the prognoses of COVID-19 patients based on the best subset, and the area under the ROC curve (AUC) of the test data was 100%. Moreover, two optimal clinical risk predictors, lactate dehydrogenase (LDH) and Myoglobin (Myo), were selected based on the Gini index. The univariable logistic analysis revealed a substantial increase in the risk for in-hospital mortality when Myo was higher than 80 ng/ml (OR = 7.54, 95% CI [3.42–16.63]) and LDH was higher than 500 U/L (OR = 4.90, 95% CI [2.13–11.25]). Conclusion We applied an RF algorithm to predict the mortality of COVID-19 patients with high accuracy and identified LDH higher than 500 U/L and Myo higher than 80 ng/ml to be potential risk factors for the prognoses of COVID-19 patients in the early stage of the disease.

Comparing Decision Tree-Based Ensemble Machine Learning Models for COVID-19 Death Probability Profiling

“…The resulting profile confirms and reinforces previous studies’ results [1-5, 18, 26, 27] on COVID-19 risk factors. The profile is also consistent with the exponential increase in death risk with age, that we analyzed inferentially.…”

“…Random forests were introduced by Breiman [17] as a way to improve tree-based ensembles' performance, while not increasing the bias significantly [16,17]. Random forests have been applied in the medical context [7][8][9][10], including, most recently, to the SARS-CoV-2 pandemic [18,19].…”

Exploring temporal varying demographic and economic disparities in COVID-19 infections in four U.S. areas: based on OLS, GWR, and random forest models

“…It is simple and easy to implement, but it shows great performance in regression. Compared with traditional statistical models, the RF can judge the importance of features and the mutual influence between different features, which is not easy to overfit (Iwendi et al, 2020 ; Wang et al, 2020 ).…”

The role of European health system characteristics in affecting Covid 19 lethality during the early days of the pandemic

“…Recently, they have been employed to compute COVID-19 mortality or to predict the risk of mortality 37 , 41 , 42 . The data used for the analyses are in prevalence based on patients’ physiological conditions, symptoms, demographic information 41 , 42 , population characteristics 43 or blood lab results and clinical data 41 , 44 – 47 .…”