A new fault diagnosis model of rolling element bearing based on a recurrent neural network

Abstract

The rolling element bearing is an important part of mechanical equipment, it has various kinds of malfunctions, the location of the fault may occur in the inner ring, outer ring, or rolling element of the bearing. Therefore, traditional methods of classification are difficult to classify and identify effectively. To improve the accuracy of bearing fault diagnosis, the deep learning method is used to diagnose the fault of the rolling element bearing. In this paper, the long short-term memory and gated recurrent unit are combined to build a bearing fault diagnosis model. On the other hand, this paper adjusts the hidden layer structure and optimizes the network parameters to establish a better long short-term memory–gated recurrent unit–long short-term memory diagnostic model and classify the fault types of bearings with Softmax. The model proposed in this paper can effectively diagnose the bearing fault under the bearing data set of Case Western Reserve University and the University of Cincinnati. Compared with the traditional long short-term memory and the gated recurrent unit, the model proposed in this paper has high accuracy in fault diagnosis as well as certain reliability and generalization ability.