Self-supervised Learning for Intelligent Fault Diagnosis of Rotating Machinery with Limited Labeled Data

Guoqiang Li,Jun Wu,Chao Deng,Meirong Wei,Xuebing Xu
DOI: https://doi.org/10.1016/j.apacoust.2022.108663
IF: 3.614
2022-01-01
Applied Acoustics
Abstract:Supervised learning-based methods have been widely used for fault diagnosis of rotating machinery. The performance of these methods usually relies on the labeled fault samples. However, the labeling of fault sample requires the support of the prior knowledge and expert experience, and it is tedious to obtain enough labeled fault samples. In this paper, an intelligent fault diagnosis method named GIDIM-CNN is proposed based on grayscale image (GI) of the vibration signal, a self-supervised learning (SSL) method called deep InfoMax (DIM) and a convolutional neural network (CNN). Specifically, the vibration signals are sampled and converted into the GIs through a developed signal-to-image method. Then, a CNN-based feature encoder is trained by using the introduced DIM, during which a new loss function is designed by considering the maximum mutual information and statistical constraints of different unlabeled GI samples. Finally, limited labeled GI samples are used to optimize a desired model, which is constructed by combining the trained feature encoder with a fault classifier. Experimental studies on gearboxes and bearings show that the proposed method has good diagnostic performance and alleviates the model training overfitting problem caused by limited labeled samples. Meanwhile, the proposed method can be extended to different fault diagnosis tasks with limited labeled samples to improve the diagnostic accuracy.(c) 2022 Elsevier Ltd. All rights reserved.
What problem does this paper attempt to address?