Semi-Supervised Transfer Learning Method for Bearing Fault Diagnosis with Imbalanced Data

Xia Zong, Rui Yang*, Hongshu Wang, Minghao Du, Pengfei You, Su Wang, Hao Su

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

11 Citations (Scopus)


Fault diagnosis is essential for assuring the safety and dependability of rotating machinery systems. Several emerging techniques, especially artificial intelligence-based technologies, are used to overcome the difficulties in this field. In most engineering scenarios, machines perform in normal conditions, which implies that fault data may be hard to acquire and limited. Therefore, the data imbalance and the deficiency of labels are practical challenges in the fault diagnosis of machinery bearings. Among the mainstream methods, transfer learning-based fault diagnosis is highly effective, as it transfers the results of previous studies and integrates existing resources. The knowledge from the source domain is transferred via Domain Adversarial Training of Neural Networks (DANN) while the dataset of the target domain is partially labeled. A semi-supervised framework based on uncertainty-aware pseudo-label selection (UPS) is adopted in parallel to improve the model performance by utilizing abundant unlabeled data. Through experiments on two bearing datasets, the accuracy of bearing fault classification surpassed the independent approaches.

Original languageEnglish
Article number515
Issue number7
Publication statusPublished - Jul 2022


  • fault diagnosis
  • imbalanced data
  • semi-supervised learning
  • transfer learning
  • uncertainty-aware pseudo-label selection


Dive into the research topics of 'Semi-Supervised Transfer Learning Method for Bearing Fault Diagnosis with Imbalanced Data'. Together they form a unique fingerprint.

Cite this