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



Zong, Xia, Yang, Rui ORCID: 0000-0002-5634-5476, Wang, Hongshu, Du, Minghao, You, Pengfei, Wang, Su and Su, Hao
(2022) Semi-Supervised Transfer Learning Method for Bearing Fault Diagnosis with Imbalanced Data. MACHINES, 10 (7). p. 515.

Access the full-text of this item by clicking on the Open Access link.

Abstract

<jats:p>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.</jats:p>

Item Type: Article
Uncontrolled Keywords: fault diagnosis, imbalanced data, semi-supervised learning, transfer learning, uncertainty-aware pseudo-label selection
Divisions: Faculty of Science and Engineering > School of Electrical Engineering, Electronics and Computer Science
Depositing User: Symplectic Admin
Date Deposited: 09 Sep 2022 14:17
Last Modified: 15 Mar 2024 15:43
DOI: 10.3390/machines10070515
Open Access URL: https://doi.org/10.3390/machines10070515
Related URLs:
URI: https://livrepository.liverpool.ac.uk/id/eprint/3164034