Rolling bearing fault diagnosis based on improved convolutional neural networks with attention mechanism under strong noise

Zhiyao Tang; Gang Li

doi:10.1117/12.2682130

16 June 2023 Rolling bearing fault diagnosis based on improved convolutional neural networks with attention mechanism under strong noise

Zhiyao Tang, Gang Li

Author Affiliations +

Proceedings Volume 12639, Third International Conference on Mechanical Design and Simulation (MDS 2023); 126392P (2023) https://doi.org/10.1117/12.2682130
Event: Third International Conference on Mechanical Design and Simulation (MDS 2023), 2023, Xi'an, China

Abstract

Aiming at the problem that the vibration signal characteristics of rolling bearings are unstable under the background of strong noise, resulting in poor generalization of the diagnostic algorithm, weak noise resistance, and difficulty in achieving effective fault diagnosis, a new method named improved convolutional neural networks with attention mechanism under high noise is proposed. Based on the deep convolutional neural network, the gated recurrent unit (GRU) is introduced to solve the gradient explosion problem in the neural network, the attention mechanism (attention) is introduced to improve the adaptive ability of the network and reduce the difficulty of hyperparameter selection, and the SVM classifier is used to replace the classification layer of the deep convolutional neural network to improve the accuracy of classification. In order to verify the robustness and generalization of the proposed method in a strong noise environment, the bearing dataset of Western Reserve University was used for verification. The experimental results show that the classification accuracy of the proposed algorithm is 24.7 percentage points higher than that of the WDCNN algorithm, which proves that the proposed method has good noise immunity and generalization under the background of high noise.

Citation Download Citation

Zhiyao Tang and Gang Li "Rolling bearing fault diagnosis based on improved convolutional neural networks with attention mechanism under strong noise", Proc. SPIE 12639, Third International Conference on Mechanical Design and Simulation (MDS 2023), 126392P (16 June 2023); https://doi.org/10.1117/12.2682130

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available