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.

A novel method based on nonlinear auto-regression neural network and convolutional neural network for imbalanced fault diagnosis of rotating machinery

A novel data-driven method based on sample reliability assessment and improved CNN for machinery fault diagnosis with non-ideal data

An Imbalanced Fault Diagnosis Method Based on TFFO and CNN for Rotating Machinery

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

Fault Diagnosis Method for Imbalanced Data of Rotating Machinery Based on Time Domain Signal Prediction and SC-ResNeSt

Fault Diagnosis for Rotating Machinery Based on Convolutional Neural Network and Empirical Mode Decomposition

A Novel Deep Learning Method for Intelligent Fault Diagnosis of Rotating Machinery Based on Improved CNN-SVM and Multichannel Data Fusion

Ensemble Dilated Convolutional Neural Network and Its Application in Rotating Machinery Fault Diagnosis

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

A Hierarchical Training-Convolutional Neural Network for Imbalanced Fault Diagnosis in Complex Equipment

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

Intelligent fault diagnosis of rotating machinery based on continuous wavelet transform-local binary convolutional neural network

A new bearing fault diagnosis method based on modified convolutional neural networks

Fault Diagnosis of Rotating Machinery Based on the Improved Multidimensional Normalization ResNet

A novel diagnostic framework based on vibration image encoding and multi-scale neural network

A Fault Diagnosis Method Based on a Rainbow Recursive Plot and Deep Convolutional Neural Networks

A fault diagnosis method based on improved convolutional neural network for bearings under variable working conditions

An Adaptive Weighted Multiscale Convolutional Neural Network for Rotating Machinery Fault Diagnosis under Variable Operating Conditions

Application of Multiscale Learning Neural Network Based on CNN in Bearing Fault Diagnosis

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