Remaining Useful Life Prediction of Bearings Based on Self-attention Mechanism, Multi-scale Dilated Causal Convolution, and Temporal Convolution Network

Hao Wei,Qinghua Zhang,Yu Gu
DOI: https://doi.org/10.1088/1361-6501/acb0e9
IF: 2.398
2023-01-08
Measurement Science and Technology
Abstract:Effective remaining useful life (RUL) prediction of bearings is essential for the predictive maintenance of rotating machinery. However, the effectiveness of many existing RUL prediction methods depends on expert experience and signal processing algorithms, which limiting the application of these methods in real-life scenarios. This study proposes a novel end-to-end deep learning framework consisting of a multi-scale attention-based dilated causal convolutional (MADCC) module and a multi-layer temporal convolutional network (MTCN) to predict the RUL of bearings using raw vibration data. First, the MADCC module extracts multi-scale temporal features of the bearing degradation process (BDP) and provides fused feature vectors containing comprehensive BDP information for the MTCN module. Subsequently, the MTCN module mines deep temporal dependencies hidden in the fused feature vector to predict the RUL of bearings. Ablation experiments are conducted to analyze the contribution of the framework's components. Three evaluation metrics (root mean square error (RMSE), mean absolute error (MAE), and coefficient of determination (R2)) are used to verify the effectiveness of the proposed framework and other state-of-the-art methods on two public bearing datasets. The experimental results show that the proposed framework achieves the lowest RMSE and MAE and the highest R2, demonstrating excellent performance and potential for RUL prediction of bearings.
engineering, multidisciplinary,instruments & instrumentation
What problem does this paper attempt to address?