Supervised Contrastive Learning-Based Domain Adaptation Network for Intelligent Unsupervised Fault Diagnosis of Rolling Bearing
Yongchao Zhang, Zhaohui Ren, Shihua Zhou, Ke Feng, Kun Yu, Zheng Liu
Abstract
Fault diagnosis of rolling bearing is essential to guarantee production efficiency and avoid catastrophic accidents. Domain adaptation is emerging as a critical technology for the intelligent fault diagnosis of rolling bearing. Most existing solutions learn domain-invariant features by statistical moment matching, adversarial training, or fusing two algorithms. However, these domain adaptation methodologies overemphasized learning domain-invariant features and ignored the generalization of classification performance on the target domain, which leads to inevitable misclassification. To address this issue, we propose a supervised contrastive learning-based domain adaptation network (SCLDAN) for cross-domain fault diagnosis of the rolling bearing in this paper. The SCLDAN develops a 1-D convolutional residual network to learn the raw signal features and employs the maximum mean discrepancy loss to achieve global domain alignment. In addition, a novel supervised contrastive learning approach is proposed, where a supervised contrastive loss and a mutual information loss are established to learn the class-specific information and improve the reliability of target prediction labels. Thus, the ambiguous data samples residing near the class boundaries of the target domain can be accurately identified, and the diagnosis accuracy is significantly improved. Extensive experiments on two experimental scenarios demonstrate the effectiveness of the proposed method.