Abstract
Good prognostic health management (PHM) plays a crucial role in industrial production and other fields. The accurate prediction of remaining useful life (RUL) can ensure good working condition of machines, and the selection of health indicators (HI) is the key to prediction. Due to the length and noise of the data, the selection of features requires a lot of prior knowledge. Therefore, a novel convolution-based attention mechanism bidirectional long and short-term memory (CABLSTM) network is proposed to achieve the end-to-end lifetime prediction of rotating machinery in this paper. Unlike concatenating two networks, the model in this paper is used to convolute the cell states of Bi-LSTM. Firstly, the input signal is performed through CNN to obtain feature information. Secondly, the obtained features are fed into the Bi-LSTM network with attention mechanism for convolution operation to obtain time-frequency information to construct HI. Finally, the training data are normalized to predict RUL. Bi-LSTM can capture features in longer time-frequency information, and attention mechanism can give input influence weight, and highlight its effective characteristics to obtain better prediction accuracy. The complex process of feature extraction, HI construction, and RUL prediction is combined in one algorithm by deep learning. To verify the performance of the method in this paper, experiments were conducted on the bearing dataset of PRONOSTIA, and compared with other methods. The results showed that the method outperforms other methods due to its better accuracy and prediction precision.


















Similar content being viewed by others
Explore related subjects
Discover the latest articles, news and stories from top researchers in related subjects.References
Wu B, Li W, Qiu MQ (2017) Remaining useful life prediction of bearing with vibration signals based on a novel indicator[J]. Shock Vib 2017(PT.6):1–10
Ren L, Sun Y, Wang H et al (2018) Prediction of bearing remaining useful life with deep convolution neural network[J]. IEEE Access 2018:13041–13049
Zhu J, Chen N, Peng W (2019) Estimation of bearing remaining useful life based on multiscale convolutional neural network[J]. IEEE Trans Ind Electron 66(4):3208–3216
Liu R, Yang B, Hauptmann AG (2019) Simultaneous bearing fault recognition and remaining useful life prediction using joint loss convolutional neural network[J]. IEEE Trans Ind Inf PP(99):1–1
Qiu G, Gu Y, Chen J (2019) Selective health indicator for bearings ensemble remaining useful life prediction with genetic algorithm and Weibull proportional hazards model[J]. Measurement 150:107097
Motahari-Nezhad Mohsen and Jafari Seyed Mohammad (2021) Bearing remaining useful life prediction under starved lubricating condition using time domain acoustic emission signal processing[J]. Expert Systems With Applications 168
Pang Y, Jia L, Liu Z (2020) Discrete cosine transformation and temporal adjacent convolutional neural network-based remaining useful life estimation of bearings[J]. Shock Vib 2020:1–14
Ren L, Sun Y, Wang H et al (2018) Prediction of bearing remaining useful life with deep convolution neural network[J]. IEEE Access 2018:13041–13049
Zou P, Hou B, Lei J, et al (2020) Bearing Fault Diagnosis Method Based on EEMD and LSTM[J]. Int J Comput Commun Control (IJCCC) 15(1)
Jouin M, Gouriveau R, Hissel D et al (2016) Particle filter-based prognostics: Review, discussion and perspectives[J]. Mech Syst Signal Process 72–73(May):2–31
Zhou F, Yang S, Fujita H et al (2020) Deep learning fault diagnosis method based on global optimization GAN for unbalanced data[J]. Knowl-Based Syst 187(Jan.):104837.1-104837.19
Hu C, He S, Wang Y (2020) A classification method to detect faults in a rotating machinery based on kernelled support tensor machine and multilinear principal component analysis[J]. Appl Intell 1-13
Yang B, Liu R, Zio E (2019) Remaining useful life prediction based on a double-convolutional neural network architecture[J]. IEEE Trans Ind Electron 66(12):9521–9530
Yu X M, Feng W Z, Wang H, et al (2020) An attention mechanism and multi-granularity-based Bi-LSTM model for Chinese Q&A system[J]. Soft Comput 24(8):5831-5845
Zhang W, Li C, Peng G et al (2017) A deep convolutional neural network with new training methods for bearing fault diagnosis under noisy environment and different working load[J]. Mech Syst Signal Process 100(FEB.1):439–453
Jun Z, Nan C, Weiwen P (2018) Estimation of bearing remaining useful life based on multiscale convolutional neural network[J]. IEEE Trans Ind Electron PP:1–1
Wang S, Wang X, Wang S et al (2019) Bi-directional long short-term memory method based on attention mechanism and rolling update for short-term load forecasting[J]. Int J Electr Power Energy Syst 109(JUL.):470–479
Yu Y, Hu C, Si X, et al (2020) Averaged Bi-LSTM networks for RUL prognostics with non-life-cycle labeled dataset[J]. Neurocomputing 402:134-147
Mao W, He J, Tang J, et al (2018) Predicting remaining useful life of rolling bearings based on deep feature representation and long short-term memory neural network[J]. Adv Mech Eng 10(12)
Xia T, Song Y, Zheng Y et al (2020) An ensemble framework based on convolutional bi-directional LSTM with multiple time windows for remaining useful life estimation[J]. Comput Ind 115:103182
Liu C, Zhu L (2020) A two-stage approach for predicting the remaining useful life of tools using bidirectional long short-term memory[J]. Measurement 164:108029
Pei H, Si XS, Hu CH et al (2021) An adaptive prognostics method for fusing CDBN and diffusion process: Application to bearing data[J]. Neurocomputing 421(1):303–315
Hu C, Wang Y, Gu J (2020) Cross-domain intelligent fault classification of bearings based on tensor-aligned invariant subspace learning and two-dimensional convolutional neural networks - ScienceDirect[J]. Knowledge-Based Syst 209
Wu Y, Zhao R, Jin W, et al (2020) Intelligent fault diagnosis of rolling bearings using a semi-supervised convolutional neural network[J]. Appl Intell 1-17
Huang G, Li H, Ou J et al (2020) A reliable prognosis approach for degradation evaluation of rolling bearing using MCLSTM[J]. Sensors 20(7):1864
Jiang JR, Lee JE, Zeng YM (2020) Time Series Multiple Channel Convolutional Neural Network with Attention-Based Long Short-Term Memory for Predicting Bearing Remaining Useful Life[J]. Sensors (Basel, Switzerland) 20(1)
Tian Q, Wang H (2020) An ensemble learning and RUL prediction method based on bearings degradation indicator construction[J]. Appl Sci 10(1):346
Wu Q, Zhang C (2020) Cascade fusion convolutional Long-short time memory network for remaining useful life prediction of rolling bearing[J]. IEEE Access PP (99):1–1
Ma M, Mao Z (2020) Deep convolution-based LSTM network for remaining useful life prediction[J]. IEEE Trans Ind Inf 99:1–1
Kong X, Yang J (2019) Remaining useful life prediction of rolling bearings based on RMS-MAVE and dynamic exponential regression model[J]. IEEE Access PP(99):1–1
Lei Y, Li N, Gontarz S et al (2016) A model-based method for remaining useful life prediction of machinery[J]. IEEE Trans Reliab 65(3):1314–1326
Qin Y, Xiang S, Chai Y et al (2020) Macroscopic–microscopic attention in LSTM networks based on fusion features for gear remaining life prediction[J]. IEEE Trans Ind Electron 67(12):10865–10875
She D, Jia M (2020) A BiGRU method for remaining useful life prediction of machinery[J]. Measurement 167:108277
Youngji Y, Jun-Geol B (2018) A novel image feature for the remaining useful lifetime prediction of bearings based on continuous wavelet transform and convolutional neural network[J]. Appl Sci 8(7):1102
Zhu ZA, Peng GA, Chen YA et al (2019) A convolutional neural network based on a capsule network with strong generalization for bearing fault diagnosis[J]. Neurocomputing 323:62–75
Yuan M, Wu Y, Lin L (2016) Fault diagnosis and remaining useful life estimation of aero engine using LSTM neural network[C]. IEEE International Conference on Aircraft Utility Systems. IEEE
Guo L, Li N, Jia F et al (2017) A recurrent neural network based health indicator for remaining useful life prediction of bearings[J]. Neurocomputing 240(May31):98–109
Ma M, Mao Z (2020) Deep convolution-based LSTM network for remaining useful life prediction[J]. IEEE Trans Ind Inf PP(99):1–1
Chen Y, Peng G, Zhu Z, et al (2019) A novel deep learning method based on attention mechanism for bearing remaining useful life prediction[J]. Appl Soft Comput 86:105919
Sutrisno E, Oh H, Vasan ASS et al (2012) Estimation of remaining useful life of ball bearings using data driven methodologies[C]. IEEE Conference on Prognostics & Health Management. IEEE
Hinchi AZ, Tkiouat M (2018) Rolling element bearing remaining useful life estimation based on a convolutional long-short-term memory network[J]. Procedia Comput Sci 127:123–132
Hong S, Zhou Z, Zio E et al (2014) Condition assessment for the performance degradation of bearing based on a combinatorial feature extraction method[J]. Digit Signal Proc 27:159–166
Acknowledgements
The authors would like to thank the FEMTO-ST Institute for their public data.
Author information
Authors and Affiliations
Corresponding author
Ethics declarations
Competing interests
The authors declare that they have no conflict of financial interests or personal relationships in this article.
Additional information
Publisher’s Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
About this article
Cite this article
Luo, J., Zhang, X. Convolutional neural network based on attention mechanism and Bi-LSTM for bearing remaining life prediction. Appl Intell 52, 1076–1091 (2022). https://doi.org/10.1007/s10489-021-02503-2
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s10489-021-02503-2