Skip to main navigation Skip to search Skip to main content

A Lightweight and Adaptive Knowledge Distillation Framework for Remaining Useful Life Prediction

  • Lei Ren*
  • , Tao Wang
  • , Zidi Jia
  • , Fangyu Li
  • , Honggui Han
  • *Corresponding author for this work
  • Zhongguancun Laboratory
  • Beihang University
  • Beijing University of Technology

Research output: Contribution to journalArticlepeer-review

Abstract

For prognostics and health management of industrial systems, machine remaining useful life (RUL) prediction is an essential task. While deep learning-based methods have achieved great successes in RUL prediction tasks, large-scale neural networks are still difficult to deploy on edge devices owing to the constraints of memory capacity and computing power. In this article, we propose a lightweight and adaptive knowledge distillation (KD) framework to alleviate this problem. First, multiple teacher models are compressed into a student model through KD to improve the industrial prediction accuracy. Second, a dynamic exiting method is studied to enable an adaptive inference on the distilled student model. Finally, we develop a reparameterization scheme to further lessen the student network. Experiments on two turbofan engine degradation datasets and a bearing degradation dataset demonstrate that our method significantly outperforms the state-of-the-art KD methods and enables the distilled model with an adaptive inference ability.

Original languageEnglish
Pages (from-to)9060-9070
Number of pages11
JournalIEEE Transactions on Industrial Informatics
Volume19
Issue number8
DOIs
StatePublished - 1 Aug 2023

Keywords

  • Adaptive inference
  • knowledge distillation (KD)
  • remaining useful life (RUL) prediction
  • reparameterization

Fingerprint

Dive into the research topics of 'A Lightweight and Adaptive Knowledge Distillation Framework for Remaining Useful Life Prediction'. Together they form a unique fingerprint.

Cite this