Remaining useful life prediction of rolling bearings based on time convolutional network and transformer in parallel
Youfu Tang, Ruifeng Liu, Chunhui Li, Na Lei
Abstract
Abstract Remaining useful life prediction (RUL) is crucial for maintaining the reliability and safety of industrial equipment. Recently, the Transformer has been widely used due to its ability to effectively extract global feature information in rolling bearing RUL prediction. However, the Transformer is weak in acquiring local feature information and cannot extract temporal features from the degradation process. Conversely, a temporal convolutional network (TCN) can effectively extract local features but is weak in global feature extraction. Therefore, to address the above problems, this paper proposes a prediction method based on the parallel combination of TCN and Transformer. The method first extracts the time domain, frequency domain, and time-frequency domain features from the original vibration signals. After screening through the evaluation indices, the features are fused using K-means clustering and principal component analysis (PCA) to construct health indicators (HI) that characterize the degradation of rolling bearings. Then, a TCN-Transformer parallel network model is constructed to extract both local and global features, and a feed-forward neural network (FNN) is used for the prediction of RUL. Finally, experimental validation is carried out on the PHM2012 bearing dataset, and the results show that the method achieves higher RUL prediction accuracy compared to other methods.