A New Meta-transfer Learning Method with Freezing Operation for Few-shot Bearing Fault Diagnosis

Peiqi Wang,Jingde Li,Shubei Wang,Fusheng Zhang,Juanjuan Shi,Changqing Shen
DOI: https://doi.org/10.1088/1361-6501/acc67b
IF: 2.398
2023-03-24
Measurement Science and Technology
Abstract:Deep learning (DL) for bearing fault diagnosis often requires a large quantity of comprehensive data to support in the field of rotating machinery fault diagnosis. However, large-quantity dataset for model training is difficult to obtain in the actual working environment. Therefore, bearing fault diagnosis problems under practical working conditions are often few-shot problems. Meta-learning can be adopted to solve few-shot problems. Traditional meta-learning method will lead to model overfitting, and shallow neural networks are usually used to avoid overfitting. As a result, features extracted by the shallow neural network are insufficiently rich to exploit the optimal performance of the model. A few-shot fault diagnosis method based on meta-learning named meta-transfer learning method with freezing operation (MTLFO) is proposed in this study to solve these problems. MTLFO can learn new knowledge rapidly through a small number of samples. Hyperparameter self-regulation ability of meta-learning is adopted by MTLFO, and freezing operation (FO) is used to deal with the neuronal of meta-learning to ensure that neurons from different tasks are transferred in the manner of scaling and shifting. MTLFO avoids the overfitting problem in traditional meta-learning methods and presents more advantages in solving few-shot problems in fault diagnosis compared with other groups of methods.
engineering, multidisciplinary,instruments & instrumentation
What problem does this paper attempt to address?