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

Weiwei Zhang,Deji Chen,Yang Xiao,Haibing Yin
DOI: https://doi.org/10.1109/tii.2023.3233960
IF: 12.3
2023-01-01
IEEE Transactions on Industrial Informatics
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 article offers a novel semi-supervised contrastive learning (SSCL) method based on a multiscale attention (MSA) mechanism and multitarget contrastive learning (MCL) for rolling bearing fault diagnosis under limited labeled samples. First, the proposed SSCL utilizes improved MCL to pretrain the model as a means of capturing potentially generic features in unlabeled data. Then, semi-supervised contrast learning is constructed and based on the pretrained 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 an MSA mechanism. Case Western Reserve University, the Society for Mechanical Failure Prevention Techniques, and Paderborn University are common motor-bearing datasets that are used in this article for testing the proposed technique. As a result of the experimental findings, SSCL is demonstrated to be superior to other approaches in similar situations.
What problem does this paper attempt to address?