Self-supervised bi-classifier adversarial transfer network for cross-domain fault diagnosis of rotating machinery

Abstract

In real industrial scenarios, deep learning-based fault diagnosis has been a popular topic lately. Unfortunately, the source-trained model typically usually underperforms in target domain owning to changeable working conditions. To resolve this problem, a novel self-supervised bi-classifier adversarial transfer learning (SBATL) network by introducing self-supervised learning (SSL) and class-conditional entropy minimization is presented. Concretely, the SBATL is made up of a feature extractor, a discrepancy detector of two classifiers, and a clustering metric based on SSL, which jointly conducts self-supervised and supervised optimization in a two-stream training procedure. In the self-supervised stream, target pseudo labels obtained by SSL are used to construct the topological clustering metric for target feature optimization. In the supervised stream, the feature extractor and classifiers compete with each other in adversarial training, which bridges the discrepancy between two classifiers. Additionally, the class-conditional entropy minimization of target domain is further embedded into both streams to amend the decision boundaries of two classifiers to pass low-density regions. The results indicate that the SBATL gets better cross-domain fault diagnosis performances when compared with other popular methods.

Keywords: Bi-classifier adversarial transfer learning; Cross-domain fault diagnosis; Entropy minimization; Self-supervised learning.