Residual Adversarial Subdomain Adaptation Network Based on Wasserstein Metrics for Intelligent Fault Diagnosis of Bearings

Abstract

Subdomain adaptation plays a significant role in the field of bearing fault diagnosis. It effectively aligns the pertinent distributions across subdomains and addresses the frequent issue of lacking local category information in domain adaptation. Nonetheless, this approach overlooks the quantitative discrepancies in distribution between samples from the source and target domains, leading to the vanishing gradient issue during the training of models. To tackle this challenge, this paper proposes a bearing fault diagnosis method based on Wasserstein metric residual adversarial subdomain adaptation. The Wasserstein metric is introduced as the optimized objective function of the domain discriminator in RASAN-W. The distribution discrepancy between the source domain and target domain samples is quantitatively measured, achieving the alignment of the relevant subdomain distributions between the source domain and the target domain. Ultimately, extensive experiments conducted on two real-world datasets reveal that the diagnostic accuracy of this method is significantly enhanced when compared to various leading bearing fault diagnosis techniques.

Keywords

• domain adaptation
• fault diagnosis
• Wasserstein metrics
• deep learning