Abstract
The deep learning model can fully extract the rich information in the signal and provide a better recognition effect as compared to traditional fault diagnosis approaches, which rely on manual analysis. However, the deep learning model has the problem that too many parameters lead to high training cost and low generalization ability. Therefore, this paper proposes a lighter fault diagnosis model for rolling bearings using Gramian Angular Field (GAF) and EfficientNet-B0. Firstly, Gramian Angular Field is used to encode one-dimensional vibration signal into a two-dimensional temporal image, the two-dimensional image is then loaded into the selected EfficientNet-B0 for training in automatic feature extraction and classification recognition before a test set is used to confirm the model’s recognition accuracy. The results show that the recognition rate of bearing faults of the lightweight fault diagnosis model proposed in this paper based on Gramian Angle field and EfficientNet-B0 reaches 99.27%, and the number of parameters of the EfficientNet-B0 is about 1/5 of that of the ResNet-50. Compared with common fault diagnosis methods, this model has a lighter weight convolutional network model, better generalization and higher recognition rate.
Supported by Shanghai Rising-Star Program(No.23QA1403800), National Natural Science Foundation of China (No.62076160) and Natural Science Foundation of Shanghai (No. 21ZR1424700).
This is a preview of subscription content, log in via an institution to check access.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
Similar content being viewed by others
References
Zhang, L., Hu, Y., Zhao, L., Zhang, N., Wang, X., Wen, P.: Fault diagnosis of rolling bearings using recursive graph coding technique and residual network. J. Xi’an Jiaotong Univ. 57(02), 110–120 (2023)
Liu, R., Yang, B., Zio, E., et al.: Artificial intelligence for fault diagnosis of rotating machinery: a review. Mech. Syst. Signal Process. 108, 33–47 (2018)
Xinwei, S., Ji Aimin, D., Zhantao, C.X., Xinhai, L.: Diagnosis method of variable speed fault of rolling bearing in gearbox of rolling stock. J. Harbin Instit. Technol. 55(01), 106–115 (2023)
Fenglin, Y., Changkai, X., Shining, L., Hao, Y., Zhe, M.: Research on rolling bearing fault diagnosis based on wavelet packet transform and ELM. J. Saf. Environ. 21(06), 2466–2472 (2021). https://doi.org/10.13637/j.issn.1009-6094.2020.0999
Qiang, M., Yachao, L., Zheng, L., Zhaojian, G.: Fault feature extraction of rolling bearings based on variational modal decomposition and Teager energy operator. Vibration and Shock 35(13), 134–139 (2016). https://doi.org/10.13465/j.cnki.jvs.2016.13.022
Heng, L., Hydrogen, Z., Xianrong, Q., Yuantao, S.: A bearing fault diagnosis method based on short-time Fourier transform and convolutional neural network. Vibr. Shock 37(19), 124–131 (2018). https://doi.org/10.13465/j.cnki.jvs.2018.19.020
Sun, X., Wang, M., Zhan, B., et al.: An intelligent diagnostic method for multisource coupling faults of complex mechanical systems. Shock and Vibration (2023)
Zheng, W., Lin, R.Q., Wang, J., Li, Z.J.: Power quality disturbance classification based on GAF and convolutional neural network. Power System Protect. Control 49(11), 97–104 (2021). https://doi.org/10.19783/j.cnki.pspc.200997
Yao, L., Mianjun, S., Chenbo, M.: A rolling bearing fault diagnosis method based on Gram’s angular field and CNN-RNN. Bearings (02), 61–67 (2022). https://doi.org/10.19533/j.issn1000-3762.2022.02.012
Han, B., Zhang, H., Sun, M., et al.: A new bearing fault diagnosis method based on capsule network and Markov transition field/Gramian angular field. Sensors 21(22), 7762 (2021)
Tan, M., Le, Q.: EfficientNet: rethinking model scaling for convolutional neural networks. In: International Conference on Machine Learning, pp. 6105–6114. PMLR (2019)
Atila, U., UcSar, M., Akyol, K., et al.: Plant leaf disease classification using EfficientNet deep learning model. Ecol. Inform. 61, 101182 (2021)
Li, W., Zhu, X., Gong, S.: Person re-identification by deep joint learning of multi-loss classification. arXiv preprint arXiv:1705.04724 (2017)
Weiss, K., Khoshgoftaar, T.M., Wang, D.D.: A survey of transfer learning. J. Big Data 3(1), 1–40 (2016)
Yu, G., Qingwen, G., Chuntao, W., et al.: Crop pest identification based on improved EfficientNet model. J. Agric. Eng., 038-001 (2022)
Shorten, C., Khoshgoftaar, T.M.: A survey on image data augmentation for deep learning. J. Big Data 6(1), 1–48 (2019)
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2024 ICST Institute for Computer Sciences, Social Informatics and Telecommunications Engineering
About this paper
Cite this paper
Dai, Y., Li, J., Ying, Y., Zhang, B., Shi, T., Zhao, H. (2024). A Lightweight Fault Diagnosis Model of Rolling Bearing Based on Gramian Angular Field and EfficientNet-B0. In: Li, J., Zhang, B., Ying, Y. (eds) 6GN for Future Wireless Networks. 6GN 2023. Lecture Notes of the Institute for Computer Sciences, Social Informatics and Telecommunications Engineering, vol 554. Springer, Cham. https://doi.org/10.1007/978-3-031-53404-1_16
Download citation
DOI: https://doi.org/10.1007/978-3-031-53404-1_16
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-031-53403-4
Online ISBN: 978-3-031-53404-1
eBook Packages: Computer ScienceComputer Science (R0)