Relationship Transfer Domain Generalization Network for Rotating Machinery Fault Diagnosis Under Different Working Conditions
Quan Qian, Jianghong Zhou, Yi Qin
Abstract
Many domain adaptation (DA) models have been explored for fault transfer diagnosis. However, their successes completely rely on the availability of target-domain samples during the training process. As target domain is usually unseen, the domain-adaptation-based diagnostic models cannot meet the requirement of real-time diagnosis in actual engineering. To achieve the domain confusion in the actual diagnosis scenario, a novel relationship transfer (RT) diagnosis framework is first proposed, which can indirectly measure and reduce the distribution discrepancy between the source domain and unseen target domain. Based on the proposed RT framework, a new domain generalization transfer method, called relationship transfer domain generalization network (RTDGN) is constructed. RTDGN is divided into two phases including task-irrelevant domain adaptation (TIDA) and task-relevant domain generalization (TRDG). In the TIDA phase, a DA adversarial network with several domain discriminators is built to enhance the domain confusion of RT framework. Furthermore, to bring the adversarial network a more general domain confusion ability, a new inverse entropy loss is designed. In the TRDG phase, a residual fusion classifier is constructed to improve the generalization ability of fault classifier. Finally, the experimental results on the wind turbine planetary gearbox dataset and bearing dataset verify the effectiveness and superiority of the proposedRTDGN.