Deep Residual Multiscale Convolutional Neural Network With Attention Mechanism for Bearing Fault Diagnosis Under Strong Noise Environment

• Published in: IEEE sensors journal Vol. 24; no. 6; pp. 9073 - 9081
• Main Authors: Han, Shuzhen, Sun, Shengke, Zhao, Zhanshan, Luan, Ziqian, Niu, Pingjuan
• Format: Journal Article
• Language: English
• Published: New York IEEE 15.03.2024; The Institute of Electrical and Electronics Engineers, Inc. (IEEE)
• Subjects: Accuracy; Artificial neural networks; Attention mechanism; bearing fault diagnosis; Computer architecture; Convolution; Convolutional neural networks; Environment models; Fault detection; Fault diagnosis; Feature extraction; intelligent fault diagnosis; Model accuracy; Neural networks; Sensors; strong noise; Vibrations
• ISSN: 1530-437X; 1558-1748
• DOI: 10.1109/JSEN.2023.3345400

In recent years, deep learning (DL) methods have gained much success in the area of intelligent fault diagnosis. However, due to the fact that the working conditions are various and the noise is inevitable, degradation of previous model is very serious. To address the challenge of bearing fault detection under strong noise environment, this article proposed a novel antinoise deep residual multiscale convolutional neural network with attention mechanism named Attention-MSCNN. First, dynamic dropout is used to improve the antinoise ability by introducing artificial noise into the training process. In addition, we design a residual connection between input and the convolved features to fully capture the characteristics of the initial input. Finally, a novel denoised multihead attention mechanism is applied to remove excess noise in raw input and obtain the relationships between long time series. The experimental results show that Attention-MSCNN can achieve robust anti strong noise performance with over 85% accuracy on the Case Western Reserve University (CWRU) dataset. On the self-collected two-stage gear drive test bench, our model achieves an accuracy of over 99% under strong noise environment. Thus, Attention-MSCNN successfully solves the problem of low detection accuracy of previous models under strong noise environment.