A Prototype Learning Framework Based on Continual Learning for Motor Incremental Fault Diagnosis Under Few-Shot Conditions

Abstract

In the actual industrial scenario, motor fault classes gradually increase, and the fault data need to be acquired incrementally during the service. Continual learning (CL) has been introduced to the field of fault diagnosis (FD), aiming at constructing an FD model with continuous evolution capability. However, the current research has the following limitations: 1) assigning parameters for each task independently leads to a linear increase in model complexity and 2) it relies heavily on the quality and completeness of historical data and the research on incremental FD under few-shot conditions is not sufficient. To address the above limitations, this article proposes a prototype learning framework for motor incremental FD (PLFIFD) under few-shot conditions. First, a prototype classifier based on distance metric is proposed to avoid uncontrollable changes due to the update of parameters in the output layer. The prototype set is dynamically expanded instead of assigning network parameters individually. Second, a multiobjective loss function is designed to jointly optimize the classification boundary and prototype space distribution to enhance intraclass compactness and interclass separability and to improve the generalization ability under few-shot conditions and stability in dynamic scenarios. Finally, the effectiveness of PLFIFD is verified on induction motor (IM) and permanent magnet synchronous motor (PMSM).