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.

Fault Diagnosis of Rolling Bearings in Primary Mine Fans under Sample Imbalance Conditions

Intelligent Rolling Bearing Fault Diagnosis Method Using Symmetrized Dot Pattern Images and CBAM-DRN

Rolling Bearing Fault Diagnosis Model Based on External Attention Integrated Convolutional Neural Network under Imbalanced Data Conditions

Intelligent fault diagnosis of rolling bearings under imbalanced data conditions using attention-based deep learning method

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

A novel fault diagnosis method of rotating machinery via VMD, CWT and improved CNN

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

Fault Diagnosis of Rolling Bearings Based on the Improved Symmetrized Dot Pattern Enhanced Convolutional Neural Networks

Rolling Bearing Fault Diagnosis Under Data Imbalance and Variable Speed Based on Adaptive Clustering Weighted Oversampling

Intelligent fault diagnosis of rolling bearings based on normalized CNN considering data imbalance and variable working conditions

An imbalance multi-faults data transfer learning diagnosis method based on finite element simulation optimization model of rolling bearing

Imbalanced Bearing Fault Diagnosis Based on RFH-GAN and PSA-DRSN

A fault diagnosis method based on an improved diffusion model under limited sample conditions

Bearing Fault Diagnosis Method Based on Deep Convolutional Neural Network and Random Forest Ensemble Learning

Intelligent fault diagnosis of rolling bearing under unbalanced samples based on simulation data fusion

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

A rolling bearing fault diagnosis method based on interactive generative feature space oversampling-based autoencoder under imbalanced data

Bearing Fault Diagnosis of Hot-Rolling Mill Utilizing Intelligent Optimized Self-Adaptive Deep Belief Network with Limited Samples

Rolling Bearing Fault Diagnosis Based on Convolutional Neural Network and Support Vector Machine

A fault diagnosis method for nuclear power plants rotating machinery based on deep learning under imbalanced samples

Rolling Bearing Fault Diagnosis Based on Convolutional Neural Network and Multi-sensor Information Fusion