Semi-Supervised Contrast Learning Based on Multiscale Attention and Multitarget Contrast Learning for Bearing Fault Diagnosis

Abstract

In the field of bearing fault diagnosis it is difficult to obtain a large amount of labeled data for training, and therefore it is easy to overfit the model, which makes it less robust and less generalizable. To address this problem, this work offers a novel semi-supervised contrastive learning (SSCL) method based on a multi-scale attention (MSA) mechanism and multi-target contrastive learning (MCL) for rolling bearing fault diagnosis under limited labeled samples. Firstly, the proposed SSCL utilizes improved multi-objective contrast learning (MCL) to pre-train the model as a means of capturing potentially generic features in unlabeled data. Then, semi-supervised contrast learning is constructed and based on the pre-trained model combining a limited amount of labeled data and a large amount of unlabeled data to jointly learn the feature mapping and further enhance the feature extraction capability of the model through a multi-scale attention (MSA) mechanism. Case Western Reserve University (CWRU), the Society for Mechanical Failure Prevention Techniques (MFPT), and Paderborn University (PU) are common motor-bearing datasets that are used in this research for testing the proposed technique. As a result of the experimental findings, SSCL is demonstrated to be superior to other approaches in similar situations.