Skip to main navigation Skip to search Skip to main content

DLformer: A Dynamic Length Transformer-Based Network for Efficient Feature Representation in Remaining Useful Life Prediction

  • Lei Ren*
  • , Haiteng Wang*
  • , Gao Huang
  • *Corresponding author for this work
  • Zhongguancun Laboratory
  • Beihang University
  • Tsinghua University

Research output: Contribution to journalArticlepeer-review

Abstract

Representation learning-based remaining useful life (RUL) prediction plays a crucial role in improving the security and reducing the maintenance cost of complex systems. Despite the superior performance, the high computational cost of deep networks hinders deploying the models on low-compute platforms. A significant reason for the high cost is the computation of representing long sequences. In contrast to most RUL prediction methods that learn features of the same sequence length, we consider that each time series has its characteristics and the sequence length should be adjusted adaptively. Our motivation is that an "easy" sample with representative characteristics can be correctly predicted even when short feature representation is provided, while "hard" samples need complete feature representation. Therefore, we focus on sequence length and propose a dynamic length transformer (DLformer) that can adaptively learn sequence representation of different lengths. Then, a feature reuse mechanism is developed to utilize previously learned features to reduce redundant computation. Finally, in order to achieve dynamic feature representation, a particular confidence strategy is designed to calculate the confidence level for the prediction results. Regarding interpretability, the dynamic architecture can help human understand which part of the model is activated. Experiments on multiple datasets show that DLformer can increase up to 90% inference speed, with less than 5% degradation in model accuracy.

Original languageEnglish
JournalIEEE Transactions on Neural Networks and Learning Systems
Issue number99
DOIs
StatePublished - 2023

Keywords

  • Adaptive inference
  • deep learning
  • feature representation
  • interpretability
  • remaining useful life (RUL) prediction.

Fingerprint

Dive into the research topics of 'DLformer: A Dynamic Length Transformer-Based Network for Efficient Feature Representation in Remaining Useful Life Prediction'. Together they form a unique fingerprint.

Cite this