DOI: 10.3390/make8080249 ISSN: 2504-4990

Knowledge Transfer-Based Heterogeneous Distillation Network for Remaining Useful Life Prediction Under Cross-Working Conditions

Jiehua Qi, Haoran Wang, Rui Wang, Xinxiao Wu, Hanhong Hu, Bingcong Chen

Remaining useful life (RUL) estimation is a fundamental task in Prognostics and Health Management (PHM), supporting condition-based and predictive maintenance of engineering systems. Data-driven methods contribute to many effective strategies for RUL prediction. However, two problems need to be solved when they are used in industrial applications: (1) The amount of data under one working condition is limited, and data from different working conditions suffer from domain discrepancies. These methods are constrained by distribution differences in data under different working conditions. (2) There is an urgent need to quickly achieve prediction with much less computing resources. To address these issues, a lightweight RUL prediction method called a knowledge transfer-based heterogeneous distillation network is proposed by combining knowledge distillation and transfer learning. First, the adversarial training mechanism is introduced for the extraction of domain-invariant features. Subsequently, a heterogeneous knowledge distillation framework is further designed for remaining useful life prediction, in which a bi-directional long short-term memory model serves as the teacher network and a compact fully connected network acts as the student model. The teacher model is used to learn informative degradation patterns and guide the training of the lightweight student model through knowledge transfer. Results obtained on the N-CMAPSS dataset verify that the proposed method achieves promising effectiveness and strong generalizability, reducing the average RMSE and MAE by 44.83% and 41.30%, respectively.

More from our Archive