Abstract
To address the issues of poor noise resistance and insufficient generalization performance in traditional fault diagnosis methods, an end-to-end rolling bearing fault diagnosis method based on Bidirectional Interactive Convolutional Neural Network (BICNN) is proposed. Firstly, the bearing vibration signal is directly input into the wide convolutional kernel for rapid feature extraction, reducing the interference of high-frequency noise. Secondly, a modified Rectified Linear Unit (M-ReLU) activation function is designed to solve the problem of "neuron death" in the ReLU activation function. Then, a bidirectional interactive feature extraction module is constructed, and the features extracted are input into the bidirectional interactive feature extraction module to capture the channel and spatial feature information simultaneously. Next, the extracted information is imported the presented feature enhancement module to achieve more valuable information transmission and accumulation. Finally, a small convolutional kernel is applied to further extract feature information, and a global average pooling layer is used to replace the fully connected layer, reducing the number of parameters while avoiding the problem of model overfitting. The Softmax is utilized to classify the types of bearing faults. Two different datasets are adopted to validate the fault diagnosis performance of the proposed model under − 4dB signal-to-noise ratio and variable working conditions. Experimental results show that compared with other fault diagnosis methods, the proposed model has higher fault diagnosis accuracy, stronger noise resistance, and generalization ability.