Machine Learning Classifiers and Data Synthesis Techniques to Tackle with Highly Imbalanced COVID-19 Data

Computer and Knowledge Engineering

مقالات آماده انتشار، پذیرفته شده، انتشار آنلاین از تاریخ 16 آبان 1403   

نوع مقاله: Original Article

شناسه دیجیتال (DOI): 10.22067/cke.2024.88940.1121

نویسندگان

Avaz Naghipour^{* 1}؛ Mohammad Reza Abbaszadeh Bavil Soflaei¹؛ mostafa ghader-zefrehei²

¹Department of Computer Engineering, University College of Nabi Akram, Tabriz, Iran

²Department of Animal Science, Faculty of Agriculture, University of Yasouj, Yasouj, Iran

چکیده

The COVID-19 pandemic has highlighted the urgent need for rapid and accurate diagnostic methods. In this study, we evaluate three machine learning models—Random Forest (RF), Logistic Regression (LR) and Decision Tree (DT)—for detecting COVID-19 trained on preprocessed imbalanced datasets with 5086 negative and 558 positive cases. To this end, we demonstrate the capability of two advanced data synthesis algorithms, Conditional Tabular Generative Adversarial Network (CTGAN) and Tabular Variational Autoencoder (TVAE), in addressing the class imbalance inherent in the dataset. The classifiers trained on the original as well as the balanced datasets were evaluated for comparison. Our findings reveal that RF obtains the highest accuracy of 98.83% on the CTGAN-balanced dataset. In conclusion, our results verify the potential of coupling data synthesis with traditional machine learning for the diagnosis of COVID-19. We hope that this research will make a significant contribution to the current AI (Artificial Intelligence) efforts in combating the pandemic.

کلیدواژه‌ها

COVID-19 Detection؛ Machine Learning؛ CTGAN؛ TVAE؛ Class Imbalance