An Iterative Resampling Deep Decoupling Domain Adaptation method for class-imbalance bearing fault diagnosis under variant working conditions

Abstract

Bearing fault diagnosis suffers from unsatisfied accuracy under variant working conditions, due to class-imbalance and distribution discrepancy of faulty data. Class-imbalance will cause classification bias that makes the diagnostic model tend to misclassify faulty classes, while the distribution discrepancy caused by variant working conditions will further reduce the model’s generalization. Thus, this paper is mainly focusing on solving the two challenges simultaneously, and an Iterative Resampling Deep Decoupling Domain Adaptation (IRDDDA) model is proposed. The model consists of a feature extractor, domain classifier, label classifier and feature resampler modules, and it adopts a decoupling two-stage training framework. In the first stage, the model learns domain invariant features from the source domain and target domain based on original class-imbalance datasets. In the second stage, the feature extractor is partially fixed, and the feature-based resampling method is used to adjust the biased classifier caused by the class-imbalance problem. An iterative training strategy is adopted to simultaneously update the upper-layers of the feature extractor and adjust the classifier till convergence. The method is verified on the CWRU open dataset and the dataset collected from our bearing test rig. The verification results show that our method can achieve a better fault diagnostic performance overall, especially in minority faulty classes.