Feature Extraction Based on DWT and CNN for Rotating Machinery Fault Diagnosis

Yuan Xie,Tao Zhang
DOI: https://doi.org/10.1109/ccdc.2017.7979176
2017-01-01
Abstract:Fault diagnosis technique and condition monitoring for rotating machinery have become important issues as the development of modern industry. A novel fault diagnosis feature extraction approach is presented for rolling bearing signals based on discrete wavelet transform (DWT) and convolutional neural network (CNN) in this paper. Vibration signals of rolling bearings usually carry dynamic information of the rotating system, therefore are very useful for fault feature extraction. Wavelet analysis is an effective tool for signal processing and feature extraction. However classic fault diagnosis methods based on wavelet analysis only use statistic features or transformation of the decomposed signals as fault feature, there are still rich information contained in the signal yet to explore. In this paper, a two-level DWT is applied on the vibration signals and four components of the signal are acquired. A CNN structure is designed and trained on the frequency spectrums of the four components after Fourier-transform. The extracted features can reflect the characteristic of the original signals and are trained in a softmax classifier for rotating machine fault diagnosis. Experiments are carried out using vibration signals under 52 different machine working conditions. The experiment results indicate that our proposed approach has better accuracy and efficiency than traditional fault diagnose approaches.
What problem does this paper attempt to address?