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

CDCGAN: Class Distribution-aware Conditional GAN-based Minority Augmentation for Imbalanced Node Classification

Statistics Enhancement Generative Adversarial Networks for Diverse Conditional Image Synthesis

ImGAGN:Imbalanced Network Embedding via Generative Adversarial Graph Networks

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

Synthetic Over-sampling for Imbalanced Node Classification with Graph Neural Networks

ECGN: A Cluster-Aware Approach to Graph Neural Networks for Imbalanced Classification

A GAN Approach for Node Embedding in Heterogeneous Graphs Using Subgraph Sampling

IDA-GAN: A Novel Imbalanced Data Augmentation GAN

Cluster-based Generative Adversarial Network Imbalanced Data Generation Method

GraphSMOTE: Imbalanced Node Classification on Graphs with Graph Neural Networks

GraphSR: A Data Augmentation Algorithm for Imbalanced Node Classification

Wacml: based on graph neural network for imbalanced node classification algorithm

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

CGGM: A conditional graph generation model with adaptive sparsity for node anomaly detection in IoT networks

Feature Augmentation for Imbalanced Classification with Conditional Mixture WGANs.

Supervised Class Distribution Learning for GANs-Based Imbalanced Classification

A novel graph oversampling framework for node classification in class-imbalanced graphs

SGBGAN: minority class image generation for class-imbalanced datasets

Minority-Weighted Graph Neural Network for Imbalanced Node Classification in Social Networks of Internet of People

Distance-wise Prototypical Graph Neural Network in Node Imbalance Classification

Imbalanced data enhancement method based on improved DCGAN and its application