Semi-supervised source-free domain adaptation method via diffusive label propagation for rotating machinery fault diagnosis

Zhiheng Su, Penglong Lian, Penghui Shang, Jiyang Zhang, Hongbing Xu, Jianxiao Zou and Shicai Fan

Reliability Engineering and System Safety, 2024, vol. 252, issue C

Abstract: Traditional domain adaptation methods address the performance degradation of deep learning models in diagnostic tasks under varying working conditions, assuming that source data is available. With the increasing demand for data privacy protection, access to source data has become restricted, leading to the rise of source-free domain adaptation methods. However, without any labels in the target domain, the adaptation process and performance of the model may be unstable and impractical for real-world scenarios. To address these issues, a semi-supervised source-free domain adaptation method via diffusive label propagation (SSFDA-DLP) is proposed in this paper. With only one labeled target sample provided for each class, SSFDA-DLP can diffuse the label information to the unlabeled target data through repeated iterations of pre-training with labeled target data and annotating new target data that are adjacent to the labeled ones. Considering that label propagation may incorrectly annotate some unlabeled samples, and to make full use of the unlabeled target data, feature and probability spaces consistency regularization is utilized to further improve the performance of the pre-trained model. The effectiveness and superiority of our method in source-free domain adaptation diagnostic tasks were evaluated on four datasets, including bearings and gears.

Keywords: Intelligent fault diagnosis; Transfer learning; Domain adaptation; Semi-supervised source-free; Label propagation (search for similar items in EconPapers)
Date: 2024
References: View references in EconPapers View complete reference list from CitEc
Citations:

Downloads: (external link)
http://www.sciencedirect.com/science/article/pii/S0951832024004800
Full text for ScienceDirect subscribers only

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:eee:reensy:v:252:y:2024:i:c:s0951832024004800

DOI: 10.1016/j.ress.2024.110408

Access Statistics for this article

Reliability Engineering and System Safety is currently edited by Carlos Guedes Soares

More articles in Reliability Engineering and System Safety from Elsevier
Bibliographic data for series maintained by Catherine Liu ().