Intelligent Fault Diagnosis of Rolling Bearings under Small Samples Based on Lightweight UNet with Attention-Fused Residual Block
Xiaochun Sun,Hua Ding,Ning Li,Xiaoxin Dong,Yaoyao Liao,Zeping Liu,Xin Bai
DOI: https://doi.org/10.1088/1361-6501/ad3a08
IF: 2.398
2024-04-04
Measurement Science and Technology
Abstract:In rotating machinery, rolling bearings are crucial, and their failure can lead to severe acci-dents and economic losses. Therefore, fault diagnosis of bearings is necessary to maintain the safe and stable operation of modern machinery and equipment. Currently, data-driven intelligent diagnosis methods are mainstream; however, in practical applications, problems such as insuffi-cient fault samples and strong interference signals often exist. At the same time, a large number of edge-end and mobile devices put higher requirements for the size of the diagnostic model. This study addresses these issues by proposing a lightweight UNet (LWUNet) model, which integrates wavelet packet decomposition (WPD) and attention-fused residual block (AFRB) for fault diag-nosis in rolling bearings under varying operating conditions, particularly for small sample sizes. Firstly, WPD is used to decompose and reconstruct the fault signal to achieve effective denoising. Secondly, a lightweight UNet network (LW-UNet) is constructed for pixel-level feature learning to reduce the number of parameters and improve the accuracy rate. Thereafter, to further enhance the model feature extraction capability, the attention-fused residual block is proposed and em-bedded into the LWUNet to develop the AFRB-LWUNet model. Finally, the reconstructed signals are input to the proposed model for training, and the model performance is examined using a test set. The proposed method is compared with other fault diagnosis models using small sample data of rolling bearings from the Case Western Reserve University (CWRU), USA and the University of Paderborn, Germany. The results confirm the higher recognition accuracy, stronger generaliza-tion ability, and robustness of the proposed method for small samples under various working conditions and intense noise.
engineering, multidisciplinary,instruments & instrumentation