Utilizing Autoregressive Networks for Full Lifecycle Data Generation of Rolling Bearings for RUL Prediction

Junliang Wang,Qinghua Zhang,Guanhua Zhu,Guoxi Sun
2024-01-02
Abstract:The prediction of rolling bearing lifespan is of significant importance in industrial production. However, the scarcity of high-quality, full lifecycle data has been a major constraint in achieving precise predictions. To address this challenge, this paper introduces the CVGAN model, a novel framework capable of generating one-dimensional vibration signals in both horizontal and vertical directions, conditioned on historical vibration data and remaining useful life. In addition, we propose an autoregressive generation method that can iteratively utilize previously generated vibration information to guide the generation of current signals. The effectiveness of the CVGAN model is validated through experiments conducted on the PHM 2012 dataset. Our findings demonstrate that the CVGAN model, in terms of both MMD and FID metrics, outperforms many advanced methods in both autoregressive and non-autoregressive generation modes. Notably, training using the full lifecycle data generated by the CVGAN model significantly improves the performance of the predictive model. This result highlights the effectiveness of the data generated by CVGans in enhancing the predictive power of these models.
Machine Learning,Artificial Intelligence
What problem does this paper attempt to address?
### Problems Addressed by the Paper The paper aims to address the issue of data scarcity in the prediction of rolling bearing lifespan. Specifically, high-quality, full-lifecycle data is a major limiting factor in achieving accurate predictions. To tackle this challenge, the authors propose the CVGAN model, a novel framework capable of generating 1-dimensional vibration signals based on historical vibration data and Remaining Useful Life (RUL). Additionally, the paper introduces an autoregressive generation method that iteratively uses previously generated vibration information to guide the generation of current signals. The main contributions of the paper include: 1. **CVGAN Model**: A new framework based on Conditional Variational Autoencoder (CVAE) and Generative Adversarial Network (GAN) that can generate 1-dimensional vibration signals in both horizontal and vertical directions. 2. **Autoregressive Generation Method**: Applied for the first time to the generation of full-lifecycle vibration data for rolling bearings. 3. **Evaluation Metrics**: Using Maximum Mean Discrepancy (MMD) and Fréchet Inception Distance (FID) as metrics to evaluate the task of generating 1-dimensional vibration signals. 4. **Experimental Results**: Experiments on the PHM 2012 dataset show that the CVGAN model outperforms many advanced methods in both autoregressive and non-autoregressive generation methods, and the use of generated full-lifecycle data significantly improves the performance of prediction models. Through these contributions, the paper demonstrates the effectiveness of generated data in enhancing the performance of prediction models.