A rolling bearing fault diagnosis method based on Markov transition field and multi-scale Runge-Kutta residual network

Abstract

In order to address the problem that one- dimensional convolutional neural networks is difficult to extract the local correlation information and mine multi-scale information of rolling bearing fault signals under variable working conditions, a novel fault diagnosis method for rolling bearings based on Markov transition field (MTF) and multi-scale Runge–Kutta residual attention network (MRKRA-Net) is proposed in this paper. Firstly, the original signal is encoded into a two-dimensional image using the MTF method. Then, a multi-scale network is constructed using pre-activation Runge–Kutta residual blocks to extract multi-level features. Secondly, a feature-guided attention mechanism is designed and embedded into the network model to enhance its generalization ability. Finally, the MRKRA-Net model is validated on two different bearing datasets, and the results show that compared with other popular intelligent fault diagnosis methods, MRKRA-Net has higher fault diagnosis accuracy and stronger robustness under both given and variable working conditions.