Zhenyu Wu1, Hongkui Zhang1, Juchuan Guo1, Yang Ji1 and Michael Pecht2
1 Engineering Research Center for Information Network, Ministry of Education, Beijing University of Posts and Telecommunications, Beijing 100876, PR China
2Center for Advanced Life Cycle Engineering, University of Maryland College Park, MD 20742, United States
Bearing fault diagnosis suffers from class imbalances and distributional discrepancies of fault data under different working conditions. The class imbalance of the fault class increases the difficulty of learning the classification boundary of the diagnostic model for the minority class. Furthermore, the diversities of feature distributions decrease the diagnostic model’s generalizability under various working conditions. Thus, in this paper, a deep adversarial transfer learning model for imbalanced bearing fault diagnosis (deep Imba-DA) is proposed to overcome these problems. In the proposed method, a cost-sensitive deep classifier is used to solve the class imbalance problem, and the domain adversarial subnet with the intraclass maximum mean discrepancy (MMD) is used to minimize the marginal and conditional distributional discrepancy between the source (data under one working condition) and target domain (data under another working condition) simultaneously. The performance of deep Imba-DA is evaluated and analyzed on the Case Western Reserve University (CWRU) and Paderborn datasets. The results show that deep Imba-DA outperforms other baseline methods on bearing diagnostic tasks.