Deep learning method for remaining useful life prediction of the rolling bearing of high-speed train

Abstract

In this paper, an optimized long short-term memory (LSTM) network is proposed for the remaining useful life (RUL) prediction of the rolling bearings based on whale optimized algorithm (WOA). The multi-domain features are extracted to construct the feature dataset as the single domain features are difficult to characterize the performance degeneration of the rolling bearing. Considering the possible gradient explosion by training of the rolling bearing lifetime data and the difficulties in selecting the key network parameters, an optimized LSTM network, namely, WOA-LSTM network is proposed. Experiment results show that, compared with the LSTM network, the RUL prediction accuracy of the rolling bearing are improved by the proposed WOA-LSTM network.