Bearing fault diagnosis presents challenges, such as insufficient fault samples and significant fault data distribution variation in different bearing operating conditions. These problems cause traditional deep learning models to show poor generality and accuracy during bearing fault diagnosis. To address these challenges, this paper proposed a few-shot bearing fault diagnosis method based on an Ensemble Empirical Mode Decomposition (EEMD) parallel neural network and a relation network (RN). First, the original bearing fault vibration signal was decomposed by EEMD, while the decomposed signal components were processed via Short Time Fourier Transfor (STFT) to obtain a two-dimensional time-frequency feature map. Then, a parallel neural network was used for initial fault feature extraction, after which the extracted features were fused to construct a more accurate multi-dimensional fault feature map. A more precise fault feature vector was generated via the feature embedding module of the RN, and the fault features of the support and query sets were stitched to create a fault feature vector set. Finally, the relation module of the RN was used for the nonlinear distance determination of the fault feature vector set and to generate the relation score for the few-shot variable condition bearing fault diagnosis. In this paper, EEMD module is introduced into RN to construct multi-dimensional fault characteristics of the original fault signal. Original signal decomposition, STFT transformation and splicing effectively improve the randomness and blindness of convolution operations, improve the accuracy of fault feature extraction in RN, and thus improve the overall diagnostic performance of the model. The experimental results showed that the proposed model obtained higher diagnostic accuracy than the matching network (MN) and RN meta-learning fault diagnosis (MLFD) methods. The accuracy of fault diagnosis of the model in 5Way-Nshot is >80%, and the accuracy of fault diagnosis in 5Way-10shot is the highest at 95.2%.
Read full abstract