Subdomain adaptation joint attention network enabled two-stage strategy towards few-shot fault diagnosis of LRE turbopump

Dongfang Zhao,Shulin Liu,Zhonghua Miao,Hongli Zhang,Wei Dou
DOI: https://doi.org/10.1016/j.aei.2024.102366
IF: 8.8
2024-01-27
Advanced Engineering Informatics
Abstract:Turbopump is the key component of liquid rocket engine (LRE), and its working status directly affects the performance of the launch system. However, it is usually difficult to obtain abundant training samples for turbopump, which has become a major factor limiting the application of deep learning based fault diagnosis technologies. In this paper, a two-stage strategy towards turbopump fault diagnosis under both limited labeled and unlabeled data based on variational auto-encoder (VAE) and subdomain adaptation joint attention network (SAJANet) is developed. In the suggested strategy, to better coordinate the importance of different layers during domain adaptation , the joint attention mechanism that can take the relationship between different layers into consideration is proposed and embedded into a subdomain adaptation network to developed the SAJANet. Subsequently, by combining the developed SAJANet with VAE, the "Generation-Transfer" two-stage diagnosis strategy for LRE turbopump under both limited labeled and unlabeled data is developed. In the suggested two-stage strategy, the function of VAE is to preliminary expand the data volume utilizing extremely few labeled data, while the SAJANet followed by is utilized to perform transfer learning from the VAE manual samples to the actual samples. Through the above strategy, the SAJANet can be well pre-trained by the VAE data, and the introduction of the adaptation stage can further improve the model performance against actual data by fully utilizing the limited actual data. The superiority of the proposed method is verified by the data of an actual LRE turbopump.
engineering, multidisciplinary,computer science, artificial intelligence
What problem does this paper attempt to address?