Е-Публикации
Технически университет - София

Abstract: This paper presents a comparative analysis of bearing and rotating component fault classification based on different time–frequency representations using vision transformer (ViT). Four different time–frequency transformation techniques—short-time Fourier transform (STFT), continuous wavelet transform (CWT), Hilbert–Huang transform (HHT), and Wigner–Ville distribution (WVD)—were applied to convert the signals into 2D images. A pretrained ViT-Base architecture was fine-tuned on the resulting images for classification tasks. The model was evaluated on two separate scenarios: (i) eight-class rotating component fault classification and (ii) four-class bearing fault classification. Importantly, in each task, the samples were collected under varying conditions of the other component (i.e., different rotating conditions in bearing classification and vice versa). This design allowed for an independent assessment of the model’s ability to generalize across fault domains. The experimental results demonstrate that the ViT-based approach achieves high classification performance across various time–frequency representations, highlighting its potential for mechanical fault diagnosis in rotating machinery. Notably, the model achieved higher accuracy in bearing fault classification compared to rotating component faults, suggesting higher sensitivity to bearing-related anomalies.

References

1. Zhang Q. Deng L. An Intelligent Fault Diagnosis Method of Rolling Bearings Based on Short-Time Fourier Transform and Convolutional Neural Network J. Fail. Anal. Prev. 2023 23 795 811 10.1007/s11668-023-01616-9
2. Zhong D. Guo W. He D. An Intelligent Fault Diagnosis Method based on STFT and Convolutional Neural Network for Bearings Under Variable Working Conditions Proceedings of the 2019 Prognostics and System Health Management Conference (PHM-Qingdao) Qingdao, China 25–27 October 2019 1 6 10.1109/PHM-Qingdao46334.2019.8943026
3. Peng B. Xia H. Lv X. Annor-Nyarko M. Zhu S. Liu Y. Zhang J. An intelligent fault diagnosis method for rotating machinery based on data fusion and deep residual neural network Appl. Intell. 2022 52 3051 3065 10.1007/s10489-021-02555-4
4. He M. He D. Deep learning based approach for bearing fault diagnosis IEEE Trans. Ind. Appl. 2017 53 3057 3065 10.1109/TIA.2017.2661250
5. Tao H. Wang P. Chen Y. Stojanovic V. Yang H. An unsupervised fault diagnosis method for rolling bearing using STFT and generative neural networks J. Frankl. Inst. 2020 357 7286 7307 10.1016/j.jfranklin.2020.04.024
6. Zhang Y. Xing K. Bai R. Sun D. Meng Z. An enhanced convolutional neural network for bearing fault diagnosis based on time–frequency image Measurement 2020 157 107667 10.1016/j.measurement.2020.107667
7. Liu H. Li L. Ma J. Rolling bearing fault diagnosis based on STFT-deep learning and sound signals Shock Vib. 2016 2016 6127479 10.1155/2016/6127479
8. Cheng Y. Lin M. Wu J. Zhu H. Shao X. Intelligent fault diagnosis of rotating machinery based on continuous wavelet transform-local binary convolutional neural network Knowl.-Based Syst. 2021 216 106796 10.1016/j.knosys.2021.106796
9. Xu Y. Li Z. Wang S. Li W. Sarkodie-Gyan T. Feng S. A hybrid deep-learning model for fault diagnosis of rolling bearings Measurement 2021 169 108502 10.1016/j.measurement.2020.108502
10. Mian T. Choudhary A. Fatima S. Vibration and infrared thermography based multiple fault diagnosis of bearing using deep learning Nondestruct. Test. Eval. 2023 38 275 296 10.1080/10589759.2022.2118747
11. Zhao H. Liu J. Chen H. Chen J. Li Y. Xu J. Deng W. Intelligent diagnosis using continuous wavelet transform and gauss convolutional deep belief network IEEE Trans. Reliab. 2022 72 692 702 10.1109/TR.2022.3180273
12. Ahsan M. Hassan M.W. Rodriguez J. Abdelrahem M. Enhanced Fault Diagnosis in Rotating Machinery Using a Hybrid CWT-LeNet-5-LSTM Model: Performance Across Various Load Conditions IEEE Access 2024 13 1026 1045 10.1109/ACCESS.2024.3522948
13. Yu S. Li Z. Gu J. Wang R. Liu X. Li L. Guo F. Ren Y. CWMS-GAN: A small-sample bearing fault diagnosis method based on continuous wavelet transform and multi-size kernel attention mechanism PLoS ONE 2025 20 e0319202 10.1371/journal.pone.0319202
14. Liu H. Wang X. Lu C. Rolling Bearing Fault Diagnosis under Variable Conditions Using Hilbert-Huang Transform and Singular Value Decomposition Math. Probl. Eng. 2014 2014 765621 10.1155/2014/765621
15. Suthar V. Vakharia V. Patel V.K. Shah M. Detection of compound faults in ball bearings using multiscale-SinGAN, heat transfer search optimization, and extreme learning machine Machines 2022 11 29 10.3390/machines11010029
16. Lin S.L. Intelligent fault diagnosis and forecast of time-varying bearing based on deep learning VMD-DenseNet Sensors 2021 21 7467 10.3390/s21227467
17. Wei P. Liu M. Wang X. Few-shot bearing fault diagnosis using GAVMD–PWVD time–frequency image based on meta-transfer learning J. Braz. Soc. Mech. Sci. Eng. 2023 45 277 10.1007/s40430-023-04202-0
18. Hua L. Qiang Y. Gu J. Chen L. Zhang X. Zhu H. Mechanical fault diagnosis using color image recognition of vibration spectrogram based on quaternion invariable moment Math. Probl. Eng. 2015 2015 702760 10.1155/2015/702760
19. Li X. Bi F. Zhang L. Lin J. Bi X. Yang X. Rotating machinery faults detection method based on deep echo state network Appl. Soft Comput. 2022 127 109335 10.1016/j.asoc.2022.109335
20. Mo C. Huang K. Ji H. Li W. Xu K. Efficient intelligent fault diagnosis method and graphical user interface development based on fusion of convolutional networks and vision transformers characteristics Sci. Rep. 2025 15 7110 10.1038/s41598-025-88668-z 40016236
21. Ren S. Lou X. Rolling Bearing Fault Diagnosis Method Based on SWT and Improved Vision Transformer Sensors 2025 25 2090 10.3390/s25072090 40218603
22. Tang X. Xu Z. Wang Z. A Novel Fault Diagnosis Method of Rolling Bearing Based on Integrated Vision Transformer Model Sensors 2022 22 3878 10.3390/s22103878
23. Zhang S. Zhou J. Ma X. Pirttikangas S. Yang C. TSViT: A Time Series Vision Transformer for Fault Diagnosis of Rotating Machinery Appl. Sci. 2024 14 10781 10.3390/app142310781
24. He Q. Li S. Bai Q. Zhang A. Yang J. Shen M. A Siamese Vision Transformer for Bearings Fault Diagnosis Micromachines 2022 13 1656 10.3390/mi13101656
25. Allen J. Short term spectral analysis, synthesis, and modification by discrete Fourier transform IEEE Trans. Acoust. Speech Signal Process. 1977 25 235 238 10.1109/TASSP.1977.1162950
26. Gabor D. Theory of communication. Part 1: The analysis of information J. Inst. Electr. Eng.-Part III Radio Commun. Eng. 1946 93 429 441 10.1049/ji-3-2.1946.0074
27. Ville J. Theorie et Applications de la Notion de Signal Analytique Cables Transm. 1948 2A 61 74
28. “Apparatus & Procedures | Case School of Engineering” Available online: https://engineering.case.edu/bearingdatacenter/apparatus-and-procedures (accessed on 1 July 2025)
29. Bechhoefer E. A Quick Introduction to Bearing Envelope Analysis, MFPT Data Available online: www.mfpt.org/fault-data-sets (accessed on 1 July 2025)
30. Lessmeier C. Kimotho J.K. Zimmer D. Sextro W. Condition monitoring of bearing damage in electromechanical drive systems by using motor current signals of electric motors: A benchmark data set for data-driven classification PHM Soc. Eur. Conf. 2016 3 1 10.36001/phme.2016.v3i1.1577
31. Graps A. An introduction to wavelets IEEE Comput. Sci. Eng. 1995 2 50 61 10.1109/99.388960
32. Polikar R. Mastorakis N. The story of wavelets inPhysics and modern topics in mechanical and electrical engineering World Sci. Eng. Soc. Press 1999 192 197
33. Huang N.E. Shen Z. Long S.R. Wu M.C. Shih H.H. Zheng Q. Yen N.-C. Tungand C.C. Liu H.H. The empirical mode decomposition and the Hilbert spectrum for nonlinear and non-stationary time series analysis Proc. R. Soc. Lond. Ser. A Math. Phys. Eng. Sci. 1998 454 903 995 10.1098/rspa.1998.0193
34. Huang H. Baddour N. Bearing vibration data collected under time-varying rotational speed conditions Data Brief 2018 21 1745 1749 10.1016/j.dib.2018.11.019 30505910
35. Boashash B. Time-Frequency Signal Analysis and Processing: A Comprehensive Reference Academic Press Cambridge, MA, USA 2015
36. Lee S. Kim T. Kim T. Multi-domain vibration dataset with various bearing types under compound machine fault scenarios: Subset 1 (deep groove ball bearing) Mendeley Data 2024 57 110640 10.17632/53vtnjy6c6.1
37. Lee S. Kim T. Kim T. Multi-domain vibration dataset with various bearing types under compound machine fault scenarios Data Brief 2024 57 110940 10.1016/j.dib.2024.110940
38. SciPy Documentation—Scipy.signal.spectrogram Available online: https://docs.scipy.org/doc/scipy/reference/generated/scipy.signal.spectrogram.html (accessed on 12 August 2025)
39. Jabloun M. Empirical mode decomposition revisited using ordinal pattern concepts Proceedings of the 2022 30th European Signal Processing Conference (EUSIPCO) Belgrade, Serbia 29 August–2 September 2022 2186 2190
40. Tsai C.W. Hsiao Y.R. Lin M.L. Hsu Y. Development of a noise-assisted multivariate empirical mode decomposition framework for characterizing PM 2.5 air pollution in Taiwan and its relation to hydro-meteorological factors Environ. Int. 2020 139 105669 10.1016/j.envint.2020.105669
41. Stallone A. Cicone A. Materassi M. New insights and best practices for the successful use of empirical mode decomposition, iterative filtering and derived algorithms Sci. Rep. 2020 10 15161 10.1038/s41598-020-72193-2
42. Yu L. Bearing fault diagnosis using time-frequency synchrosqueezing transform Proceedings of the 2020 Chinese Automation Congress (CAC) Shanghai, China 6–8 November 2020 4260 4264
43. Wang Z. Xu Z. Cai C. Wang X. Xu J. Shi K. Zhong X. Liao Z. Li Q. Rolling bearing fault diagnosis method using time-frequency information integration and multi-scale TransFusion network Knowl.-Based Syst. 2024 284 111344 10.1016/j.knosys.2023.111344
44. Verstraete D. Ferrada A. Droguett E.L. Meruane V. Modarres M. Deep learning enabled fault diagnosis using time-frequency image analysis of rolling element bearings Shock Vib. 2017 2017 5067651 10.1155/2017/5067651

Issue

Copyright MDPI