Domain Adaptation With Self-Supervised Learning and Feature Clustering for Intelligent Fault Diagnosis

• Published in: IEEE transaction on neural networks and learning systems Vol. 35; no. 6; pp. 7657 - 7670
• Main Authors: Lu, Nannan, Xiao, Hanhan, Ma, Zhanguo, Yan, Tong, Han, Min
• Format: Journal Article
• Language: English
• Published: United States IEEE 01-06-2024; The Institute of Electrical and Electronics Engineers, Inc. (IEEE)
• Subjects: Adaptation; Clustering; domain adaptation; Employee welfare; Fault diagnosis; Feature extraction; intelligent fault diagnosis; Learning; Manifolds; Neural networks; pseudolabeling; Representation learning; Self-supervised learning; Training; unbiased feature learning; Vibrations
• ISSN: 2162-237X; 2162-2388
• DOI: 10.1109/TNNLS.2022.3219896

Domain adaptation indeed promotes the progress of intelligent fault diagnosis in industrial scenarios. The abundant labeled samples are not necessary. The identical distribution between the training and testing datasets is not any more the prerequisite for intelligent fault diagnosis working. However, two issues arise subsequently: Feature learning in domain adaptation framework tends to be biased to the source domain, and unreliable pseudolabeling seriously impacts on the conditional domain adaptation. In this article, a new domain adaptation approach with self-supervised learning and feature clustering (DASSL-FC) is proposed, trying to alleviate the issues by unbiased feature learning and pseudolabels updating strategy. Taking different transformation methods as pretext, the transformed data and its pretext train a neural network in an SSL way. As to pseudolabeling, clusters are taken as the auxiliary information to correct the network predicted labels in terms of the "strong cluster" rule. Then, the updated pseudolabels and their confidence are enforced to further estimate the conditional distribution discrepancy and its confidence weight. To verify the effectiveness of the proposed method, the experiments are implemented on intraplatform and interplatforms for simulating the practical scenarios.