[go: up one dir, main page]

Skip to main content
Log in

Convolutional neural network based on attention mechanism and Bi-LSTM for bearing remaining life prediction

  • Published:
Applied Intelligence Aims and scope Submit manuscript

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.

This is a preview of subscription content, log in via an institution to check access.

Access this article

Subscribe and save

Springer+ Basic
$34.99 /Month
  • Get 10 units per month
  • Download Article/Chapter or eBook
  • 1 Unit = 1 Article or 1 Chapter
  • Cancel anytime
Subscribe now

Buy Now

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Fig. 1
Fig. 2
Fig. 3
Fig. 4
Fig. 5
Fig. 6
Fig. 7
Fig. 8
Fig. 9
Fig. 10
Fig. 11
Fig. 12
Fig. 13
Fig. 14
Fig. 15
Fig. 16
Fig. 17
Fig. 18

Similar content being viewed by others

Explore related subjects

Discover the latest articles, news and stories from top researchers in related subjects.

References

  1. 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

  2. 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

  3. 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

    Article  Google Scholar 

  4. 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

  5. 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

    Article  Google Scholar 

  6. 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

  7. 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

    Google Scholar 

  8. 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

    Article  Google Scholar 

  9. 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)

  10. 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

    Article  Google Scholar 

  11. 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

  12. 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

  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

    Article  Google Scholar 

  14. 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

  15. 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

    Google Scholar 

  16. 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

    Google Scholar 

  17. 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

    Article  Google Scholar 

  18. 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

  19. 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)

  20. 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

    Article  Google Scholar 

  21. 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

    Article  Google Scholar 

  22. 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

    Article  Google Scholar 

  23. 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

  24. 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

  25. 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

    Article  Google Scholar 

  26. 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)

  27. Tian Q, Wang H (2020) An ensemble learning and RUL prediction method based on bearings degradation indicator construction[J]. Appl Sci 10(1):346

    Article  Google Scholar 

  28. 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

  29. Ma M, Mao Z (2020) Deep convolution-based LSTM network for remaining useful life prediction[J]. IEEE Trans Ind Inf 99:1–1

    Article  Google Scholar 

  30. 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

  31. 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

    Article  Google Scholar 

  32. 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

    Article  Google Scholar 

  33. She D, Jia M (2020) A BiGRU method for remaining useful life prediction of machinery[J]. Measurement 167:108277

    Article  Google Scholar 

  34. 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

    Article  Google Scholar 

  35. 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

    Article  Google Scholar 

  36. 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

  37. 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

    Article  Google Scholar 

  38. Ma M, Mao Z (2020) Deep convolution-based LSTM network for remaining useful life prediction[J]. IEEE Trans Ind Inf PP(99):1–1

  39. 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

  40. 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

  41. 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

    Article  Google Scholar 

  42. 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

    Article  Google Scholar 

Download references

Acknowledgements

The authors would like to thank the FEMTO-ST Institute for their public data.

Author information

Authors and Affiliations

Authors

Corresponding author

Correspondence to Xu Zhang.

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

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

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

Download citation

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s10489-021-02503-2

Keywords

Navigation