Efficient Cross-Domain Fault Diagnosis via Distributed Multi-Source Domain Deep Transfer Learning

Abstract In actual industrial production, the working conditions of rotating machinery are complex and changeable, and the health-state monitoring data are increasingly large and difficult to label, which will seriously restrict the accuracy and efficiency of the cross-domain fault diagnosis (CDFD) of rotating machinery. Therefore, an efficient multi-source domain deep transfer learning (MDDTL) method for CDFD of rotating machinery is proposed. First, an MDDTL model is constructed to improve the accuracy of CDFD. In the model, a dual-phase domain alignment strategy is designed, which considers the alignment of feature distributions between each source and target domain pair in the feature space and that of the prediction probabilities between domain-specific fault classifiers in the output space. The fault prediction results from multiple different fault classifiers are merged dynamically by the proposed imbalanced adaptive prediction strategy. Secondly, a data-parallel distributed training scheme for the MDDTL model is proposed. Based on the idea of data parallelism, the distributed parallel training of the MDDTL model is performed with a Horovod-graphics processing unit platform, and the parameters are synchronously updated with the bandwidth-optimal Ring-AllReduce architecture. Under the premise of ensuring the accuracy of FD, the training time of the MDDTL model is significantly reduced. Finally, extensive experiments are conducted to verify the effectiveness of the proposed MDDTL method. The results demonstrate that the proposed method not only effectively improves the accuracy of CDFD of rotating machinery but also significantly improves the training efficiency of the MDDTL model. After adopting the proposed method, the diagnosis accuracies achieved under two different cross-working condition scenarios reach 97.09% and 97.87% respectively, and the model training time is reduced by 73.62% when facing a large-scale rotating machinery training set.