Few-Shot Rotating Machinery RUL Prediction Based on Reptile Framework and Transformer
The prediction of the remaining useful life (RUL) of rotating machinery is important in Prognostics and Health Management (PHM) of industrial equipment. Traditional RUL prediction methods rely on large amounts of labeled data, making it challenging to meet the prediction needs in data-scarce situations. This paper proposes a Transformer-based meta-learning algorithm for few-shot RUL prediction of rotating machinery. The method constructs a meta-learner based on Transformer and conducts meta-training under the Reptile framework, enabling the model to learn general knowledge across multiple tasks and rapidly adapt to new tasks with limited samples. Experiments on the CMAPSS dataset demonstrate that this method performs well under few-shot conditions.
Authors
- Guangwei Liu
- Shuai Lv (ORCID: https://orcid.org/0000-0002-2676-9622)
- Yu Wang (ORCID: https://orcid.org/0000-0002-6145-292X)
- Shujie Liu (ORCID: https://orcid.org/0009-0008-2599-6752)
Institutions
- Dalian University of Technology (CN)
Publication Details
- Published
- 2024-10-11
- DOI
- https://doi.org/10.1109/phm-beijing63284.2024.10874723
- Primary Topic
- Advanced machining processes and optimization
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- National Natural Science Foundation of China