GAN-based Data Augmentation for Chest X-ray Classification
- URL: http://arxiv.org/abs/2107.02970v1
- Date: Wed, 7 Jul 2021 01:36:48 GMT
- Title: GAN-based Data Augmentation for Chest X-ray Classification
- Authors: Shobhita Sundaram and Neha Hulkund
- Abstract summary: Generative Adrialversa Networks (GANs) offer a novel method of synthetic data augmentation.
GAN-based augmentation leads to higher downstream performance for underrepresented classes.
This suggests that GAN-based augmentation a promising area of research to improve network performance when data collection is prohibitively expensive.
- Score: 0.0
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: A common problem in computer vision -- particularly in medical applications
-- is a lack of sufficiently diverse, large sets of training data. These
datasets often suffer from severe class imbalance. As a result, networks often
overfit and are unable to generalize to novel examples. Generative Adversarial
Networks (GANs) offer a novel method of synthetic data augmentation. In this
work, we evaluate the use of GAN- based data augmentation to artificially
expand the CheXpert dataset of chest radiographs. We compare performance to
traditional augmentation and find that GAN-based augmentation leads to higher
downstream performance for underrepresented classes. Furthermore, we see that
this result is pronounced in low data regimens. This suggests that GAN-based
augmentation a promising area of research to improve network performance when
data collection is prohibitively expensive.
Related papers
- Additional Look into GAN-based Augmentation for Deep Learning COVID-19
Image Classification [57.1795052451257]
We study the dependence of the GAN-based augmentation performance on dataset size with a focus on small samples.
We train StyleGAN2-ADA with both sets and then, after validating the quality of generated images, we use trained GANs as one of the augmentations approaches in multi-class classification problems.
The GAN-based augmentation approach is found to be comparable with classical augmentation in the case of medium and large datasets but underperforms in the case of smaller datasets.
arXiv Detail & Related papers (2024-01-26T08:28:13Z) - Generative adversarial networks for data-scarce spectral applications [0.0]
We report on an application of GANs in the domain of synthetic spectral data generation.
We show that CWGANs can act as a surrogate model with improved performance in the low-data regime.
arXiv Detail & Related papers (2023-07-14T16:27:24Z) - LD-GAN: Low-Dimensional Generative Adversarial Network for Spectral
Image Generation with Variance Regularization [72.4394510913927]
Deep learning methods are state-of-the-art for spectral image (SI) computational tasks.
GANs enable diverse augmentation by learning and sampling from the data distribution.
GAN-based SI generation is challenging since the high-dimensionality nature of this kind of data hinders the convergence of the GAN training yielding to suboptimal generation.
We propose a statistical regularization to control the low-dimensional representation variance for the autoencoder training and to achieve high diversity of samples generated with the GAN.
arXiv Detail & Related papers (2023-04-29T00:25:02Z) - Performance of GAN-based augmentation for deep learning COVID-19 image
classification [57.1795052451257]
The biggest challenge in the application of deep learning to the medical domain is the availability of training data.
Data augmentation is a typical methodology used in machine learning when confronted with a limited data set.
In this work, a StyleGAN2-ADA model of Generative Adversarial Networks is trained on the limited COVID-19 chest X-ray image set.
arXiv Detail & Related papers (2023-04-18T15:39:58Z) - Leveraging GANs for data scarcity of COVID-19: Beyond the hype [1.0957528713294873]
Artificial Intelligence (AI)-based models can help in diagnosing COVID-19 from lung CT scans and X-ray images.
Many researchers studied Geneversa Adrative Networks (GANs) for producing synthetic lung CT scans and X-Ray images.
It is not well explored how good GAN-based methods performed to generate reliable synthetic data.
arXiv Detail & Related papers (2023-04-07T08:26:12Z) - Augmentation-Aware Self-Supervision for Data-Efficient GAN Training [68.81471633374393]
Training generative adversarial networks (GANs) with limited data is challenging because the discriminator is prone to overfitting.
We propose a novel augmentation-aware self-supervised discriminator that predicts the augmentation parameter of the augmented data.
We compare our method with state-of-the-art (SOTA) methods using the class-conditional BigGAN and unconditional StyleGAN2 architectures.
arXiv Detail & Related papers (2022-05-31T10:35:55Z) - Deceive D: Adaptive Pseudo Augmentation for GAN Training with Limited
Data [125.7135706352493]
Generative adversarial networks (GANs) typically require ample data for training in order to synthesize high-fidelity images.
Recent studies have shown that training GANs with limited data remains formidable due to discriminator overfitting.
This paper introduces a novel strategy called Adaptive Pseudo Augmentation (APA) to encourage healthy competition between the generator and the discriminator.
arXiv Detail & Related papers (2021-11-12T18:13:45Z) - Class Balancing GAN with a Classifier in the Loop [58.29090045399214]
We introduce a novel theoretically motivated Class Balancing regularizer for training GANs.
Our regularizer makes use of the knowledge from a pre-trained classifier to ensure balanced learning of all the classes in the dataset.
We demonstrate the utility of our regularizer in learning representations for long-tailed distributions via achieving better performance than existing approaches over multiple datasets.
arXiv Detail & Related papers (2021-06-17T11:41:30Z) - Evaluation of Deep Convolutional Generative Adversarial Networks for
data augmentation of chest X-ray images [0.0]
Medical image datasets are usually imbalanced, due to the high costs of obtaining the data and time-consuming annotations.
In this work, we performed data augmentation on the Chest X-rays dataset through generative modeling.
arXiv Detail & Related papers (2020-09-02T16:43:55Z) - On Data Augmentation for GAN Training [39.074761323958406]
We propose Data Augmentation Optimized for GAN (DAG) to enable the use of augmented data in GAN training.
We conduct experiments to apply DAG to different GAN models.
When DAG is used in some GAN models, the system establishes state-of-the-art Frechet Inception Distance (FID) scores.
arXiv Detail & Related papers (2020-06-09T15:19:26Z) - Data Augmentation using Generative Adversarial Networks (GANs) for
GAN-based Detection of Pneumonia and COVID-19 in Chest X-ray Images [0.0]
We propose a new GAN architecture for augmentation of chest X-rays for semi-supervised detection of pneumonia and COVID-19.
We show that the proposed GAN can be used to effectively augment data and improve classification accuracy of disease in chest X-rays for pneumonia and COVID-19.
arXiv Detail & Related papers (2020-06-05T18:30:20Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.