Semisupervised Momentum Prototype Network for Gearbox Fault Diagnosis Under Limited Labeled Samples
Xiaolong Zhang, Zuqiang Su, Xiaolin Hu, Yan Han, Shuxian Wang
Abstract
It is difficult to obtain expensive labeled data in industrial fault diagnosis applications, which easily leads to overfitting of deep learning and restricts its extensive usage. Aiming at this issue, this article proposed an improved few-shot semisupervised learning method, called semisupervised momentum prototype network (SSMPN), to realize gearbox fault diagnosis under limited labeled samples. First, the proposed SSMPN utilizes the powerful few-shot learning ability of the prototype network to learn the feature mapping and obtains prototypes by using limited labeled samples. Then, a threshold selection based on Monte Carlo uncertainty is adopted in pseudo label learning to increase the confidence of pseudo labels. Finally, the expended labeled dataset is utilized to optimize feature extraction and the momentum prototype method is proposed to fine-tune the prototype of each category. The experiments on both test-bench and wind turbine gearbox fault diagnosis demonstrated that SSMPN is more effective than the comparable methods under the same situation.