A Novel Rolling Bearing Fault Diagnosis Method Based on BLS and CNN with Attention Mechanism

Xiaojia Wang,Tong Hua,Sheng Xu,Xibin Zhao
DOI: https://doi.org/10.3390/machines11020279
IF: 2.899
2023-02-16
Machines
Abstract:In actual industrial application scenarios, noise pollution makes it difficult to extract fault features accurately via conventional methods. A novel method for rolling bearing fault diagnosis combining a convolutional neural network (CNN), an attention mechanism for a squeeze-and-excitation network (SENet) module and a broad learning system (BLS) is proposed (SECNN–BLS). The one-dimensional bearing vibration signal is processed by using multiple short-time Fourier transforms (STFT); the two-dimensional image in the time-frequency domain is used as the model input. In the CNN fault feature extraction process, SENet is introduced to replace the CNN convolution layer, and global information is obtained through the squeeze operation. Excitation operation captures the importance of image channels, assigns weights adaptively to improve the attention on important information and eliminates the interference of irrelevant features without increasing the spatial and temporal complexity. The weighted feature representation is then transferred to the BLS input. The BLS has the characteristics of a flat structure and ridge regression to quickly solve the weights; as a fault classifier, it can save more computing resources and improve the accuracy of the fault classification. The SECNN–BLS fault diagnosis has achieved more than 98% accuracy on the Society for Machinery Failure Prevention Technology (MFPT) dataset. We also demonstrate the excellent performance of SECNN–BLS in a noisy environment.
engineering, mechanical, electrical & electronic
What problem does this paper attempt to address?