Using Heterogeneous Extractor to Transfer Local-Global Knowledge for Cross-Domain Rolling Bearing Fault Diagnosis

Abstract

The intelligent fault diagnosis (IFD) methods obtain superior performance in ensuring the safety of mechanical systems, but varying working conditions degrade the performance of intelligent models. Fortunately, unsupervised domain adaptation (UDA) has been used to handle the bias and unannotated data. High-quality features contribute to facilitating subsequent domain alignment and enhancing diagnostic performance. This article aims to address the contradiction between using complex neural networks to extract better fault features and the resulting longer inference time. Specifically, a heterogeneous extractor is designed by integrating a pure CNN-based main network in parallel with a hybrid auxiliary network. The auxiliary network consists of a CNN and a ViT, connected in series, which extract local and global fault knowledge, respectively. Then, a training strategy is proposed to help the main branch enrich the extracted features, where the pure CNN is optimized to distinguish the hard-to-transfer samples identified by auxiliary CNN-ViT extractor. Finally, an information filter mechanism is introduced to facilitate mutual feature learning between the two branches. Experiments are constructed on two diagnosis datasets and a practical platform, where the comparison studies manifest the superiority of our method in fault diagnosis tasks under varying working conditions.