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%.

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

BSGAN: A Novel Oversampling Technique for Imbalanced Pattern Recognitions

SMOTified-GAN for class imbalanced pattern classification problems

A Novel SMOTE-Based Classification Approach to Online Data Imbalance Problem

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

A Novel Adaptive Minority Oversampling Technique for Improved Classification in Data Imbalanced Scenarios

Modified-generative adversarial networks for imbalance text classification

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

Distribution Enhancement for Imbalanced Data with Generative Adversarial Network

G-SMOTE: A GMM-based synthetic minority oversampling technique for imbalanced learning.

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

Deep convolutional neural networks with genetic algorithm-based synthetic minority over-sampling technique for improved imbalanced data classification

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

Synthetic Tabular Data Generation for Imbalanced Classification: The Surprising Effectiveness of an Overlap Class

SGBGAN: minority class image generation for class-imbalanced datasets

Improving Classification Performance on Imbalanced Medical Data using Generative Adversarial Network

A Synthetic Minority Oversampling Technique Based on Gaussian Mixture Model Filtering for Imbalanced Data Classification

GMOTE: Gaussian based minority oversampling technique for imbalanced classification adapting tail probability of outliers

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

Imbalanced data enhancement method based on improved DCGAN and its application

Oversampling Method Based on GAN for Tabular Binary Classification Problems.