Predicting the Disease Outcome in COVID-19 Positive Patients Through Machine Learning: A Retrospective Cohort Study
Fernanda Sumika Hojo De Souza1, Natália Satchiko Hojo-Souza2, Edimilson Batista Dos Santos1
1Department of Computer Science, Federal University of São João Del-Rei, São João Del-Rei, Brazil.
Insights
Machine learning accurately predicts COVID-19 patient outcomes using demographic and clinical data. This aids physicians in decision-making, improving response times and preventing healthcare system overload.
Area of Science:
- Medical Informatics
- Public Health
- Machine Learning in Healthcare
Background:
- Brazil registered its first COVID-19 case in February 2020, rapidly escalating to over 672,000 cases and 36,000 deaths by June 2020.
- Accurate diagnosis and prognosis are crucial for effective COVID-19 patient treatment and healthcare system management.
Purpose of the Study:
- To develop and evaluate a machine learning model for predicting poor prognosis and outcomes in COVID-19 patients.
- To identify key patient characteristics influencing disease severity and predict patient outcomes.
Main Methods:
- Utilized a dataset of 8,443 COVID-19 patients with demographic, clinical, and comorbidity information.
- Applied machine learning algorithms to predict disease outcomes (cure or death).
Main Results:
- The best prediction model achieved a Receiver Operating Characteristic Area Under the Curve (AUC) of 0.92.
- The model demonstrated high performance with a Sensitivity of 0.88 and Specificity of 0.82.
Conclusions:
- Machine learning models, integrating patient data, can effectively assist in prognostic prediction for COVID-19.
- This approach supports physician decision-making, enabling faster interventions and mitigating healthcare system strain.
Abstract:
The first officially registered case of COVID-19 in Brazil was on February 26, 2020. Since then, the situation has worsened with more than 672, 000 confirmed cases and at least 36, 000 reported deaths by June 2020. Accurate diagnosis of patients with COVID-19 is extremely important to offer adequate treatment, and avoid overloading the healthcare system. Characteristics of patients such as age, comorbidities and varied clinical symptoms can help in classifying the level of infection severity, predict the disease outcome and the need for hospitalization. Here, we present a study to predict a poor prognosis in positive COVID-19 patients and possible outcomes using machine learning. The study dataset comprises information of 8, 443 patients concerning closed cases due to cure or death. Our experimental results show the disease outcome can be predicted with a Receiver Operating Characteristic AUC of 0.92, Sensitivity of 0.88 and Specificity of 0.82 for the best prediction model. This is a preliminary retrospective study which can be improved with the inclusion of further data. Conclusion: Machine learning techniques fed with demographic and clinical data along with comorbidities of the patients can assist in the prognostic prediction and physician decision-making, allowing a faster response and contributing to the non-overload of healthcare systems.
Related Concept Videos
Steps in Outbreak Investigation
Residuals and Least-Squares Property
If the observed data point lies above the line, the residual is positive, and the line underestimates the actual data value for y. If the observed data point lies below the line, the residual is negative, and the line overestimates the actual data value for y.
The process of fitting the best-fit...
Classification of Illness
An illness is a response to a disease in which the person's level of functioning is changed compared with a previous level. The general classification of illness includes acute and chronic.
Acute illness is severe...
Cancer Survival Analysis
Prediction Intervals
However, the point estimate is most likely not the exact value of the population parameter, but close to it. After calculating point estimates, we construct interval estimates, called confidence intervals or prediction intervals. This prediction interval comprises a range of values unlike the point estimate and is a better predictor of the observed sample value, y.
Statistical Methods for Analyzing Epidemiological Data

