Loading...
1 results
Search Results
Now showing 1 - 1 of 1
- Exploiting graph neural network with one-shot learning for fault diagnosis of rotating machineryPublication . Shuai Yang; Xu Chen; Yu Wang; Yun Bai; Pu, ZiqiangInsufficient training data often leads to overfitting, posing a significant challenge in diagnosing faults in mechanical devices, particularly rotating machinery. To address this issue, this paper introduces a novel approach employing a graph neural network (GNN) with one-shot learning for fault diagnosis in rotating machinery. Firstly, the Short-Time Fourier Transform (STFT) is utilized for data preprocessing to convert the one-dimensional data into two-dimensional pictures. Subsequently, Feature extraction a convolutional neural network (CNN) is utilized to perform feature extraction. By introducing the adjacency matrix to explore the spatial information within data, a graph neural network (GNN) method is proposed to achieve the fault classification of rotating machinery with small sample. The method utilizes GNN to process structural information between, transferring the distance metric from Euclidean space to non-Euclidean space. Classification accuracy is thereby improved based on information processing in non-Euclidean space.Experiments were implemented on two datasets to verify the proposed method, including an open dataset of the rolling bearing and an experimental rig of the rotate vector (RV) reducer in an industrial robot. Siamese Net, Matching Net, and sparse auto-encoder with random forest (SAE + RF) wereemployed as the comparisons to further prove the effectiveness of the proposed method. Results indicate that the proposed method outperforms all the comparative methods in both rotating machineries.