SDCGAN: A CycleGAN-based single-domain generalization method for mechanical fault diagnosis
提出一种基于CycleGAN的单域泛化方法SDCGAN，通过生成与源域差异大的扩展域并提取域不变特征，结合对抗对比训练，提升机械故障诊断模型对未知域偏移的泛化能力。
In recent years, fault diagnosis based on domain generalization has attracted increasing attention as an effective approach to address the challenge of domain shift. most existing approaches depend on learning domain-invariant representations from multiple source domains, limiting their practical application in fault diagnosis. To address this issue, this paper introduces a single-domain generalization method for mechanical fault diagnosis, the Single-Domain Cycle Generative Adversarial Network (SDCGAN). A CycleGAN-based domain generation module is introduced to produce extended domains that exhibit substantial divergence from the source domain, enhancing the model's generalization capability. The diagnostic task module subsequently extracts domain-invariant features from both the source and extended domains. Furthermore, an adversarial contrastive training strategy is employed to learn generalized features robust to unknown domain shifts. Comprehensive experiments on two mechanical datasets verify the effectiveness of the proposed method, while ablation studies validate the contributions of its components, highlighting its potential for real-world applications.