Attention mechanism-guided residual convolution variational autoencoder for bearing fault diagnosis under noisy environments

Abstract Due to rolling bearings usually operate under fluctuating working conditions in practical engineering, the raw vibration signals generated by bearing faults have nonlinear and non-stationary characteristics. Additionally, there is a lot of noise interference in the collected bearing vibration signal, which indicates that it is difficult to extract bearing fault information and obtain a satisfactory diagnosis accuracy via using traditional method. Deep learning provides a shining road to address this issue. Nevertheless, traditional deep network model has the shortcomings of poor generalization performance and weak robustness in the feature learning. To improve fault recognition accuracy and obtain a favorable anti-noise robustness, this paper proposes a novel bearing fault diagnosis approach based on attention mechanism-guided residual convolutional variational autoencoder (AM-RCVAE). Firstly, the improved residual module is constructed to overcome the convergence difficulty problem caused by network degradation and promote the model generalization performance by replacing the batch normalization (BN) layer in the traditional residual module with the adaptive BN layer. Subsequently, by incorporating the convolutional block attention module and the improved residual module into convolutional variational autoencoder, a deep network model termed as AM-RCVAE is presented to automatically learn fault features from the original data and perform fault diagnosis tasks. The effectiveness of the proposed approach is verified via two experimental cases. Moreover, the recognition accuracy and diagnostic performance of the proposed approach have been certain improved compared with several representative methods.