ResNet50-3Cur-HGCN: A novel multimodal hybrid curvature space approach to bearing fault diagnosis
Cancan Yi,YuChen Wang,Yang Liu,Han Xiao
DOI: https://doi.org/10.1088/1361-6501/ad9d67
IF: 2.398
2024-12-12
Measurement Science and Technology
Abstract:Bearings are widely used in various fields due to their advantages of high load-bearing capacity, low friction loss, and minimal vibration and impact. In response to the current issues in the process of fault signal identification using graph neural networks, such as the insufficient expression of node data types, the inadequate exploration of feature information carried by the graph structure, and the singularity of mathematical space operations on graph data, this paper proposes a ResNet50-HGCN model based on the combination of Residual Network-50 (ResNet50) and Heterogeneous Graph Convolutional Neural Network (HGCN). The model is trained in a threefold mixed curvature (3 Curvature, 3Cur) space environment for semi-supervised learning, aiming to achieve accurate classification of bearing fault signals. Specifically, first, the time-domain signals and the components of the time-frequency map obtained from the Wavelet Synchrosqueezing Transform (WSST) are used as the bimodal node information for HGCN. Then, the graph network is merged in the 3Cur space for training and weighted validation, obtaining the predicted category labels of the model under the 3Cur configuration space. Finally, experimental data analysis is conducted, sparse confusion matrices for predicted categories are drawn, and four types of accuracy-related evaluation metrics are calculated. The experimental results show that the proposed ResNet50-3Cur-HGCN classification model outperforms other models in the experiment, achieving an accuracy of 97.71%, which verifies the advantage of the method in terms of accuracy and effectiveness. It also provides a good methodological reference for bearing fault diagnosis approaches centered on graph neural networks.
engineering, multidisciplinary,instruments & instrumentation