Wasserstein GAN based Chest X-Ray Dataset Augmentation for Deep Learning Models: COVID-19 Detection Use-Case
AffiliationAligarh Muslim University; University of Chester; Malaviya National Institute of Technology Jaipur; Keele University
MetadataShow full item record
AbstractThe novel coronavirus infection (COVID-19) is still continuing to be a concern for the entire globe. Since early detection of COVID-19 is of particular importance, there have been multiple research efforts to supplement the current standard RT-PCR tests. Several deep learning models, with varying effectiveness, using Chest X-Ray images for such diagnosis have also been proposed. While some of the models are quite promising, there still remains a dearth of training data for such deep learning models. The present paper attempts to provide a viable solution to the problem of data deficiency in COVID-19 CXR images. We show that the use of a Wasserstein Generative Adversarial Network (WGAN) could lead to an effective and lightweight solution. It is demonstrated that the WGAN generated images are at par with the original images using inference tests on an already proposed COVID-19 detection model.
CitationHussain, B. Z., Andleeb, I., Ansari, M. S., Joshi, A. M., & Kanwal, N. (2022). Wasserstein GAN based chest X-Ray dataset augmentation for deep learning models: COVID-19 detection use-case. 44th Annual International Conference of the IEEE Engineering in Medicine & Biology Society (EMBC), pp. 2058-2061. 10.1109/EMBC48229.2022.9871519.
Description“© 2022 IEEE. Personal use of this material is permitted. Permission from IEEE must be obtained for all other uses, in any current or future media, including reprinting/republishing this material for advertising or promotional purposes, creating new collective works, for resale or redistribution to servers or lists, or reuse of any copyrighted component of this work in other works.”
Except where otherwise noted, this item's license is described as https://creativecommons.org/licenses/by-nc-nd/4.0/