Skip to main content
SpringerLink
Log in

Intelligent fault diagnosis of rolling bearings based on the visibility algorithm and graph neural networks

  • Technical Paper
  • Published:
Journal of the Brazilian Society of Mechanical Sciences and Engineering Aims and scope Submit manuscript

Abstract

Rolling bearing fault diagnosis methods based on deep learning models usually apply regular data based on Euclidean space, and the relationship between data has not received corresponding attention. However, graph data based on non-Euclidean space has more comprehensive information expression ability and its intrinsic relationship is of great significance for fault identification. At present, the graph neural network diagnostic approach based on time domain cannot well capture the local features of the data adequately while constructing graph data, and the diagnosis accuracy is low. Therefore, a new method of bearing fault diagnosis based on graph neural network is proposed in this paper. Firstly, the time series is transformed by the visibility algorithm into non-Euclidean space graph structure data, which considerably enriches the internal relationship between data. Secondly, to identify and categorize rolling bearing failure characteristics, five different types of graph neural networks are fed with the generated graph data. At the same time, the effects of max, avg, and sum readout operations on the classification results of the graph neural network are analyzed. Finally, experiments are carried out on two bearing experimental data sets. The experimental findings demonstrate that the average accuracy of the five graph neural networks under the impact of max and avg readout operations is greater than 89% and 100%, respectively. Graph Attention Network (GAT) has the highest classification accuracy among them in this research, with an average accuracy of more than 90%. In general, graph neural networks can achieve good results in fault diagnosis of graph-level classification tasks, and the relationship information extracted by the visibility algorithm has certain advantages. The accuracy of this approach is somewhat enhanced when compared to previous time domain diagnostic methods of graph neural networks.

This is a preview of subscription content, log in via an institution to check access.

Access this article

Log in via an institution

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Institutional subscriptions

Fig. 1
Fig. 2
Fig. 3
Fig. 4
Fig. 5
Fig. 6
Fig. 7
Fig. 8
Fig. 9
Fig. 10
Fig. 11

Similar content being viewed by others

References

  1. Zhao W, Wang Z, Cai W, Zhang Q, Wang J, Du W, Yang N, He X (2022) Multiscale inverted residual convolutional neural network for intelligent diagnosis of bearings under variable load condition. Measurement 188:110511

    Article  Google Scholar 

  2. Cai W, Zhou J, Wang Z, Zheng L, Wang Z, Han X, He J, Du W (2019) A novel method for intelligent fault diagnosis of bearing based on capsule neural network. Hindawi 2019:1–17

    Google Scholar 

  3. Amirkhani A, Barshooi AH, Ebrahimi A (2022) Enhancing the robustness of visual object tracking via style transfer computers. Mater Contin 1:981–997

    Google Scholar 

  4. Gao J, Zhang T, Xu C (2019) Graph convolutional tracking ieee/cvf conference on computer vision and pattern recognition (CVPR) IEEE 2019, p 1–12

  5. Landrieu L, Simonovsky M (2018) Large-scale point cloud semantic segmentation with super-point graphs. In: Conference on computer vision and pattern recognition 2018, p 4558–4567

  6. Yan S., Xiong Y, Lin D (2018) Spatial temporal graph convolutional networks for skeleton-based action recognition. In: Conference on computer vision and pattern recognition 2018, p 7444-7452

  7. Jin W, Coley CW, Barzilay R, Jaakkola T (2017) Predicting organic reaction outcomes with weisfeiler-lehman network. In: 31st Conference on neural information processing systems 2017, p 2604–2613

  8. Ying R, He R, Chen K, Eksombatchai P, Hamilton WL, Leskovec J (2018) Graph convolutional neural networks for web-scale recommender systems. In: International conference on knowledge discovery & data mining 2018, p 1-10

  9. Bastings J , Titov I, Aziz W, Marcheggiani D, Simaan K (2017) Graph convolutional encoders for syntax-aware neural machine translation. In: Conference on empirical methods in natural language processing, p 1–10

  10. Shang J, Xiao C, Ma T , Li H, Sun J (2019) GAMENet: graph augmented memory networks for recommending medication combination. In: Proceedings of the AAAI conference on artificial intelligence, vol 33, p 1126-1133

  11. Jie M, Dong H, Yang X, Meng Z, Jia L, Qin Y, Xin G (2022) GCG: Graph Convolutional network and gated recurrent unit method for high-speed train axle temperature forecasting. Mech Syst Signal Process 163(12):108102

    Google Scholar 

  12. Chen Z, Xu J, Alippi C, Ding S, Shardt Y, Peng T, Yang C (2021) Graph neural network-based fault diagnosis: a review[J]. arXiv e-prints 2021, p 1–17

  13. Li T, Zhou Z, Li S, Sun C, Yan R, Chen X (2022) The emerging graph neural networks for intelligent fault diagnostics and prognostics: a guideline and a benchmark study. Mech Syst Signal Process 168:108653

    Article  Google Scholar 

  14. Kavianpour M, Ramezani A, Beheshti MTH (2022) A class alignment method based on graph convolution neural network forbearing fault diagnosis in presence of missing data and changing working conditions. Measurement 199:111536

    Article  Google Scholar 

  15. Jiang L, Li X, Wu L, Li Y (2022) Bearing fault diagnosis method based on a multi-head graph attention network[J]. Meas Sci Technol 33(7):075012

    Article  Google Scholar 

  16. Sun K, Huang Z, Mao H, Qin A, Li X, Tang W, Xiong J (2021) Multi-scale cluster-graph convolution network with multi-channel residual network for intelligent fault diagnosis[J]. IEEE Trans Instrum Meas 71:1–12

    Google Scholar 

  17. Zhang K, Chen J, He S, Li F, Yong F, Zhou F (2022) Triplet metric driven multi-head GNN augmented with decoupling adversarial learning for intelligent fault diagnosis of machines under varying working condition. J Manuf Syst 62:1–16

    Article  Google Scholar 

  18. Wang SH, Xing SB, Lei YG, Lu N, Li NP (2021) Vibration indicator-based graph convolutional network for semi-supervised bearing fault diagnosis. In: IOP conference series: materials science and engineering, vol 1043, p 052026

  19. Zhang D, Stewart E, Entezami M, Roberts CJ, Yu D (2020) Intelligent acoustic-based fault diagnosis of roller bearings using a deep graph convolutional network. Measurement 156:107585

    Article  Google Scholar 

  20. Gao Y, Chen M, Yu D (2021) Semi-supervised graph convolutional network and its application in intelligent fault diagnosis of rotating machinery. Measurement 186:110084

    Article  Google Scholar 

  21. Yu X, Tang B, Zhang K (2021) Fault diagnosis of wind turbine gearbox using a novel method of fast deep graph convolutional networks. IEEE Trans Instrum Meas 70:1–14

    Google Scholar 

  22. Liu J, Yang C, Zhou K (2021) Fault diagnosis of rotating machinery based on graph data deep mining. J Huazhong Univ Sci Technol Natural Sci Ed 49(9):1–5

    Google Scholar 

  23. Lacasa L, Luque B, Ballesteros F, Luque J, Nuno JC (2008) From time series to complex networks: the visibility graph. Proc Natl Acad Sci USA 105(13):4972–4975

    Article  MathSciNet  MATH  Google Scholar 

  24. Gori, M., Monfardini G, Scarselli F (2005) A new model for learning in graph domains Proceedings. In: IEEE international joint conference on neural networks, vol 2, p 729-734

  25. Scarselli F, Gori M, Tsoi AC, Hagenbuchner M, Monfardini G (2009) The graph neural network model[J]. IEEE Trans Neural Netw 20(1):61–80

    Article  Google Scholar 

  26. Amirkhani A, Barshooi AH (2022) Consensus in multi-agent systems: a review[J]. Artif Intell Rev 55(5):3897–3935

    Article  Google Scholar 

  27. Bruna J, Zaremba W, Szlam A, Lecun Y (2013) Spectral networks and locally connected networks on graphs. Comput Sci 2013:1–14

    Google Scholar 

  28. Defferrard M, Bresson X, Vandergheynst P (2016) Convolutional neural networks on graphs with fast localized spectral filtering. In: Proceedings of the 30th international conference on neural information processing systems, p 3844–3852

  29. Kipf TN, Welling M (2016) Semi-supervised classification with graph convolutional networks. In: International conference on learning representations, p 1-14

  30. Wu F, Zhang T, Souza AHD, Fifty C, Yu T, Weinberger KQ (2016) Simplifying graph convolutional networks. In: International conference on machine learning. PMLR 2016 6861-6871

  31. Hamilton WL, Ying R, Leskovec J (2017) Inductive representation learning on large graphs. In: Proceedings of the 31st international conference on neural information processing systems, p 1–19

  32. Velikovi P, Cucurull G, Casanova A , Romero A, P Liò P, Bengio Y (2018) Graph attention networks. In: International conference on learning representations 2018 1-12

  33. Xu K, Hu W, Leskovec J (2018) How powerful are graph neural networks? In: International conference on learning representations 2018, p 1-17

Download references

Funding

This work was supported by Shanxi Natural Science Foundation (201901D111239), China.

Author information

Authors and Affiliations

  1. College of Mechanical Engineering, Taiyuan University of Science and Technology, Taiyuan, China

    Shaohui Ning, Yonglei Ren & Yukun Wu

Authors
  1. Shaohui Ning
    View author publications

    You can also search for this author in PubMed Google Scholar

  2. Yonglei Ren
    View author publications

    You can also search for this author in PubMed Google Scholar

  3. Yukun Wu
    View author publications

    You can also search for this author in PubMed Google Scholar

Corresponding author

Correspondence to Shaohui Ning.

Ethics declarations

Conflict of interest

The authors declare that they have no known competing financial interests or personal relationships that could have appeared to influence the work reported in this paper.

Additional information

Technical Editor: Jarir Mahfoud.

Publisher's Note

Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Rights and permissions

Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

Cite this article

Ning, S., Ren, Y. & Wu, Y. Intelligent fault diagnosis of rolling bearings based on the visibility algorithm and graph neural networks. J Braz. Soc. Mech. Sci. Eng. 45, 72 (2023). https://doi.org/10.1007/s40430-022-03913-0

Download citation

  • Received:

  • Accepted:

  • Published:

  • DOI: https://doi.org/10.1007/s40430-022-03913-0

Keywords

Search

Navigation