Bearing fault diagnosis with parallel CNN and LSTM

Math Biosci Eng. 2024 Jan 16;21(2):2385-2406. doi: 10.3934/mbe.2024105.

Abstract

Intelligent diagnosis of bearing faults is fundamental to machinery automation and their intelligent operation. Deep learning-based analysis of bearing vibration data has emerged as one research mainstream for fault diagnosis. To enhance the quality of feature extraction from bearing vibration signals and the robustness of the model, we construct a fault diagnostic model based on convolutional neural network (CNN) and long short-term memory (LSTM) parallel network to extract their temporal and spatial features from two perspectives. First, via resampling, vibration signal is split into equal-sized slices which are then converted into time-frequency images by continuous wavelet transform (CWT). Second, LSTM extracts the time-correlation features of 1D signals as one path, and 2D-CNN extracts the local frequency distribution features of time-frequency images as another path. Third, 1D-CNN further extracts integrated features from the fusion features yielded by former parallel paths. Finally, these categories are calculated through the softmax function. According to experimental results, the proposed model has satisfactory diagnostic accuracy and robustness in different contexts on two different datasets.

Keywords: CNN; CWT; LSTM; bearing fault diagnosis; parallel path.