Residual Life Prediction of Rolling Bearings Based on a CEEMDAN Algorithm Fused with CNN–Attention-Based Bidirectional LSTM Modeling

Abstract

This paper presents a methodology for predicting the remaining usability of rolling bearings. The method combines a fully adaptive ensemble empirical modal decomposition of noise (CEEMDAN), convolutional neural network (CNN), and attention bidirectional long short-term memory network (ABiLSTM). Firstly, a finite number of intrinsic mode functions (IMFs) are obtained from breaking down the initial vibration signals using CEEMDAN. The IMFs are further screened by combining the correlation criterion and the craggy criterion. Then, time-frequency domain features, which are extracted from the screened IMFs, are reconstructed into a feature set. The SPT is recognized through some features, like the root mean square (RMS), variance, and kurtosis. Secondly, the deterioration character of rolling bearings was extracted using CNN and used to train the ABiLSTM network. Based on the output of the ABiLSTM network, it forecasts how long rolling bearings will last during use. Finally, the XJTU-SY rolling bearing dataset validated the validity of the suggested rolling bearing remaining life prediction method. We compare our algorithm with other algorithms, such as GRU, LSTM, and CNN–BiLSTM, in which the accuracy of MAE, MSE, RMSE, MAPE, and R2_score is significantly improved. Thus, the results of the validation experiments demonstrate that our proposed algorithm has excellent prediction accuracy.