COVID-19 Probability Prediction Using Machine Learning: An Infectious Approach

TL;DR

This study evaluates various machine learning models, especially deep neural networks, for predicting COVID-19 infection probability, demonstrating that DNN achieves the highest accuracy of 89% in early detection using a dataset of 4000 samples.

Contribution

It provides a comprehensive comparison of multiple ML models for COVID-19 prediction, highlighting the superior performance of deep neural networks in this context.

Findings

DNN achieved 89% accuracy in COVID-19 prediction.

Deep learning models outperform traditional ML algorithms.

The study emphasizes the potential of ML in early pandemic detection.

Abstract

The ongoing COVID-19 pandemic continues to pose significant challenges to global public health, despite the widespread availability of vaccines. Early detection of the disease remains paramount in curbing its transmission and mitigating its impact on public health systems. In response, this study delves into the application of advanced machine learning (ML) techniques for predicting COVID-19 infection probability. We conducted a rigorous investigation into the efficacy of various ML models, including XGBoost, LGBM, AdaBoost, Logistic Regression, Decision Tree, RandomForest, CatBoost, KNN, and Deep Neural Networks (DNN). Leveraging a dataset comprising 4000 samples, with 3200 allocated for training and 800 for testing, our experiment offers comprehensive insights into the performance of these models in COVID-19 prediction. Our findings reveal that Deep Neural Networks (DNN) emerge as the top-performing model, exhibiting superior accuracy and recall metrics. With an impressive accuracy rate of 89%, DNN demonstrates remarkable potential in early COVID-19 detection. This underscores the efficacy of deep learning approaches in leveraging complex data patterns to identify COVID-19 infections accurately. This study underscores the critical role of machine learning, particularly deep learning methodologies, in augmenting early detection efforts amidst the ongoing pandemic. The success of DNN in accurately predicting COVID-19 infection probability highlights the importance of continued research and development in leveraging advanced technologies to combat infectious diseases.