Remaining Useful Life Prediction of Bearings Based on Convolution Attention Mechanism and Temporal Convolution Network

Abstract

The prediction of the remaining useful life (RUL) of bearings is of great significance for reducing cost and increasing efficiency of mechanical equipment and ensuring healthy operation. Most of the traditional RUL prediction methods based on deep learning only analyze single sensor data, which makes the prediction accuracy fluctuate greatly and the reliability is low, and the contributions of data collected by different sensors to RUL prediction are inconsistent, which makes the low utilization of data lead to poor prediction results. To solve the above problems, a new RUL prediction framework with convolutional attention mechanism and temporal convolutional network (CAMTCN) is proposed. Firstly, the vibration signals collected by multi-sensor are normalized and processed directly as the input of the network. Secondly, an Efficient Adaptive Shrinkage (EAS) model is designed to adaptively eliminate noise interference and improve the accuracy of RUL prediction; Finally, we use the attention mechanism to design the convolutional attention subnet and combine it with the TCN to obtain the contribution degree of bearing degradation features collected by different sensors. Adaptive weight allocation captures more important degradation features to achieve end-to-end RUL prediction. Experiments were conducted on IEEE PHM 2012 dataset and XJTU-SY dataset respectively, and compared with some advanced in-depth learning methods. The experimental results show that this method can predict RUL more accurately on different bearing data sets, and has good generalization.