Adversarial domain adaptation of asymmetric mapping with CORAL alignment for intelligent fault diagnosis

Abstract

Rolling bearings play a vital role in the overall operation of rotating machinery. In practice, many learning methods for variable-speed fault diagnosis ignore task-specific decision boundaries, making it very difficult to completely match feature distribution between different domains. Therefore, to overcome this problem, an adversarial domain adaptation of asymmetric mapping with CORAL alignment is presented. The asymmetric mapping feature extractor is able to extract more specific-domain features with obvious distinction. Meanwhile, combining the maximum classifier discrepancy of deep transfer to give an adversarial approach and taking the task-specific decision boundaries into account, class-level alignment between the features of the source domain and target domain can be attempted. To prevent degenerate learning, which is possibly caused by asymmetric mapping and adversarial learning, the model is constrained by deep CORAL alignment to extract more domain-invariant features. Experimental results show that the proposed method can solve the variable-speed (a small span of intermediate vehicle speeds) fault diagnosis problem well, with high transfer accuracy and strong generalization.