Balance-blended adversarial distribution and smooth-suppressed labels refinement network for partial transfer fault diagnosis
Fuzheng Liu,Haomiao Wang,Mingshun Jiang,Faye Zhang
DOI: https://doi.org/10.1016/j.engappai.2024.108723
IF: 8
2024-06-07
Engineering Applications of Artificial Intelligence
Abstract:In recent years, domain adaptation has received widespread attention in solving mechanical equipment fault diagnosis tasks with domain shift, which can transfer knowledge from labeled source domain to unlabeled target domain to reduce feature distribution discrepancy across domains. However, most existing methods assume that source domain and target domain share identical label space, which fails to handle practical engineering problems of scarce labeled data and incomplete fault types under big data. To address this engineering issue, partial domain adaptation network based on balance-blended adversarial and smooth-suppressed labels is proposed for partial domain fault diagnosis. On the one hand, to promote positive transfer of network, balance-blended adversarial utilizes source domain features to enhance target domain and shorten the distance of label distributions between different domains. Meanwhile, features in source and target domains are respectively domain intra-blending to explore the diversity of samples and internal structures in each domain. On the other hand, to enhance the discriminability of potential feature space, smooth-suppressed labels adopt smooth refinement mechanism to generate pseudo-labels and increase discriminative boundaries of target domain samples. Meanwhile, low prediction scores are uniformly assigned to uncertain probabilities to suppress the propagation of uncertainty classes. Experiments on different datasets show that the proposed method outperforms the state-of-the-art methods.
automation & control systems,computer science, artificial intelligence,engineering, electrical & electronic, multidisciplinary