Mitigating Negative Transfer Learning In Source Free-Unsupervised Domain Adaptation for Rotating Machinery Fault Diagnosis

Abstract

Driven by the data-centric industrial revolution, Prognostics and Health Management (PHM) technology based on sophisticated Deep Learning (DL) methods has become crucial for industrial fault diagnosis. Based on well-labeled data, the DL methods further improve PHM development. However, the efficiency of the involved DL methods is impacted significantly due to the limited data collection. To address this problem, Source Free-Unsupervised Domain Adaptation (SF-UDA) has been proposed to improve DL efficiency without fully labeled source data. However, the traditional SF-UDA methods are constrained by predefined label numbers and insufficient distribution alignment. To solve this problem, we propose the use of Initial Pseudo Labels (IPLs) generated through Adaptive Bandwidth-based Mean Shift Clustering (ABMSC), which are then refined using cosine similarity with unlabeled target data, bolstering SF-UDA’s efficiency. Furthermore, we introduce Weight-Aware Regularization (W-AR) to effectively mitigate negative transfer. Experimental results show significant accuracy improvements, with gains of 1.5%−6.7% and 1.3%−5% in the same operating environment, and 0.4%−24.8% and 3.1%−8.6% in different operating environments.