Partial transfer learning in machinery cross-domain fault diagnostics using class-weighted adversarial networks

• Published in: Neural networks Vol. 129; pp. 313 - 322
• Main Authors: Li, Xiang, Zhang, Wei, Ma, Hui, Luo, Zhong, Li, Xu
• Format: Journal Article
• Language: English
• Published: Elsevier Ltd 01.09.2020
• Subjects: Deep learning; Domain adversarial network; Fault diagnosis; Partial transfer learning; Rotating machinery; Fault diagnosis; Deep learning; Partial transfer learning; Domain adversarial network; Rotating machinery
• ISSN: 0893-6080; 1879-2782
• DOI: 10.1016/j.neunet.2020.06.014

Recently, transfer learning has been receiving growing interests in machinery fault diagnosis due to its strong generalization across different industrial scenarios. The existing methods generally assume identical label spaces, and propose minimizing marginal distribution discrepancy between source and target domains. However, this assumption usually does not hold in real industries, where testing data mostly contain a subspace of the source label space. Therefore, transferring diagnosis knowledge from a comprehensive source domain to a target domain with limited machine conditions is motivated. This challenging partial transfer learning problem is addressed in this study using deep learning-based domain adaptation method. A class weighted adversarial neural network is proposed to encourage positive transfer of the shared classes and ignore the source outliers. Experimental results on two rotating machinery datasets suggest the proposed method is promising for partial transfer learning. •The partial transfer learning problem in machinery fault diagnostics is addressed.•Class weighted adversarial neural network is proposed for feature extraction and condition alignments.•Instead of minimizing marginal distribution gap, the conditional data alignments are focused on.•Practical experiments validate the effectiveness of the proposed method on partial transfer learning tasks.