Improved Bearing Fault Diagnosis Based on Deep Convolutional Neural Network with Wide First-Layer Kernels

Rui Huang,Qi Wang,Jianbin Xiong,Xiangjun Dong,Jianxiang Yang,Yipeng Wu
DOI: https://doi.org/10.1109/iotaai62601.2024.10692793
2024-01-01
Abstract:Deep learning (DL) is an indispensable tool in rotating machinery fault diagnosis. In order to enhance the self-learning ability of rotary machine DL and boost the precision of intelligent diagnosis using DL., this paper proposes a DL method based on the combination of deep convolutional neural network with wide first-layer kernels (WDCNN) and Squeeze-and-Excitation (SE) attention mechanism. The SE module is integrated subsequent to the initial wide convolution layer, compressing the feature map into a vector via global average pooling. Then the weight vector is learned by two fully connected layers, which is used to reweight each channel of the original feature graph. CWRU data set is used to verify the effectiveness of SE-WDCNN. The experimental results demonstrate a significant improvement in the performance of the SE-WDCNN network on the CWRU dataset, with an increase to 99%.
What problem does this paper attempt to address?