A Rolling Bearing Fault Diagnosis Method based on Synchrosqueezed Wavelet Transform and SE-ResNet

Rolling bearings are vital components in rotary machines where even minor faults can lead to the collapse of the entire mechanical system, resulting in significant economic losses. The condition monitoring and fault diagnosis of rolling bearings are crucial for safe operation of complex industrial systems. Traditional fault diagnosis methods mainly depend on expert experience, which leads to obvious deficiencies in diagnostic efficiency, especially in terms of accuracy. To address the issues of traditional bearing fault diagnosis relying on expert experience and low efficiency and accuracy of fault diagnosis, a bearing fault diagnosis model combining synchrosqueezed wavelet transform (SWT) and SE-ResNet is proposed. Firstly, the one-dimensional non-stationary bearing vibration signal is converted into a high-frequency two-dimensional time-frequency image through SWT, which serves as the input for the convolutional neural network; Secondly, an attention mechanism is introduced to construct an SE-ResNet model for extracting two-dimensional image features, which enhances the attention of important features and thus enhances the network's representation ability; Finally, the results of fault diagnosis are output through Softmax layer. To validate the capacity of the proposed model, experiments with the bearing dataset collected from Case Western Reserve University (CWRU) were conducted. The results show that the transfer learning accuracy of the CWRU dataset is 86.25% to 99%. This method has good feasibility, good convergence performance, and has high practical application value.