Rolling Bearing Fault Diagnosis based on Correlated Channel Attention-Optimized Convolutional Neural Networks
jing zhu,Ou Li,Minghui Chen,Lili Xing
DOI: https://doi.org/10.1088/1361-6501/ad5a2e
IF: 2.398
2024-06-21
Measurement Science and Technology
Abstract:In the field of intelligent fault diagnosis, traditional convolutional neural network (CNN)-based models for rolling bearing fault diagnosis effectively extract signal features, but they perform poorly in identifying weak fault features in strong noise environments. This study proposes a fault diagnosis method based on Correlated Channel Attention Optimized Deep Convolutional Neural Network (CAOCNN), aimed at addressing this challenge. The proposed method includes several innovations: firstly, expanding the width of the convolutional kernels to increase the receptive field for feature extraction, effectively extracting one-dimensional vibration signal features and suppressing high-frequency noise. Secondly, introducing a correlated channel attention mechanism to enhance focus on important features, thereby improving the model's generalization ability. Additionally, the Nesterov momentum optimization algorithm is used to optimize network model parameters, enhancing classification accuracy. Experimental results demonstrate that CAOCNN achieves outstanding performance on the rolling bearing datasets from Case Western Reserve University and Xi'an Jiaotong University, reaching high accuracies of 99.71% and 100%, respectively. Under strong noise conditions, T-SNE visual analysis shows CAOCNN's excellent robustness and feature classification capabilities. These achievements prove CAOCNN's effective diagnostic ability for rolling bearing faults in complex noise environments, contributing valuable advancements to intelligent fault diagnosis technology.
engineering, multidisciplinary,instruments & instrumentation