A Semi-supervised Fault Diagnosis Method Based on Graph Convolution for Few-shot Fault Diagnosis

##plugins.themes.bootstrap3.article.main##

##plugins.themes.bootstrap3.article.sidebar##

Published Jun 27, 2024
Yuyan Li Tian Wang Jingsong Xie

Abstract

In practical bearing fault diagnosis, labeled fault data are difficult to obtain, and limited samples will lead to training overfitting. To address the above problems, a semi-supervised fault diagnosis method based on graph convolution is proposed. Firstly, the KNN graph construction method based on Euclidean distance (ED-KNN) is used to achieve label propagation. Then, a graph convolutional network framework based on dot product attention mechanism (GPGAT) was constructed to enhance the weights of high similarity nodes and diagnose bearing faults. The proposed method is validated on a public bearing dataset. The results show that the proposed method can make full use of very few labeled samples for fault diagnosis. Compared with other state-of-the-art methods, the proposed method achieves better diagnosis performance.

How to Cite

Li, Y., Wang, T., & Xie, J. (2024). A Semi-supervised Fault Diagnosis Method Based on Graph Convolution for Few-shot Fault Diagnosis. PHM Society European Conference, 8(1), 8. https://doi.org/10.36001/phme.2024.v8i1.4106
Abstract 149 | PDF Downloads 125

##plugins.themes.bootstrap3.article.details##

Keywords

Fault diagnosis, Semi-supervised, Graph Convolution, Few-shot

References
Zhang, S., Su, L., Gu, J., Li, K., Zhou, L., & Pecht, M. (2023). Rotating machinery fault detection and diagnosis based on deep domain adaptation: A survey. Chinese Journal of Aeronautics, 36(1), 45–74. doi: 10.1016/j.cja.2021.10.006 Jiao, J., Zhao, M., Lin, J., & Liang, K. (2020). A comprehensive review on convolutional neural network in machine fault diagnosis. Neurocomputing, 417, 36–63. doi: 10.1016/j.neucom.2020.07.088 Yang, X., Song, Z., King, I., & Xu, Z. (2023). A Survey on Deep Semi-Supervised Learning. IEEE Transactions on Knowledge and Data Engineering, 35(9), 8934–8954. doi: 10.1109/TKDE.2022.3220219 Ding, Y., Ma, L., Ma, J., Wang, C., & Lu, C. (2019). A Generative Adversarial Network-Based Intelligent Fault Diagnosis Method for Rotating Machinery Under Small

Zhang, X.-Y., Shi, H., Zhu, X., & Li, P. (2019). Active semisupervised learning based on self-expressive correlation with generative adversarial networks. Neurocomputing, 345, 103–113. doi: 10.1016/j.neucom.2019.01.083 Scarselli, F., Gori, M., Tsoi, A. C., Hagenbuchner, M., & Monfardini, G. (2009). The Graph Neural Network Model. IEEE Transactions on Neural Networks, 20(1), 61–80. doi: 10.1109/TNN.2008.2005605 Xie, Z., Chen, J., Feng, Y., & He, S. (2022). Semi-supervised multi-scale attention-aware graph convolution network for intelligent fault diagnosis of machine under extremely-limited labeled samples. Journal of Manufacturing Systems, 64, 561–577. doi:

10.1016/j.jmsy.2022.08.007

Kavianpour, M., Ramezani, A., & Beheshti, M. T. H. (2022). A class alignment method based on graph convolution neural network for bearing fault diagnosis in presence of missing data and changing working conditions. Measurement, 199, 111536. doi:

10.1016/j.measurement.2022.111536

Kim, D., & Oh, A. (2022). How to Find Your Friendly Neighborhood: Graph Attention Design with SelfSupervision. arXiv. Retrieved from http://arxiv.org/abs/2204.04879 Smith, W. A., & Randall, R. B. (2015). Rolling element bearing diagnostics using the Case Western Reserve University data: A benchmark study. Mechanical Systems and Signal Processing, 64–65, 100–131. doi:

10.1016/j.ymssp.2015.04.021

Huang, H., & Baddour, N. (2018). Bearing vibration data collected under time-varying rotational speed conditions. Data in Brief, 21, 1745–1749. doi:

10.1016/j.dib.2018.11.019

Veličković, P., Cucurull, G., Casanova, A., Romero, A., Liò, P., & Bengio, Y. (2018, February 4). Graph Attention Networks. arXiv. Retrieved from http://arxiv.org/abs/1710.10903 Shi, Y., Huang, Z., Feng, S., Zhong, H., Wang, W., & Sun,

Y. (2021, May 9). Masked Label Prediction: Unified Message Passing Model for Semi-Supervised Classification. arXiv. Retrieved from http://arxiv.org/abs/2009.03509 Morris, C., Ritzert, M., Fey, M., Hamilton, W. L., Lenssen, J. E., Rattan, G., & Grohe, M. (2019). Weisfeiler and Leman Go Neural: Higher-Order Graph Neural Networks. Proceedings of the AAAI Conference on Artificial Intelligence, 33(01), 4602–4609. doi:10.1609/aaai.v33i01.33014602
Section
Technical Papers