Abstract:To address the problems of existing 2D image-based imbalanced fault diagnosis methods for rolling bearings, which generate images with inadequate texture details and color degradation, this paper proposes a novel image enhancement model based on a dual-branch generative adversarial network (GAN) combining spatial and frequency domain information for an imbalanced fault diagnosis of rolling bearing. Firstly, the original vibration signals are converted into 2D time–frequency (TF) images by a continuous wavelet transform, and a dual-branch GAN model with a symmetric structure is constructed. One branch utilizes an auxiliary classification GAN (ACGAN) to process the spatial information of the TF images, while the other employs a GAN with a frequency generator and a frequency discriminator to handle the frequency information of the input images after a fast Fourier transform. Then, a shuffle attention (SA) module based on an attention mechanism is integrated into the proposed model to improve the network's expression ability and reduce the computational burden. Simultaneously, mean square error (MSE) is integrated into the loss functions of both generators to enhance the consistency of frequency information for the generated images. Additionally, a Wasserstein distance and gradient penalty are also incorporated into the losses of the two discriminators to prevent gradient vanishing and mode collapse. Under the supervision of the frequency WGAN-GP branch, an ACWGAN-GP can generate high-quality fault samples to balance the dataset. Finally, the balanced dataset is utilized to train the auxiliary classifier to achieve fault diagnosis. The effectiveness of the proposed method is validated by two rolling bearing datasets. When the imbalanced ratios of the four datasets are 0.5, 0.2, 0.1, and 0.05, respectively, their average classification accuracy reaches 99.35% on the CWRU bearing dataset. Meanwhile, the average classification accuracy reaches 96.62% on the MFS bearing dataset.

FMRGAN: Feature Mapping Reconstruction GAN for Rolling Bearings Fault Diagnosis under Limited Data Condition

A Bearing Fault Diagnosis Method for High-Noise and Unbalanced Dataset

Imbalanced fault diagnosis of rolling bearing using improved MsR-GAN and feature enhancement-driven CapsNet

Data-augmented wavelet capsule generative adversarial network for rolling bearing fault diagnosis

An Efficient Method Based on Conditional Generative Adversarial Networks for Imbalanced Fault Diagnosis of Rolling Bearing

Image Enhancement Based on Dual-Branch Generative Adversarial Network Combining Spatial and Frequency Domain Information for Imbalanced Fault Diagnosis of Rolling Bearing

Data Augmentation Integrated with Feature-Enhanced Convolutional Neural Network for Imbalanced Fault Diagnosis in Rolling Bearings

Imbalanced data fault diagnosis of rolling bearings using enhanced relative generative adversarial network

A Novel Method for Fault Diagnosis of Bearings with Small and Imbalanced Data Based on Generative Adversarial Networks

Unsupervised Feature-Preserving CycleGAN for Fault Diagnosis of Rolling Bearings Using Unbalanced Infrared Thermal Imaging Sample

Rolling bearing fault diagnosis method based on generative adversarial enhanced multi-scale convolutional neural network model

An interpretable data augmentation scheme for machine fault diagnosis based on a sparsity-constrained generative adversarial network

Conditional GAN and 2-D CNN for Bearing Fault Diagnosis With Small Samples

Limited fault data augmentation with compressed sensing for bearing fault diagnosis

Multi-mode data augmentation and fault diagnosis of rotating machinery using modified ACGAN designed with new framework

Rolling Bearing Fault Diagnosis Based on Improved GAN and 2-D Representation of Acoustic Emission Signals

Bearing Fault Diagnosis Using ACWGAN-GP Enhanced by Principal Component Analysis

Generative adversarial networks for data augmentation in machine fault diagnosis

Improvement of Generative Adversarial Network and Its Application in Bearing Fault Diagnosis: A Review

Two-Stage Multi-Scale Fault Diagnosis Method for Rolling Bearings with Imbalanced Data

Imbalanced Sample Fault Diagnosis of Rolling Bearing Using Deep Condition Multidomain Generative Adversarial Network