Few-Shot Class-Incremental Learning with Adjustable Pseudo-Incremental Sessions for Bearing Fault Diagnosis

Hongyan Zhu,Changqing Shen,Jiaan Wang,Bojian Chen,Dong Wang,Zhongkui Zhu
DOI: https://doi.org/10.1109/jsen.2024.3395515
IF: 4.3
2024-01-01
IEEE Sensors Journal
Abstract:Rotating machinery may constantly generate new classes of faults in complex operating environments, with a finite set of fault samples that are obtainable. The incremental nature of fault types and the scarcity of samples present great challenges for fault diagnosis models employing deep learning techniques, such as overfitting due to insufficient data and stability–plasticity problem when new fault samples are introduced. A few-shot class-incremental learning (FSCIL) method with adjustable pseudo-incremental sessions (APIS) is introduced in this study to overcome the abovementioned concerns. First, the method extracts fault samples from the sufficient data of the real base session, and it sets the pseudo FSCIL task according to the data format of FSCIL. The pseudo FSCIL task is employed to extract invariant features in the base and incremental session, and the feature space is trained to provide a reference model for the real fault incremental task. Second, the adjuster based on a self-attention mechanism is used to learn more discriminative features. Feature differences are observed between the old fault classifier and a fresh fault prototype. Based on the self-attention mechanism, the distinguishing features between the fault samples in the testing dataset and the old classifier are highlighted. In this way, a fault diagnosis model of the few-shot fault class increment with more generalization ability is acquired. A case study shows that APIS can effectively alleviate stability–plasticity and overfitting problems when dealing with FSCIL bearing fault diagnosis tasks.
What problem does this paper attempt to address?