IEEE Access (Jan 2020)

A Novel Transfer Learning Method for Fault Diagnosis Using Maximum Classifier Discrepancy With Marginal Probability Distribution Adaptation

  • Sixiang Jia,
  • Jinrui Wang,
  • Baokun Han,
  • Guowei Zhang,
  • Xiaoyu Wang,
  • Jingtao He

DOI
https://doi.org/10.1109/ACCESS.2020.2987933
Journal volume & issue
Vol. 8
pp. 71475 – 71485

Abstract

Read online

Effective fault diagnosis is essential to ensure the safe and reliable operation of equipment. In recent years, several transfer learning-based methods for diagnosing faults under variable working conditions have been developed. However, these models are designed to completely match the feature distributions between different domains, which is difficult to accomplish because each domain has unique characteristics. To solve this problem, we propose a framework based on the maximum classifier discrepancy with marginal probability distribution adaptation that focuses on task-specific decision boundaries. Specifically, this method captures ambiguous target samples through the predicted discrepancy between two classifiers for the target samples. Furthermore, marginal probability distribution adaptation facilitates the capture of target samples located far from the source domain, and these target samples are brought closer to the source domain through adversarial training. Experimental results indicate that the proposed method demonstrates higher performance and generalization ability than existing fault diagnosis methods.

Keywords