Few-shot fault diagnosis of rotating machinery with two-branch prototypical networks

Cuixia Jiang (), Hao Chen (), Qifa Xu () and Xiangxiang Wang ()
Additional contact information
Cuixia Jiang: Hefei University of Technology
Hao Chen: Hefei University of Technology
Qifa Xu: Hefei University of Technology
Xiangxiang Wang: Anhui Ronds Science & Technology Incorporated Company

Journal of Intelligent Manufacturing, 2023, vol. 34, issue 4, No 8, 1667-1681

Abstract: Abstract In the fault diagnosis of rotating machinery, vibration signal or spectrum is usually used. As a data-driven method, deep learning has been introduced into the field of fault diagnosis. But it often confronts with two critical difficulties: few fault cases and single data source. To this end, we employ the prototype network to solve the problem of few fault cases, and use the two-branch technique to combine data sources in time domain and frequency domain. We introduce the two-branch network structure into the framework of the prototype network and develop a two-branch prototype network (TBPN) for fault diagnosis. The TBPN model is constructed through three main steps. First, we extract information from vibration signals in time domain and frequency domain respectively, and feed them into the model as two branches. Second, the prototype representation of each fault in time domain and frequency domain can be learned through metric learners, and the distances between fault prototypes and query faults features are then calculated. Third, the distances in time domain and frequency domain are integrated and incorporated into the softmax function for multi-classification. The performance of TBNP is verified by a real-world application on fault diagnosis of rotating machinery with the case data accumulated by an industrial Internet enterprise in China. The results show that the TBPN model is suitable for fault diagnosis in the case of small data. Compared with using time domain signals or spectrum alone, their combination use can improve the effectiveness of fault diagnosis.

Keywords: Fault diagnosis; Few-shot learning; Prototypical networks; Two-branch prototypical networks (search for similar items in EconPapers)
Date: 2023
References: View references in EconPapers View complete reference list from CitEc
Citations: View citations in EconPapers (2)

Downloads: (external link)
http://link.springer.com/10.1007/s10845-021-01904-x 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:34:y:2023:i:4:d:10.1007_s10845-021-01904-x

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

DOI: 10.1007/s10845-021-01904-x

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 ().