An Improved Rolling Bearing Fault Diagnosis Model of Long Short-Term Memory Network Based on VMD Denoised Vibration Signals

Abstract

Data driven fault diagnosis methods are increasingly being used for condition monitoring of rotating machinery in the era of Industry 4.0. The effectiveness of Variational Mode Decomposition (VMD) denoised vibration signals in improving the Long Short-Term Memory (LSTM) method for the intelligent fault diagnosis of a rolling bearing is reported. The raw and VMD denoised vibration signals of rolling bearings are provided as inputs to the LSTM network for classification of bearing condition. The efficacy of the methodology to extract the fault information is assessed through datasets obtained from experiment test rig and through the open-source dataset of Case Western Reserve University (CWRU). A comparative analysis is also carried out using both VMD based decomposition and denoising techniques along with four machine learning classifiers viz. Decision Tree, k-Nearest Neighbour (k-NN), Support Vector Machine (SVM) and Artificial Neural Network (ANN) by using the statistical features of the VMD modes. Among the different methods evaluated, VMD denoised signals when fed to LSTM result in the maximum classification accuracy of 99.14 % with experimental dataset. For the case of CWRU dataset, VMD denoised signals as input to the SVM resulted in maximum classification accuracy of 98.16%.