A Lifelong Learning Method for Gearbox Diagnosis With Incremental Fault Types

Abstract

Deep learning (DL)-based fault diagnosis models have to collect the most comprehensive data of mechanical fault types to ensure reliability. In real scenarios, due to complex, variable operating conditions, machines often generate unexpected faults that lead to an increment of fault types, causing the diagnosis model to be invalid. Therefore, the data of new fault types are needed to retrain the model. However, DL models suffer from catastrophic forgetting when incrementally learning new classes. To solve the problem of the diagnosis of increasing fault types, a lifelong learning method for fault diagnosis (LLMFD) is proposed in this paper under the lifelong learning paradigm. The key of LLMFD is a proposed dual-branch aggregation networks (DBANets) framework that is combined with reserved exemplars to learn the new fault types without forgetting the old ones. In DBANets, each residual block layer has a dynamic block and a steady block to solve the stability–plasticity dilemma in lifelong learning. The aggregation weights are adopted to balance stability and plasticity. LLMFD is applied to a diagnosis case of incremental fault types. Results verify that LLMFD is superior to other lifelong learning methods and has satisfactory robustness.