The Detection of COVID-19 in Chest X-rays Using Ensemble CNN Techniques

Advances in the field of image classification using convolutional neural networks (CNNs) have greatly improved the accuracy of medical image diagnosis by radiologists. Numerous research groups have applied CNN methods to diagnose respiratory illnesses from chest X-rays and have extended this work to...

Full description

Bibliographic Details
Main Authors: Domantas Kuzinkovas, Sandhya Clement
Format: Article
Language:English
Published: MDPI AG 2023-06-01
Series:Information
Subjects:
Online Access:https://www.mdpi.com/2078-2489/14/7/370
_version_ 1797588900431003648
author Domantas Kuzinkovas
Sandhya Clement
author_facet Domantas Kuzinkovas
Sandhya Clement
author_sort Domantas Kuzinkovas
collection DOAJ
description Advances in the field of image classification using convolutional neural networks (CNNs) have greatly improved the accuracy of medical image diagnosis by radiologists. Numerous research groups have applied CNN methods to diagnose respiratory illnesses from chest X-rays and have extended this work to prove the feasibility of rapidly diagnosing COVID-19 with high degrees of accuracy. One issue in previous research has been the use of datasets containing only a few hundred images of chest X-rays containing COVID-19, causing CNNs to overfit the image data. This leads to lower accuracy when the model attempts to classify new images, as would be clinically expected. In this work, we present a model trained on the COVID-QU-Ex dataset containing 33,920 chest X-ray images, with an equal share of COVID-19, Non-COVID pneumonia, and Normal images. The model is an ensemble of pre-trained CNNs (ResNet50, VGG19, and VGG16) and GLCM textural features. The model achieved a 98.34% binary classification accuracy (COVID-19/no COVID-19) on a test dataset of 6581 chest X-rays and 94.68% for distinguishing between COVID-19, Non-COVID pneumonia, and normal chest X-rays. The results also demonstrate that a higher 98.82% three-class test accuracy can be achieved using the model if the training dataset only contains a few thousand images. However, the generalizability of the model suffers due to the smaller dataset size. This study highlights the benefits of both ensemble CNN techniques and larger dataset sizes for medical image classification performance.
first_indexed 2024-03-11T00:59:34Z
format Article
id doaj.art-4cda17185d074be09be21d44507a2c51
institution Directory Open Access Journal
issn 2078-2489
language English
last_indexed 2024-03-11T00:59:34Z
publishDate 2023-06-01
publisher MDPI AG
record_format Article
series Information
spelling doaj.art-4cda17185d074be09be21d44507a2c512023-11-18T19:46:39ZengMDPI AGInformation2078-24892023-06-0114737010.3390/info14070370The Detection of COVID-19 in Chest X-rays Using Ensemble CNN TechniquesDomantas Kuzinkovas0Sandhya Clement1School of Biomedical Engineering, The University of Sydney, Sydney, NSW 2006, AustraliaSchool of Biomedical Engineering, The University of Sydney, Sydney, NSW 2006, AustraliaAdvances in the field of image classification using convolutional neural networks (CNNs) have greatly improved the accuracy of medical image diagnosis by radiologists. Numerous research groups have applied CNN methods to diagnose respiratory illnesses from chest X-rays and have extended this work to prove the feasibility of rapidly diagnosing COVID-19 with high degrees of accuracy. One issue in previous research has been the use of datasets containing only a few hundred images of chest X-rays containing COVID-19, causing CNNs to overfit the image data. This leads to lower accuracy when the model attempts to classify new images, as would be clinically expected. In this work, we present a model trained on the COVID-QU-Ex dataset containing 33,920 chest X-ray images, with an equal share of COVID-19, Non-COVID pneumonia, and Normal images. The model is an ensemble of pre-trained CNNs (ResNet50, VGG19, and VGG16) and GLCM textural features. The model achieved a 98.34% binary classification accuracy (COVID-19/no COVID-19) on a test dataset of 6581 chest X-rays and 94.68% for distinguishing between COVID-19, Non-COVID pneumonia, and normal chest X-rays. The results also demonstrate that a higher 98.82% three-class test accuracy can be achieved using the model if the training dataset only contains a few thousand images. However, the generalizability of the model suffers due to the smaller dataset size. This study highlights the benefits of both ensemble CNN techniques and larger dataset sizes for medical image classification performance.https://www.mdpi.com/2078-2489/14/7/370automatic diagnosisCOVID-19neural networkartificial intelligenceensemble machine learning
spellingShingle Domantas Kuzinkovas
Sandhya Clement
The Detection of COVID-19 in Chest X-rays Using Ensemble CNN Techniques
Information
automatic diagnosis
COVID-19
neural network
artificial intelligence
ensemble machine learning
title The Detection of COVID-19 in Chest X-rays Using Ensemble CNN Techniques
title_full The Detection of COVID-19 in Chest X-rays Using Ensemble CNN Techniques
title_fullStr The Detection of COVID-19 in Chest X-rays Using Ensemble CNN Techniques
title_full_unstemmed The Detection of COVID-19 in Chest X-rays Using Ensemble CNN Techniques
title_short The Detection of COVID-19 in Chest X-rays Using Ensemble CNN Techniques
title_sort detection of covid 19 in chest x rays using ensemble cnn techniques
topic automatic diagnosis
COVID-19
neural network
artificial intelligence
ensemble machine learning
url https://www.mdpi.com/2078-2489/14/7/370
work_keys_str_mv AT domantaskuzinkovas thedetectionofcovid19inchestxraysusingensemblecnntechniques
AT sandhyaclement thedetectionofcovid19inchestxraysusingensemblecnntechniques
AT domantaskuzinkovas detectionofcovid19inchestxraysusingensemblecnntechniques
AT sandhyaclement detectionofcovid19inchestxraysusingensemblecnntechniques