Cross-domain open-set rolling bearing fault diagnosis based on feature improvement adversarial network under noise condition

Article type: Research Article

Authors: Wang, Haomiao^a | Li, Yibin^{a; *} | Jiang, Mingshun^b | Zhang, Faye^{b; *}

Affiliations: [a] Institute of Marine Science and Technology, Shandong University, Qigndao, Shandong, China | [b] School of Control Sciences and Engineering, Shandong University, Jinan, Shandong, China

Correspondence: [*] Corresponding author. Yibin Li. E-mail: [email protected] and Faye Zhang. Email: [email protected].

Abstract: Domain adaptation (DA) technology has the ability to solve fault diagnosis (FD) problems under variable operating conditions. However, DA technology faces two issues: (1) in general, vibration signals inevitably contain noise, which makes it difficult to extract discriminant features.(2) there are unknown fault types in target domain. These issues will lead to poor diagnostic performance. To solve above issues, a new cross-domain open-set transfer FD method called feature improvement adversarial network (FIAN) is proposed in this article. Specifically, to alleviate noise interference, a feature improvement module (FIM) is proposed and embedded into the backbone convolutional neural network to form new feature extractor. FIM uses soft threshold function to enhance important information and suppresses redundant information. Furthermore,open-set DA by back-propagation (OSBP) is introduced into FIAN. OSBP can predict the probability that a target domain sample belongs to an unknown category, so that it can effectively identify unknown and known category samples. Experimental results demonstrated its effectiveness and superiority in two bearing datasets.

Keywords: Fault diagnosis, rolling bearing, open-set domain adaptation, feature improvement module, adversarial network

DOI: 10.3233/JIFS-236593

Journal: Journal of Intelligent & Fuzzy Systems, vol. 46, no. 2, pp. 5073-5085, 2024