Abstract:The traditional fault diagnosis methods of rotating machinery based on deep learning have made some achievements. However, the fault samples are generally difficult to collect, which will lead to data imbalance problem. In such cases, it will degrade their diagnosis performances. Therefore, a data rebalancing solution based on Wasserstein Generative Adversarial Network (WGAN) was proposed to improve data quality, and combined with long and short-term memory full convolutional network (LSTM-FCN) to achieve high-dimensional vibration signal fault diagnosis. The method first utilizes WGAN to expand the imbalanced fault samples. The unique one-dimensional convolution architecture in WGAN and the loss function based on Wasserstein distance ensure the stable feature learning ability of the model and the high quality of the generated data. Hence, it fully solves the problem of data imbalance. Subsequently, the balanced high-dimensional fault signals are fed into LSTM-FCN model. LSTM-FCN simultaneously combines the characteristics of LSTM and FCN in timing feature extraction, and thus extracts more robust and representative fault features through parallel feature extraction and fusion technology. Finally, SoftMax classifier is employed to identify different fault conditions. The proposed method is verified on the multiple unbalanced fault datasets of rolling bearings and gears. Compared with the traditional data imbalance resolution algorithms, the experimental results demonstrate that WGAN can effectively deal with the data imbalance problem, and the diagnosis accuracy of LSTM-FCN is maintained at about 99%. In comparison with the traditional fault diagnosis algorithms, the proposed method shows better diagnostic performance and generalization ability under complex working conditions with small samples and strong noise.

Wasserstein Generative Adversarial Network and Convolutional Neural Network (WG-CNN) for Bearing Fault Diagnosis

Research on a Bearing Fault Diagnosis Method Based on an Improved Wasserstein Generative Adversarial Network

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

CWGAN: Conditional Wasserstein Generative Adversarial Nets for Fault Data Generation

Improved Generative Adversarial Network for Bearing Fault Diagnosis with a Small Number of Data and Unbalanced Data

Intelligent Fault Diagnosis of Rotating Machinery Via Wavelet Transform, Generative Adversarial Nets and Convolutional Neural Network

Fault diagnosis of rotating machinery based on combination of Wasserstein generative adversarial networks and long short term memory fully convolutional network

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

Intelligent Fault Diagnosis of Bearings under Variable Working Conditions and Small Samples with Generative Adversarial Network

Rolling Bearing Fault Diagnosis Based on CWT and Two-Stream Convolutional Neural Network

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

A Novel Bearing Fault Diagnosis Method Based on the DLM-CNN Framework

A multi convolution pooling group fault diagnosis model with high generalization across data sets and large receptive field characteristics considering industrial environmental noise

Bearing Fault Diagnosis Method Based on CWT-ResNet18

Rolling Bearing Fault Diagnosis Based on Wavelet Packet Transform and Convolutional Neural Network

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

A Wasserstein Gradient-Penalty Generative Adversarial Network with Deep Auto-Encoder for Bearing Intelligent Fault Diagnosis

A Lightweight Dynamic Dual-Damped Wavelet-Based Convolutional Neural Network for Interpretable Bearing Fault Diagnosis

A bearing fault diagnosis model based on CNN with wide convolution kernels

A New Data-Driven Intelligent Fault Diagnosis by Using Convolutional Neural Network

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