Abstract In the 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 be labeled, which will seriously restrict the accuracy and efficiency of 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. Firstly, a 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 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 MDDTL model is proposed. Based on the idea of data parallelism, the distributed parallel training of MDDTL model is performed with Horovod-GPU platform, and the parameters are synchronously updated with the bandwidth-optimal Ring-AllReduce architecture. Under the premise of ensuring the accuracy of fault diagnosis (FD), the training time of 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 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.
Read full abstract