Multi-scale residual neural network with enhanced gated recurrent unit for fault diagnosis of rolling bearing

Abstract

As the key component of rotating machinery, effective and reliable fault diagnosis of rolling bearing is particularly critical for promoting production safety and economic benefits. The powerful representation learning ability of convolutional neural network (CNN) enables it to effectively extract fault information from vibration signals of rolling bearing. Nevertheless, challenges are faced by CNN in extracting features at multi-scale and capturing temporal features. With regard to this issue, a hybrid deep learning model that incorporates the multi-scale residual neural network (MSRN) with the enhanced gated recurrent unit (EGRU), namely MSRN-EGRU, is proposed in this paper. To begin with, MSRN is designed by introducing multi-scale structure and residual connections into CNN for extracting local features effectively and improving the feature representation of the model. Then, the extracted local features are input into EGRU to further extract temporal features, where EGRU is proposed by improving GRU structure and embedding scaled exponential liner unit (SELU), which enhances the nonlinear modeling and memory ability. Eventually, the obtained features are processed by α-Dropout and global average pooling before being inputted into the softmax layer for fault diagnosis. To validate the effectiveness of the proposed model, three baseline models and two ablation models were employed for comparative experiments with two bearing datasets. The experimental results reveal that the proposed model achieves commendable performance in terms of accuracy, robustness, and convergence for fault diagnosis of rolling bearing.