Contrastive learning via swapped cluster assignments for bearing fault diagnosis

Abstract

In industrial scenarios, machinery and equipment operate in environments with many uncontrollable factors. These factors have a significant impact on vibration signals. For instance, differences in rotational speeds can make the collected data inconsistent. As a result, the tags collected from industrial environments are untrustworthy. It is impossible to know how many types of vibration signals are included in the dataset. In this paper, we propose a self-supervised pre-training method for bearing fault diagnosis called Contrastive Learning via Swapped Cluster Assignments (CLSCA). Technically, CLSCA is based on Momentum Contrast (MoCo) and uses an unsupervised clustering algorithm to replace the classification of positive and negative samples in MoCo. By autonomously extracting representations from unlabelled datasets and then generating pseudo-labels using a clustering algorithm, CLSCA is able to learn in a self-supervised manner. When the value of the clustering center is in the appropriate range, the classification accuracy of CLSCA is improved by about 4% compared to MoCo.

Robust transfer subspace learning based on low-rank and sparse representation for bearing fault diagnosis

“…Unlike the LRSR, the Gaussian noise can be well characterized by introducing N, so the above model is called LRSR-G. Now we show how to solve Problem (8). According to the alternating direction method of multipliers (ADMM) [37,38], one can iteratively update each variable while fixing the others.…”

“…We would also like to point out that the constraint on R can make the margin between one-hot encoding for each sample greater than a fixed constant 1. In comparison to Y + B • V in (8), R is no longer constrained by the 0 and 1 of the binary labeling matrix Y. It is able to learn label features directly from the source domain data in the subspace and therefore provides the possibility to improve the discriminative power and flexibility.…”

“…http://ise.thss.tsinghua.edu.cn/∼mlong/doc/transfer-joint-matching-cvpr14.zip 8. https://github.com/jindongwang/transferlearning/tree/master/code/ traditional/BDA.…”