COVID-19 CT-images diagnosis and severity assessment using machine learning algorithm

Zaid Albataineh; Fatima Aldrweesh; Mohammad A Alzubaidi

doi:10.1007/s10586-023-03972-5

COVID-19 CT-images diagnosis and severity assessment using machine learning algorithm

Cluster Comput. 2023 Jan 24:1-16. doi: 10.1007/s10586-023-03972-5. Online ahead of print.

Authors

Zaid Albataineh¹, Fatima Aldrweesh², Mohammad A Alzubaidi²

Affiliations

¹ Department of Electronic Engineering, Yarmouk University, Irbid, 21163 Jordan.
² Department of Computer Engineering, Yarmouk University, Irbid, 21163 Jordan.

Abstract

As a pandemic, the primary evaluation tool for coronavirus (COVID-19) still has serious flaws. To improve the existing situation, all facilities and tools available in this field should be used to combat the pandemic. Reverse transcription polymerase chain reaction is used to evaluate whether or not a person has this virus, but it cannot establish the severity of the illness. In this paper, we propose a simple, reliable, and automatic system to diagnose the severity of COVID-19 from the CT scans into three stages: mild, moderate, and severe, based on the simple segmentation method and three types of features extracted from the CT images, which are ratio of infection, statistical texture features (mean, standard deviation, skewness, and kurtosis), GLCM and GLRLM texture features. Four machine learning techniques (decision trees (DT), K-nearest neighbors (KNN), support vector machines (SVM), and Naïve Bayes) are used to classify scans. 1801 scans are divided into four stages based on the CT findings in the scans and the description file found with the datasets. Our proposed model divides into four steps: preprocessing, feature extraction, classification, and performance evaluation. Four machine learning algorithms are used in the classification step: SVM, KNN, DT, and Naive Bayes. By SVM method, the proposed model achieves 99.12%, 98.24%, 98.73%, and 99.9% accuracy for COVID-19 infection segmentation at the normal, mild, moderate, and severe stages, respectively. The area under the curve of the model is 0.99. Finally, our proposed model achieves better performance than state-of-art models. This will help the doctors know the stage of the infection and thus shorten the time and give the appropriate dose of treatment for this stage.

Keywords: COVID-19; CT scans; Decision tree; KNN; Mild stage; Moderate stage; Naïve Bayes; SVM; Segmentation; Severe stage; The severity of infection.

© The Author(s), under exclusive licence to Springer Science+Business Media, LLC, part of Springer Nature 2023, Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.