Bayesian Adversarial Adaptation Network With Feature Disentanglement for Remaining Useful Life Prediction

Remaining useful life (RUL) prediction is vital for the safety of engineering assets. In the real scenario, due to the lack of failure data and variable working conditions, the accuracy of predictive RUL is significantly compromised as models struggle to generalize across diverse operating environments. Existing solutions manage to shift the degradation information from the ideal laboratory environment to the complex real-world environment. However, they fail to consider the heterogeneity of operating machines under different working conditions. This ignorance of inherent properties will eventually hamper the accuracy of RUL prediction. Consequently, a novel Bayesian adversarial Fast Linear Attention with a Single Head (FLASH) Transformer with feature disentanglement model (BAFTFD) was proposed in this article to tackle with the problem. The proposed BAFTFD model can disentangle the private feature representations from the raw data, preserving the shared feature representation for the prediction. The adversarial training method is also exploited to facilitate the transfer of degradation knowledge. Besides, the feature extractor is equipped with the effective FLASH Transformer model to retain the most informative degradation features for model training, improving the efficiency of feature extraction. Moreover, considering the impact of insufficient training data, inherent data noise on the trustworthiness of the predictive results, the Bayesian DL method is adopted to quantify the prediction uncertainties, ensuring the reliability of maintenance decisions. Two commercial turbofan datasets are leveraged to validate the designed model.