A Channel Attention Based Large Kernel Convolutional Neural Network for Bearing Fault Diagnosis

Fan Li,Liping Wang,Decheng Wang,Jun Wu,Ying Wang,Jiayu Sun
DOI: https://doi.org/10.1109/nnice58320.2023.10105739
2023-01-01
Abstract:We propose an improved end-to-end Channel Attention based Large Kernel Convolutional Neural Network (CALKCNN) in this paper. The CALKCNN can automatically extract features from the original vibration signal and accurately diagnose bearing fault without any manual feature selection operations. We break the stereotype and innovatively use a large kernel in the first layer of CALKCNN to extract the long-time scale characteristics and suppress high-frequency noise, which improves the anti-noise and generalization ability. The CALKCNN can adaptively adjust the weights of feature maps generated by large kernel convolutions based on the channel attention mechanism, selectively focus on essential features and ignore unimportant features, which improves the adaptability and robustness. Compared with five excellent baseline models, the CALKCNN achieves state-of-the-art performance in bearing fault diagnosis by experiments using CWRU bearing fault dataset.
What problem does this paper attempt to address?