A Prototypical Networks-based Multi-task Model for Few-shot Fault Diagnosis

Zhihao Huo,Xiaoyu Yang,Tao Yang,Huijin Fan,Housheng Su,Ying Zheng
DOI: https://doi.org/10.1109/ddcls55054.2022.9858586
2022-01-01
Abstract:Due to the powerful classification performance, Prototypical Networks is widely used in few-shot fault diagnosis. However, insufficient feature extraction ability and negligence of inter-class distance in the Prototypical Networks may lead to the decrease of diagnosis accuracy. To solve the above problems, a Prototypical Networks-based multi-task model (PNBMTM) is proposed in this paper. Firstly, a convolutional autoencoder is built based on the feature extraction network of the Prototypical Networks, in which the optimization of the reconstruction loss is taken as a new task to obtain more efficient features. Secondly, a channel attention module (CAM) is designed to be applied to each layer of the encoder, which focuses on important features and suppresses useless features. Thirdly, relative distance loss is introduced while considering the effect of intra-class distance and inter-class distance to make inter-class distance larger and intra-class distance smaller, which helps to improve the quality of the features. And the optimization of relative distance loss is consequently taken as a new task to constitute a multi-task fault diagnosis model. Finally, a mechanical fault simulation experiment platform is built to verify the validity of the proposed model. The experimental results show that the proposed model has superior performance compared with other models.
What problem does this paper attempt to address?