Unsupervised domain adaptation bearing fault diagnosis method based on joint feature alignment

Abstract

In this paper, the issue of cross-condition fault diagnosis of bearing is studied. During actual operation, the conditions of bearing vary due to changes in factors such as rotation speed and load, and the data distribution between different working conditions varies. Deep learning models that perform well in one condition are not ideal when applied to another condition directly. To address this problem, a novel unsupervised domain adaptation fault diagnosis method based on joint feature alignment is proposed in this paper. 1D-CNN is used as a weight-shared feature extractor to extract the features from both the source and target domains. The discrepancies in marginal and conditional distributions between the source and target domains are comprehensively considered by multi-layer multi-bandwidth Cauchy kernel maximum mean discrepancy (MB-CMMD) and mutual information (MI). The domain drift is reduced by aligning the feature representations of source and target domains. The network after feature alignment demonstrates a notable enhancement in the diagnostic accuracy of unlabeled samples within the target domain. The experimental results demonstrate that, in comparison to other domain adaptation approaches, The proposed approach can significantly enhance the accuracy of fault diagnosis while realizing feature alignment.