Abstract:Purpose The purpose is using generative adversarial network (GAN) to solve the problem of sample augmentation in the case of imbalanced bearing fault data sets and improving residual network is used to improve the diagnostic accuracy of the bearing fault intelligent diagnosis model in the environment of high signal noise. Design/methodology/approach A bearing vibration data generation model based on conditional GAN (CGAN) framework is proposed. The method generates data based on the adversarial mechanism of GANs and uses a small number of real samples to generate data, thereby effectively expanding imbalanced data sets. Combined with the data augmentation method based on CGAN, a fault diagnosis model of rolling bearing under the condition of data imbalance based on CGAN and improved residual network with attention mechanism is proposed. Findings The method proposed in this paper is verified by the western reserve data set and the truck bearing test bench data set, proving that the CGAN-based data generation method can form a high-quality augmented data set, while the CGAN-based and improved residual with attention mechanism. The diagnostic model of the network has better diagnostic accuracy under low signal-to-noise ratio samples. Originality/value A bearing vibration data generation model based on CGAN framework is proposed. The method generates data based on the adversarial mechanism of GAN and uses a small number of real samples to generate data, thereby effectively expanding imbalanced data sets. Combined with the data augmentation method based on CGAN, a fault diagnosis model of rolling bearing under the condition of data imbalance based on CGAN and improved residual network with attention mechanism is proposed.

A Unified Framework Incorporating Predictive Generative Denoising Autoencoder and Deep Coral Network for Rolling Bearing Fault Diagnosis with Unbalanced Data

Combining Kernel Principal Component Analysis and Spatial Group-Wise Enhance Convolutional Neural Network for Fault Recognition of Rolling Element Bearings

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

A Deep Ensemble Dense Convolutional Neural Network for Rolling Bearing Fault Diagnosis

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

Imbalanced Fault Diagnosis of Rolling Bearing Using a Deep Gradient Improved Generative Adversarial Network

Deep dynamic adaptation network: a deep transfer learning framework for rolling bearing fault diagnosis under variable working conditions

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

A Domain Adaptation Method Based on Deep Coral for Rolling Bearing Fault Diagnosis

Fault Diagnosis of Rolling Bearing under Marine Noisy Environments and Varying Working Conditions

Improved Knowledge Transfer Based on Stacked Denoising Autoencoder and Graph Convolutional Network for Fault Diagnosis of Rolling Bearings

Denoising Diffusion Implicit Model Combined with TransNet for Rolling Bearing Fault Diagnosis Under Imbalanced Data

Dual-feature enhanced hybrid convolutional network for imbalanced fault diagnosis of rolling bearings

Fault Diagnosis Method for Rolling Mill Multi Row Bearings Based on AMVMD-MC1DCNN under Unbalanced Dataset

A deep reinforcement learning-based intelligent fault diagnosis framework for rolling bearings under imbalanced datasets

A New Bearing Fault Diagnosis Method Based on Deep Transfer Network and Supervised Joint Matching

Rolling Bearing Fault Diagnosis Using Variational Autoencoding Generative Adversarial Networks with Deep Regret Analysis

Data synthesis using deep feature enhanced generative adversarial networks for rolling bearing imbalanced fault diagnosis

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

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

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