Dual Classifier-Discriminator Adversarial Networks for Open Set Fault Diagnosis of Train Bearings

Fault diagnosis of train bearings is crucial to ensure the reliability and safety of a running train. However, it is a big challenge to recognize new fault types of the train bearings under variable working conditions and heavy noise environments. This article proposes a new domain adaptation model, called dual classifier-discriminator adversarial networks (DCDAN), for open set fault diagnosis of the train bearings. The main contributions of the proposed DCDAN are that a novel weighting strategy is designed by constructing a weighting module with a dual classifier-discriminator structure to separate the new fault types in the target domain from the shared health types between the source and target domains, and a parallel channel attention module (PCAM) is embedded in the feature extractor of the DCDAN to promote feature extraction capability from noisy monitoring data. Specifically, the monitoring data are first input to the feature extractor to extract rich key health state information with the help of the PCAM. Then, the features are input to the weighting module to learn credible weights by unifying the similarity between the samples in the two domains evaluated from different perspectives. Finally, the weights are assigned to the adversarial training between the feature extractor and one of the classifiers for accurate separation of the new fault types and identification of the shared health types. Experimental results on two train bearing datasets verified the effectiveness and superiority of the proposed method, indicating that the proposed method has great potential for application in practical new fault recognition of train bearings.