Intelligent fault diagnosis of rotating machinery using a multi-source domain adaptation network with adversarial discrepancy matching

In the health management of modern rotating machinery, domain adaptation is an effective method to solve the diagnostic problems of insufficient labeled signals and poor generalization performance. In engineering scenarios, obtaining signals from various source domains can ensure abundant feature information and contribute to diagnostic ability improvement compared with learning from a single source domain. This paper presents a deep multi-source adversarial discrepancy matching adaptation network (MADMAN) for enhancing the accuracy of cross-domain intelligent diagnosis. Firstly, the proposed method makes use of the generalization knowledge learned from multiple domains to diagnose the unknown task, and adaptively adjusts the weight factors of multiple source domains utilizing the self-attention mechanism. Secondly, to better alleviate the domain shift phenomenon between different domains, the discrepancy matching technique is applied to dynamically align the feature distributions of different domains. Thirdly, an adversarial classifier training method is incorporated to raise the transferability by considering the decision boundary of specific tasks. The proposed method is verified by extensive experiments using two bearing datasets, and the superiority of the presented approach is demonstrated by comparison with advanced methods.