DC-GAN-based synthetic X-ray images augmentation for increasing the performance of EfficientNet for COVID-19 detection

Currently, many deep learning models are being used to classify COVID‐19 and normal cases from chest X‐rays. However, the available data (X‐rays) for COVID‐19 is limited to train a robust deep‐learning model. Researchers have used data augmentation techniques to tackle this issue by increasing the n...

Full description

Bibliographic Details
Main Authors: Pir Masoom Shah, Hamid Ullah, Rahim Ullah, Dilawar Shah, Yulin Wang, Saif ul Islam, Abdullah Gani, Rodrigues, Joel J. P. C.
Format: Article
Language:English
English
Published: Wiley-Blackwell Publishing 2021
Subjects:
Online Access:https://eprints.ums.edu.my/id/eprint/32820/1/DC-GAN-based%20synthetic%20X-ray%20images%20augmentation%20for%20increasing%20the%20performance%20of%20EfficientNet%20for%20COVID-19%20detection.pdf
https://eprints.ums.edu.my/id/eprint/32820/2/DC-GAN-based%20synthetic%20X-ray%20images%20augmentation%20for%20increasing%20the%20performance%20of%20EfficientNet%20for%20COVID-19%20detection%20_ABSTRACT.pdf
_version_ 1825714763986894848
author Pir Masoom Shah
Hamid Ullah
Rahim Ullah
Dilawar Shah
Yulin Wang
Saif ul Islam
Abdullah Gani
Rodrigues, Joel J. P. C.
author_facet Pir Masoom Shah
Hamid Ullah
Rahim Ullah
Dilawar Shah
Yulin Wang
Saif ul Islam
Abdullah Gani
Rodrigues, Joel J. P. C.
author_sort Pir Masoom Shah
collection UMS
description Currently, many deep learning models are being used to classify COVID‐19 and normal cases from chest X‐rays. However, the available data (X‐rays) for COVID‐19 is limited to train a robust deep‐learning model. Researchers have used data augmentation techniques to tackle this issue by increasing the numbers of samples through flipping, translation, and rotation. However, by adopting this strategy, the model compromises for the learning of high‐dimensional features for a given problem. Hence, there are high chances of overfitting. In this paper, we used deep‐convolutional generative adversarial networks algorithm to address this issue, which generates synthetic images for all the classes (Normal, Pneumonia, and COVID‐19). To validate whether the generated images are accurate, we used the k‐mean clustering technique with three clusters (Normal, Pneumonia, and COVID‐19). We only selected the X‐ray images classified in the correct clusters for training. In this way, we formed a synthetic dataset with three classes. The generated dataset was then fed to The EfficientNetB4 for training. The experiments achieved promising results of 95% in terms of area under the curve (AUC). To validate that our network has learned discriminated features associated with lung in the X‐rays, we used the Grad‐CAM technique to visualize the underlying pattern, which leads the network to its final decision.
first_indexed 2024-03-06T03:16:39Z
format Article
id ums.eprints-32820
institution Universiti Malaysia Sabah
language English
English
last_indexed 2024-03-06T03:16:39Z
publishDate 2021
publisher Wiley-Blackwell Publishing
record_format dspace
spelling ums.eprints-328202022-06-16T07:33:41Z https://eprints.ums.edu.my/id/eprint/32820/ DC-GAN-based synthetic X-ray images augmentation for increasing the performance of EfficientNet for COVID-19 detection Pir Masoom Shah Hamid Ullah Rahim Ullah Dilawar Shah Yulin Wang Saif ul Islam Abdullah Gani Rodrigues, Joel J. P. C. RC581-951 Specialties of internal medicine Currently, many deep learning models are being used to classify COVID‐19 and normal cases from chest X‐rays. However, the available data (X‐rays) for COVID‐19 is limited to train a robust deep‐learning model. Researchers have used data augmentation techniques to tackle this issue by increasing the numbers of samples through flipping, translation, and rotation. However, by adopting this strategy, the model compromises for the learning of high‐dimensional features for a given problem. Hence, there are high chances of overfitting. In this paper, we used deep‐convolutional generative adversarial networks algorithm to address this issue, which generates synthetic images for all the classes (Normal, Pneumonia, and COVID‐19). To validate whether the generated images are accurate, we used the k‐mean clustering technique with three clusters (Normal, Pneumonia, and COVID‐19). We only selected the X‐ray images classified in the correct clusters for training. In this way, we formed a synthetic dataset with three classes. The generated dataset was then fed to The EfficientNetB4 for training. The experiments achieved promising results of 95% in terms of area under the curve (AUC). To validate that our network has learned discriminated features associated with lung in the X‐rays, we used the Grad‐CAM technique to visualize the underlying pattern, which leads the network to its final decision. Wiley-Blackwell Publishing 2021 Article PeerReviewed text en https://eprints.ums.edu.my/id/eprint/32820/1/DC-GAN-based%20synthetic%20X-ray%20images%20augmentation%20for%20increasing%20the%20performance%20of%20EfficientNet%20for%20COVID-19%20detection.pdf text en https://eprints.ums.edu.my/id/eprint/32820/2/DC-GAN-based%20synthetic%20X-ray%20images%20augmentation%20for%20increasing%20the%20performance%20of%20EfficientNet%20for%20COVID-19%20detection%20_ABSTRACT.pdf Pir Masoom Shah and Hamid Ullah and Rahim Ullah and Dilawar Shah and Yulin Wang and Saif ul Islam and Abdullah Gani and Rodrigues, Joel J. P. C. (2021) DC-GAN-based synthetic X-ray images augmentation for increasing the performance of EfficientNet for COVID-19 detection. Expert Systems, 39. pp. 1-13. https://onlinelibrary.wiley.com/doi/epdf/10.1111/exsy.12823 https://doi.org/10.1111/exsy.12823 https://doi.org/10.1111/exsy.12823
spellingShingle RC581-951 Specialties of internal medicine
Pir Masoom Shah
Hamid Ullah
Rahim Ullah
Dilawar Shah
Yulin Wang
Saif ul Islam
Abdullah Gani
Rodrigues, Joel J. P. C.
DC-GAN-based synthetic X-ray images augmentation for increasing the performance of EfficientNet for COVID-19 detection
title DC-GAN-based synthetic X-ray images augmentation for increasing the performance of EfficientNet for COVID-19 detection
title_full DC-GAN-based synthetic X-ray images augmentation for increasing the performance of EfficientNet for COVID-19 detection
title_fullStr DC-GAN-based synthetic X-ray images augmentation for increasing the performance of EfficientNet for COVID-19 detection
title_full_unstemmed DC-GAN-based synthetic X-ray images augmentation for increasing the performance of EfficientNet for COVID-19 detection
title_short DC-GAN-based synthetic X-ray images augmentation for increasing the performance of EfficientNet for COVID-19 detection
title_sort dc gan based synthetic x ray images augmentation for increasing the performance of efficientnet for covid 19 detection
topic RC581-951 Specialties of internal medicine
url https://eprints.ums.edu.my/id/eprint/32820/1/DC-GAN-based%20synthetic%20X-ray%20images%20augmentation%20for%20increasing%20the%20performance%20of%20EfficientNet%20for%20COVID-19%20detection.pdf
https://eprints.ums.edu.my/id/eprint/32820/2/DC-GAN-based%20synthetic%20X-ray%20images%20augmentation%20for%20increasing%20the%20performance%20of%20EfficientNet%20for%20COVID-19%20detection%20_ABSTRACT.pdf
work_keys_str_mv AT pirmasoomshah dcganbasedsyntheticxrayimagesaugmentationforincreasingtheperformanceofefficientnetforcovid19detection
AT hamidullah dcganbasedsyntheticxrayimagesaugmentationforincreasingtheperformanceofefficientnetforcovid19detection
AT rahimullah dcganbasedsyntheticxrayimagesaugmentationforincreasingtheperformanceofefficientnetforcovid19detection
AT dilawarshah dcganbasedsyntheticxrayimagesaugmentationforincreasingtheperformanceofefficientnetforcovid19detection
AT yulinwang dcganbasedsyntheticxrayimagesaugmentationforincreasingtheperformanceofefficientnetforcovid19detection
AT saifulislam dcganbasedsyntheticxrayimagesaugmentationforincreasingtheperformanceofefficientnetforcovid19detection
AT abdullahgani dcganbasedsyntheticxrayimagesaugmentationforincreasingtheperformanceofefficientnetforcovid19detection
AT rodriguesjoeljpc dcganbasedsyntheticxrayimagesaugmentationforincreasingtheperformanceofefficientnetforcovid19detection