Remaining Useful Life Prediction for Bearing Based on Coupled Diffusion Process and Temporal Attention
Yixiang Lu, Daiqi Tang, De Zhu, Qingwei Gao, Dawei Zhao, Junwen Lyu
Abstract
Remaining Useful Life (RUL) prediction is one of the difficulties in prognostics and system health management (PHM) of bearings. Moreover, due to the limitations of test cost and equipment, it isn’t easy to obtain abundant full life cycle experimental samples of bearings, which cannot provide enough training samples to further improve the accuracy of RUL prediction. To address these problem, this paper proposes a rolling bearing RUL prediction method based on coupled diffusion probabilistic model and time attention mechanism (CDTA). This method first augments the original sequence with the coupled diffusion process, which preserves its randomness and reduces the difficulty of training the subsequent inference network. Then, it introduces a time attention unit (TAU) to enhance the network’s ability to extract and fuse temporal features and dependencies within and between samples, which improves prediction accuracy. This paper conducts experimental verification on PHM2012 datasets and shows that the proposed method achieves significant improvement in prediction accuracy compared with existing RUL prediction methods.