Abstract
AbstractThe manuscript proposes a fault diagnosis method based on graph neural network (GNN) with one-shot learning to effectively diagnose rolling bearings under variable operating conditions. In this proposed method, the convolutional neural network is utilized for feature extraction, reducing loss in the process. Subsequently, GNN applies an adjacency matrix to generate codes for one-shot learning. Experimental verification is conducted using open data from Case Western Reserve University Rolling Bearing Data Center, where four different working conditions with six types of typical faults are selected as input signals. The classification accuracy of the proposed method reaches 98.02%. To further validate its effectiveness, traditional single-learning neural networks such as Siamese, Matching Net, Prototypical Net and (Stacked Auto Encoder) SAE are introduced as comparisons. Simulation results that the proposed method outperforms all chosen methods.
Funder
National Natural Science Foundation of China
Publisher
Springer Science and Business Media LLC
Reference38 articles.
1. M. Ali, J.H. Shah, M.A. Khan, M. Alhaisoni, U. Tariq, Brain tumor detection and classification using pso and convolutional neural network. Comput. Mater. Contin. 73(3), 4501–4518 (2022)
2. M. Bilardo, G. Fraisse, M. Pailha, E. Fabrizio, Design and experimental analysis of an integral collector storage (ICS) prototype for DHW production. Appl. Energy 259, 114104 (2020). https://doi.org/10.1016/j.apenergy
3. Y. Ding, X. Tian, L. Yin, X. Chen, S. Liu, Multi-scale relation network for few-shot learning based on meta-learning, in International Conference on Computer Vision Systems (Springer, Cham, 2019), pp. 343–352
4. W. Fuan, J. Hongkai, S. Haidong et al., An adaptive deep convolutional neural network for rolling bearing fault diagnosis. Meas. Sci. Technol. 28(9), 95005 (2017)
5. H.Z. Gao, L. Lin, X.G. Chen, Feature extraction and recognition for rolling element bearing faultutilizing short-time fourier transform and non-negative matrix factorization. Chin. J. Mech. Eng. 28, 96–105 (2015)