A new bearing fault diagnosis method using elastic net transfer learning and LSTM

Abstract

Although the existing transfer learning method based on deep learning can realize bearing fault diagnosis under variable load working conditions, it is difficult to obtain bearing fault data and the training data of fault diagnosis model is insufficient£¬which leads to the low accuracy and generalization ability of fault diagnosis model, A fault diagnosis method based on improved elastic net transfer learning under variable load working conditions is proposed. The improved elastic net transfer learning is used to suppress the over fitting and improve the training efficiency of the model, and the long short-term memory network is introduced to train the fault diagnosis model, then a small amount of target domain data is used to fine tune the model parameters. Finally, the fault diagnosis model under variable load working conditions based on improved elastic net transfer learning is constructed. Finally, through model experiments and comparison with conventional deep learning fault diagnosis models such as long short-term memory network (LSTM), gated recurrent unit (GRU) and Bi-LSTM, it shows that the proposed method has higher accuracy and better generalization ability, which verifies the effectiveness of the method.