Deep prototypical networks based domain adaptation for fault diagnosis

Huanjie Wang, Xiwei Bai, Jie Tan () and Jiechao Yang
Additional contact information
Huanjie Wang: Institute of Automation, Chinese Academy of Sciences
Xiwei Bai: Institute of Automation, Chinese Academy of Sciences
Jie Tan: Institute of Automation, Chinese Academy of Sciences
Jiechao Yang: Institute of Automation, Chinese Academy of Sciences

Journal of Intelligent Manufacturing, 2022, vol. 33, issue 4, No 5, 973-983

Abstract: Abstract Due to the existence of domain shifts, the distributions of data acquired from different machines show significant discrepancies in industrial applications, which leads to performance degradation of traditional machine learning methods. In this paper, we propose a novel method that combines supervised domain adaptation with prototype learning for fault diagnosis. The proposed method consists of two modules, i.e., feature learning and condition recognition. The module of feature learning applies the Siamese architecture based on one-dimensional convolutional neural networks to learn a domain invariant subspace, which reduces the inter-domain discrepancy of distributions. The module of condition recognition applies a prototypical layer to learn the prototypes of each class. Then the classification task is simplified to find the nearest class prototype. Compared with existing intelligent fault diagnosis methods, this proposed method can be extended to address the problem when the classes from the source and target domains are partially overlapped. The model must generalize to unknown classes in the source domain, given only a few samples of each new target class. The effectiveness of the proposed method is verified using two bearing datasets. The model quickly converges with high classification accuracy using a few labeled target samples in training, even one per class can be effective.

Keywords: Bearing; Fault diagnosis; Domain adaptation; Prototype learning (search for similar items in EconPapers)
Date: 2022
References: View complete reference list from CitEc
Citations:

Downloads: (external link)
http://link.springer.com/10.1007/s10845-020-01709-4 Abstract (text/html)
Access to the full text of the articles in this series is restricted.

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:spr:joinma:v:33:y:2022:i:4:d:10.1007_s10845-020-01709-4

Ordering information: This journal article can be ordered from
http://www.springer.com/journal/10845

DOI: 10.1007/s10845-020-01709-4

Access Statistics for this article

Journal of Intelligent Manufacturing is currently edited by Andrew Kusiak

More articles in Journal of Intelligent Manufacturing from Springer
Bibliographic data for series maintained by Sonal Shukla () and Springer Nature Abstracting and Indexing ().