Abstract:Data imbalance issue generally exists in most medical image analysis problems and maybe getting important with the popularization of data-hungry deep learning paradigms. We explore the cutting-edge Wasserstein generative adversarial networks (WGANs) to address the data imbalance problem with oversampling on the minority classes. The WGAN can estimate the underlying distribution of a minority class to synthesize more plausible and helpful samples for the classification model. In this paper, the WGAN-based over-sampling technique is applied to augment the data to balance for the fine-grained classification of seven semantic attributes of lung nodules in computed tomography images. The fine-grained classification is carried out with a normal convolutional neural network (CNN). To further illustrate the efficacy of the WGAN-based over-sampling technique, the conventional data augmentation method commonly used in many deep learning works, the generative adversarial networks (GANs), and the deep convolutional generative adversarial networks (DCGANs) are implemented for comparison. The whole schemes of the minority oversampling and fine-grained classification are tested with the public lung imaging database consortium dataset. The experimental results suggest that the WGAN-based oversampling technique can synthesize helpful samples for the minority classes to assist the training of the CNN model and to boost the fine-grained classification performance better than the conventional data augmentation method and the two schemes of the GAN and DCGAN techniques do. It may thus suggest that the WGAN technique offers an alternative methodological option for the further deep learning on imbalanced classification studies.

SGBGAN: minority class image generation for class-imbalanced datasets

Conditional Variational Autoencoder with Balanced Pre-training for Generative Adversarial Networks

BAGAN: Data Augmentation with Balancing GAN

An ensemble oversampling method for imbalanced classification with prior knowledge via generative adversarial network

IDA-GAN: A Novel Imbalanced Data Augmentation GAN

CEGAN: Classification Enhancement Generative Adversarial Networks for unraveling data imbalance problems

A new imbalanced data oversampling method based on Bootstrap method and Wasserstein Generative Adversarial Network

Multi-class imbalanced image classification using conditioned GANs

Using Bidirectional GAN with Improved Training Architecture for Imbalanced Tasks

Incremental Focal Loss GANs.

Enhancing and improving the performance of imbalanced class data using novel GBO and SSG: A comparative analysis

An improved generative adversarial network to oversample imbalanced datasets

WGAN-Based Synthetic Minority Over-Sampling Technique: Improving Semantic Fine-Grained Classification for Lung Nodules in CT Images.

Distribution Enhancement for Imbalanced Data with Generative Adversarial Network

BSGAN: A Novel Oversampling Technique for Imbalanced Pattern Recognitions

IB-GAN: A Unified Approach for Multivariate Time Series Classification under Class Imbalance

Class Balancing GAN with a Classifier in the Loop

SSGAN: A Semantic Similarity-Based GAN for Small-Sample Image Augmentation

SMOTified-GAN for class imbalanced pattern classification problems

CMAGAN: classifier-aided minority augmentation generative adversarial networks for industrial imbalanced data and its application to fault prediction

GIU-GANs: Global Information Utilization for Generative Adversarial Networks