Few-shot bearing fault diagnosis by semi-supervised meta-learning with graph convolutional neural network under variable working conditions

Abstract

Aiming at the problems of low accuracy and weak generalization ability in fault diagnosis caused by complex working conditions and limited fault samples of bearings, a few-shot bearing fault diagnosis method by semi-supervised meta-learning with simplifying graph convolutional neural network (semi-meta-sgc) is proposed. Firstly, the time domain signal is converted into the frequency domain by fast Fourier transform (FFT). Secondly, considering each spectrum sample as a node, a k-nearest neighbor (KNN) graph is constructed according to the adjacency between nodes, thus transforming the bearing fault classification into a graphical node classification. Then, relying on meta-learning, graph convolutional neural network, and semi-supervised learning, a high-accuracy node classifier is obtained by using a small number of training samples to achieve the classification of bearing faults under variable working conditions. Finally, the proposed method is verified by four cases and compared with other methods. The results show that this method has higher recognition accuracy and generalization performance.