Mutual-assistance semisupervised domain generalization network for intelligent fault diagnosis under unseen working conditions

Abstract

Generalizing deep models to unseen working conditions is an essential topic for intelligent fault diagnosis. Existing domain generalization-based fault diagnosis (DGFD) methods usually require sufficient annotated samples from all observed domains during the training phase, while annotating abundant samples is an expensive and difficult task. Therefore, this study proposes a mutual-assistance network for semisupervised domain generalization fault diagnosis (SemiDGFD), where only one source domain is labeled along with several unlabeled source domains. Reliable pseudo labels are assigned to unlabeled data with knowledge assistance from labeled data. Then, an entropy-based sample purification mechanism is designed to improve the quality of pseudo-labeled samples. Finally, pseudo-labeled samples cooperate with real-labeled samples to serve as the input of a low-rank decomposition, which discovers domain invariance against domain shift. Extensive diagnostic experiments demonstrate that the proposed method can obtain higher precision than other popular SemiDGFD methods and achieve comparable performance with up-to-date fully-labeled DGFD methods.