A Feature Extension and Reconstruction Method with Incremental Learning Capabilities under Limited Samples for Intelligent Diagnosis

Kui Hu,Zhihao Bi,Qingbo He,Zhike Peng
DOI: https://doi.org/10.1016/j.aei.2024.102796
IF: 8.8
2024-01-01
Advanced Engineering Informatics
Abstract:Existing data-driven Intelligent Fault Diagnosis (IFD) methods often struggle with continuous learning and integrating new diagnostic knowledge. These methods necessitate a complete retraining process when new fault data emerges. Furthermore, collecting sufficient data for model training is impractical in real-world applications. This study proposes a Feature Extension and Reconstruction (FER) method for model incremental updating with limited samples. FER employs scalable and modular feature extractors for continuous representation learning of new fault modes. In addition, a domain mapping algorithm based on optimal transport is devised to leverage old classification nodes in aiding new nodes’ training, enhancing models’ generalization performance under limited samples. A feature reconstruction technique is also developed to compress the expanded multi-backbone model, eliminating redundant parameters and dimensions for adaptability in subsequent incremental tasks. Validated on simulated transmission systems and real planetary gearboxes, FER demonstrates a better stability-plasticity trade-off in incremental tasks with limited training samples, offering a novel solution for adaptive IFD model updates in sample-constrained scenarios.
What problem does this paper attempt to address?