Abstract:Machinery operates well under normal conditions in most cases; far fewer samples are collected in a fault state (minority samples) than in a normal state, resulting in an imbalance of samples. Common machine learning algorithms such as deep neural networks require a significant amount of data during training to avoid overfitting. These models often fail to detect minority samples when the input samples are imbalanced, which results in missed diagnoses of equipment faults. As an effective method to enhance minority samples, Deep Convolution Generative Adversarial Network (DCGAN) does not fundamentally address the problem of unstable Generative Adversarial Network (GAN) training This study proposes an improved DCGAN model with improved stability and sample balance for achieving greater classification accuracy over minority samples. First, spectral normalization is performed on each convolutional layer, improving stability in the DCGAN discriminator. Then, the improved DCGAN model is trained to generate new samples that are different from the original samples but with a similar distribution when the Nash equilibrium is reached. Four indices-Inception Score (IS), Frechet Inception Distance Score (FID), Peak Signal to Noise Ratio (PSNR), and Structural Similarity (SSIM)-were used to quantitatively evaluate of the generated images. Finally, the Balance Degree of Samples (BDS) index was proposed, and the new samples are proportionally added to the original samples to improve sample balance, resulting in the formation of several groups of datasets with different balance degrees, and Convolutional Neural Network (CNN) models are used to classify these samples. With experimental analysis on the reciprocating compressor, the variance of lost data is found to be less than 1% of the original value, representing an increase in stability of the model to generate diverse and high-quality sample images, as compared with that of the unmodified model. The classification accuracy exceeds 95% and tends to remain stable when the balance degree of samples is greater than 80%.

An improved generative adversarial network to oversample imbalanced datasets

Distribution Enhancement for Imbalanced Data with Generative Adversarial Network

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

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

Oversampling Method Based on GAN for Tabular Binary Classification Problems.

Conditional Wasserstein generative adversarial network-gradient penalty-based approach to alleviating imbalanced data classification

Annealing Genetic GAN for Minority Oversampling

An Improved D2GAN‐based oversampling algorithm for imbalanced data classification

Generative adversarial minority enlargement—A local linear over-sampling synthetic method

Using Bidirectional GAN with Improved Training Architecture for Imbalanced Tasks

Global-And-Local Aware Data Generation For The Class Imbalance Problem

IDA-GAN: A Novel Imbalanced Data Augmentation GAN

BSGAN: A Novel Oversampling Technique for Imbalanced Pattern Recognitions

GenSample: A Genetic Algorithm for Oversampling in Imbalanced Datasets

Imbalanced data enhancement method based on improved DCGAN and its application

Incremental Focal Loss GANs.

Oversampling Imbalanced Data Based on Convergent WGAN for Network Threat Detection

Over-sampling method for tackling class imbalance in software defect prediction based on generative adversarial networks

SGBGAN: minority class image generation for class-imbalanced datasets

Improving GAN Training via Feature Space Shrinkage

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