Class-Imbalance Adversarial Transfer Learning Network for Cross-Domain Fault Diagnosis With Imbalanced Data

Abstract

Recently, cross-domain fault diagnosis based on transfer learning methods has been extensively explored and well-addressed when class-balance data with supervision information are available. However, data under machine faulty states are mostly difficult to collect; there is a huge divide between current transfer learning methods based on implicit class-balance data and real industrial applications. In this article, we propose a class-imbalance adversarial transfer learning (CIATL) network with input being imbalanced data to learn domain-invariant and knowledge. Within this framework, class-imbalance learning is embedded into the adversarial training process to learn class-separate diagnostic knowledge with imbalanced data, double-level adversarial transfer learning including marginal and conditional distribution adaptations is conducted to learn domain-invariant knowledge. Extensive experiments on a planetary gearbox rig with imbalanced data verify the effectiveness and generalization of the proposed method and show its superior performance over contrastive transfer learning methods. Moreover, the proposed method relaxes the underlying assumption that the success of current transfer learning regimes is rooted in class-balance data and extends the application of the transfer learning method for real-industrial cross-domain diagnosis tasks.