Automated system for classification of COVID-19 infection from lung CT images based on machine learning and deep learning techniques.

Sci Rep

Department of Information Technology, College of Computer and Information Sciences, Majmaah University, Al Majmaah, 11952, Saudi Arabia.

Published: October 2022

AI Article Synopsis

  • The study aimed to segment CT images using a k-means clustering algorithm and extract textural features with the gray level co-occurrence matrix (GLCM).
  • It compared the effectiveness of machine learning classifiers (Naïve Bayes, Bagging, REPTree) to classify images as COVID or non-COVID, alongside three pre-trained CNN models (AlexNet, ResNet50, SqueezeNet).
  • The results showed that the Naïve Bayes classifier achieved a 97% accuracy, while the ResNet50 model led with 99%, indicating that deep learning networks were more effective than traditional machine learning methods for classifying COVID-19 images.

Article Abstract

The objectives of our proposed study were as follows: First objective is to segment the CT images using a k-means clustering algorithm for extracting the region of interest and to extract textural features using gray level co-occurrence matrix (GLCM). Second objective is to implement machine learning classifiers such as Naïve bayes, bagging and Reptree to classify the images into two image classes namely COVID and non-COVID and to compare the performance of the three pre-trained CNN models such as AlexNet, ResNet50 and SqueezeNet with that of the proposed machine learning classifiers. Our dataset consists of 100 COVID and non-COVID images which are pre-processed and segmented with our proposed algorithm. Following the feature extraction process, three machine learning classifiers (Naive Bayes, Bagging, and REPTree) were used to classify the normal and covid patients. We had implemented the three pre-trained CNN models such as AlexNet, ResNet50 and SqueezeNet for comparing their performance with machine learning classifiers. In machine learning, the Naive Bayes classifier achieved the highest accuracy of 97%, whereas the ResNet50 CNN model attained the highest accuracy of 99%. Hence the deep learning networks outperformed well compared to the machine learning techniques in the classification of Covid-19 images.

Download full-text PDF

Source
http://www.ncbi.nlm.nih.gov/pmc/articles/PMC9579174PMC
http://dx.doi.org/10.1038/s41598-022-20804-5DOI Listing

Publication Analysis

Top Keywords

machine learning
28
learning classifiers
16
learning
9
classification covid-19
8
deep learning
8
learning techniques
8
bayes bagging
8
bagging reptree
8
reptree classify
8
covid non-covid
8

Similar Publications

Want AI Summaries of new PubMed Abstracts delivered to your In-box?

Enter search terms and have AI summaries delivered each week - change queries or unsubscribe any time!