Few-shot Sample Multi-Class Incremental Fault Diagnosis for Gearbox Based on Convolutional-Attention Fusion Network

Zhen Guo,Wenliao Du,Zhiping Liu,Tao Hu,Yannan Yu,Chuan Li
DOI: https://doi.org/10.1016/j.eswa.2024.125918
IF: 8.5
2024-01-01
Expert Systems with Applications
Abstract:Incremental fault diagnosis refers to overcoming catastrophic forgetting by constantly learning new knowledge from the collected data stream, thus ensuring that the model can adapt to the changing environment. However, in practical applications, especially in the case of few-shot samples and increasing fault categories, gearbox fault diagnosis faces significant challenges in stability learning ability. To address these challenges, we proposed a convolutional-attention fusion network (CAFNet) based approach to optimize gearbox fault diagnosis capabilities. First, we built a knowledge base to store the information continuously from the data stream. This knowledge base is used to store historical data and includes data preprocessing to ensure data quality and consistency. Secondly, an adaptive weight updating algorithm is designed to dynamically adjust the weights according to the actual performance of the model to improve the adaptive ability of the model. At the same time, the L2 regularization parameter is introduced into the loss function to effectively prevent overfitting. The validity and reliability of this method are verified by two datasets, which provide a new solution for mechanical fault diagnosis.
What problem does this paper attempt to address?