Cross-Domain Data Fusion Generation: A Novel Composite Label-Guided Generative Solution for Adaptation Diagnosis

Tian Zhang,Jing Lin,Jinyang Jiao,Hao Li
DOI: https://doi.org/10.1016/j.knosys.2024.112284
IF: 8.139
2024-01-01
Knowledge-Based Systems
Abstract:Unsupervised domain adaptation (UDA) has been widely studied in the fault diagnosis community, and remarkable success has been witnessed. Nonetheless, existing UDA methods always assumes that data is sufficient in source and target domains. In real industrial scenarios, access to massive amount of data is expensive, especially for the target domain. A direct and effective solution is to expand the amount of the original dataset through the generative models. However, when the available information on the target domain is exceptionally scarce, obtaining a perform-well generative model is significantly challenging. In light of this, we propose a novel composite-label guided cross-domain data fusion generation (CCDFG) method, which trains a latent diffusion model by integrating the data, class labels, pseudo-labels, and domain labels from the source and target domains. Notably, we redesign the classifier-free guidance conditional generative strategy for diffusion models, enhancing the data generation quality of the target domain by borrowing information from the source domain. Subsequently, we present a sample selection strategy based on entropy-oriented prediction uncertainty, aiming to improve the transfer performance of the UDA models by retaining high-value generated data. Comprehensive experiments and evaluations on three mechanical systems demonstrate the effectiveness and application prospects of our method.
What problem does this paper attempt to address?