A Generalized Graph Contrastive Learning Framework for Few-Shot Machine Fault Diagnosis
Chaoying Yang, Jie Liu, Qi Xu, Kaibo Zhou
Abstract
Graph data-driven machine fault diagnosis methods make success using sufficient data recently. However, in the actual industry, there are rare failure data in historical data, leading to insufficient graph representation ability and reducing diagnosis performance. In this article, a generalized graph contrastive learning (GCL) framework for few-shot machine fault diagnosis is proposed. First, spectrum features of vibration data-based samples are used to calculate Euclidean distance matrix for constructing K-nearest neighborhood graph (KNNG), where <italic xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">K</i> adjacent neighbors of each sample are connected. Avoiding excess calculation cost for graph construction, positive and negative KNNGs are constructed by changing parameter <italic xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">K</i> . To make full use of few-shot samples, an unsupervised GCL subtask is set for pretraining graph deep learning model. Further, the unsupervised pretrained model is semisupervised trained using original KNNGs for outputting unlabeled nodes’ labels. The proposed method achieves 99.83%, 99.56% in bearing and gearbox dataset, respectively, and the proposed GCL framework works for different graph neural networks.