Reserving embedding space for new fault types: A new continual learning method for bearing fault diagnosis

Hongyan Zhu, Changqing Shen, Lin Li, Dong Wang, Weiguo Huang and Zhongkui Zhu

Reliability Engineering and System Safety, 2024, vol. 252, issue C

Abstract: In complex operating environments, rotating equipment may continually generate new fault categories, affecting the safety of equipment operation, and the number of collected fault samples is limited, which make it difficult to establish a reliable diagnostic model. Few-shot continual learning (FSCL) can continuously learn and summarize fault knowledge from limited samples. These traditional FSCL models, when learning from limited samples of new types of bearing faults, exhibit model overfitting and catastrophic forgetting issues after self-updating (new model). This phenomenon significantly limits the reliability of traditional models. This paper presents a new approach coined as reserving embedding space for new fault types (RESNFT) for continual learning of newly occurred bearing faults for machinery fault diagnosis, which includes several key and new ideas. Preliminary, a class prototype (prototype vector) is derived to replace a classifier weight. Then, embedding space reserved by forcibly assigning real fault samples to both real and virtual categories can mitigate catastrophic forgetting. Finally, virtual faults formed by sample mixing guide new fault samples to the reserved embedding space, and realize continuous classification of new faults. Results show that the RESNFT can effectively alleviate the catastrophic forgetting and overfitting problems, and it is superior to other existing methods.

Keywords: Bearing; Fault diagnosis; Catastrophic forgetting; Continual learning; Prototypical networks; Few-shot learning (search for similar items in EconPapers)
Date: 2024
References: View references in EconPapers View complete reference list from CitEc
Citations:

Downloads: (external link)
http://www.sciencedirect.com/science/article/pii/S0951832024005052
Full text for ScienceDirect subscribers only

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:eee:reensy:v:252:y:2024:i:c:s0951832024005052

DOI: 10.1016/j.ress.2024.110433

Access Statistics for this article

Reliability Engineering and System Safety is currently edited by Carlos Guedes Soares

More articles in Reliability Engineering and System Safety from Elsevier
Bibliographic data for series maintained by Catherine Liu ().