Evaluation of Time-Frequency Representations for Deep Learning-Based Rotating Machinery Fault Diagnosis

Delanyo Kwame Bensah Kulevome; Man Qiu; Feng Cao; Edward Opoku-Mensah

doi:10.46604/ijeti.2024.14774

Authors

Delanyo Kwame Bensah Kulevome Department of Computer Science and Information Engineering, Qilu Institute of Technology, Jinan, China https://orcid.org/0000-0001-5854-0788
Man Qiu Department of Intelligent Manufacturing and Control Engineering, Qilu Institute of Technology, Jinan, China
Feng Cao Department of Intelligent Manufacturing and Control Engineering, Qilu Institute of Technology, Jinan, China
Edward Opoku-Mensah Department of Information Technologies, HEC Montréal, Montreal, Canada

DOI:

https://doi.org/10.46604/ijeti.2024.14774

Keywords:

deep learning, time-frequency representation, fault diagnosis, rotating equipment

Abstract

This study evaluates and compares five time-frequency representation (TFR) methods for fault diagnosis in rotating machinery, aiming to ensure operational reliability and reduce unexpected downtime. The methods—short-time Fourier transform (STFT), continuous wavelet transform (CWT), modified S-transform (MS-transform), smoothed pseudo Wigner-Ville distribution (SPWVD), and Hilbert-Huang transform (HHT)—are investigated. Vibration signals from benchmark bearing and gearbox datasets are converted into two-dimensional TFR data and classified using a convolutional neural network (CNN). Results show that MS-transform achieves the highest accuracy (up to 99.87%) under ideal conditions. STFT and CWT demonstrate better robustness in noisy environments, maintaining over 99% accuracy at 15 dB signal-to-noise ratio (SNR). SPWVD is computationally intensive with moderate performance, while HHT performs poorly under noise. Renyi entropy, energy conservation, and training time are also used to assess TFR quality. These findings support selecting appropriate TFR methods for industrial fault diagnosis.

References

I. D. P. Karyatanti, A. Noersena, F. R. Purnomo, R. S. Zulkifli, and A. Wijayanto, “Analysis of Outer Race Bearing Damage by Calculation of Sound Signal Frequency Based on the FFT Method,” International Journal of Engineering and Technology Innovation, vol. 13, no. 1, pp. 28-39, 2023.

X. Zhang, L. Jiang, L. Wang, T. Zhang, and F. Zhang, “A Pruned-Optimized Weighted Graph Convolutional Network for Axial Flow Pump Fault Diagnosis with Hydrophone Signals,” Advanced Engineering Informatics, vol. 60, article no. 102365, 2024.

X. Zhang, J. Liu, X. Zhang, and Y. Lu, “Self-Supervised Graph Feature Enhancement and Scale Attention for Mechanical Signal Node-Level Representation and Diagnosis,” Advanced Engineering Informatics, vol. 65, part A, article no. 103197, 2025.

A. Raqib, H. Jahanger, M. Sumner, T. Cox, T. Klumpner, I. Agha, et al., “Comparison of Frequency Transform Methods for Condition Monitoring of a DC-Link Capacitor,” Proceedings of the 10th International Conference on Power Electronics, Machines and Drives, vol. 2020, no. 7, pp. 621-626, 2020.

D. K. B. Kulevome, H. Wang, and X. Wang, “Deep Neural Network Based Classification of Rolling Element Bearings and Health Degradation through Comprehensive Vibration Signal Analysis,” Journal of Systems Engineering and Electronics, vol. 33, no. 1, pp. 233-246, 2022.

G. Wu, G. Wang, X. Liu, and R. Duan, “Bearing Fault Diagnosis Method Based on STFT Image and AlexNet Network,” International Conference on the Efficiency and Performance Engineering Network, pp. 1056-1068, 2022.

G. Xin, Z. Li, L. Jia, Q. Zhong, H. Dong, N. Hamzaoui, et al., “Fault Diagnosis of Wheelset Bearings in High-Speed Trains Using Logarithmic Short-Time Fourier Transform and Modified Self-Calibrated Residual Network,” IEEE Transactions on Industrial Informatics, vol. 18, no. 10, pp. 7285-7295, 2022.

Q. Zhang and L. Deng, “An Intelligent Fault Diagnosis Method of Rolling Bearings Based on Short-Time Fourier Transform and Convolutional Neural Network,” Journal of Failure Analysis and Prevention, vol. 23, no. 2, pp. 795-811, 2023.

R. F. Ribeiro Junior, I. A. dos Santos Areias, M. M. Campos, C. E. Teixeira, L. E. B. da Silva, and G. F. Gomes, “Fault Detection and Diagnosis in Electric Motors Using Convolution Neural Network and Short-Time Fourier Transform,” Journal of Vibration Engineering & Technologies, vol. 10, no. 7, pp. 2531-2542, 2022.

D. K. B. Kulevome, H. Wang, and X. Wang, “Rolling Bearing Fault Diagnostics Based on Improved Data Augmentation and ConvNet,” Journal of Systems Engineering and Electronics, vol. 34, no. 4, pp. 1074-1084, 2023.

W. Fan, Q. Zhou, J. Li, and Z. Zhu, “A Wavelet-Based Statistical Approach for Monitoring and Diagnosis of Compound Faults with Application to Rolling Bearings,” IEEE Transactions on Automation Science and Engineering, vol. 15, no. 4, pp. 1563-1572, 2018.

X. Zhang, X. Zhang, J. Liu, B. Wu, and Y. Hu, “Graph Features Dynamic Fusion Learning Driven by Multi-Head Attention for Large Rotating Machinery Fault Diagnosis with Multi-Sensor Data,” Engineering Applications of Artificial Intelligence, vol. 125, article no. 106601, 2023.

H. K. R. Pillai, M. Nanappan, M. V. Prabhakaran, and S. P. Sathyabhama, “A Robust Technique for Detection, Diagnosis, and Localization of Switching Faults in Electric Drives Using Discrete Wavelet Transform,” International Journal of Engineering and Technology Innovation, vol. 13, no. 1, pp. 14-27, 2023.

M. Q. Tran, M. K. Liu, Q. V. Tran, and T. K. Nguyen, “Effective Fault Diagnosis Based on Wavelet and Convolutional Attention Neural Network for Induction Motors,” IEEE Transactions on Instrumentation and Measurement, vol. 71, article no. 3501613, 2022.

R. Yan, Z. Shang, H. Xu, J. Wen, Z. Zhao, X. Chen, et al., “Wavelet Transform for Rotary Machine Fault Diagnosis: 10 Years Revisited,” Mechanical Systems and Signal Processing, vol. 200, article no. 110545, 2023.

D. K. B. Kulevome, H. Wang, B. M. Cobbinah, E. S. Mawuli, and R. Kumar, “Effective Time-Series Data Augmentation with Analytic Wavelets for Bearing Fault Diagnosis,” Expert Systems with Applications, vol. 249, part A, article no. 123536, 2024.

J. Zheng, J. Wang, H. Wang, J. Ding, and C. Yi, “Diagnosis and Classification of Gear Composite Faults Based on S-Transform and Improved 2D Convolutional Neural Network,” International Journal of Dynamics and Control, vol. 12, no. 6, pp. 1659-1670, 2024.

Y. Peng and X. Ma, “A Fault Diagnosis Method of Rolling Bearings Based on Parameter Optimization and Adaptive Generalized S-Transform,” Machines, vol. 10, no. 3, article no. 207, 2022.

Y. Xu, L. Wang, A. Hu, and G. Yu, “Time-Extracting S-Transform Algorithm and Its Application in Rolling Bearing Fault Diagnosis,” Science China Technological Sciences, vol. 65, no. 4, pp. 932-942, 2022.

Z. Cui, G. Yu, and W. Tian, “S-Transform Based Time–Frequency Analysis Tool with Application to Detection of Bearing Fault,” Journal of Physics: Conference Series, vol. 2352, no. 1, article no. 012004, 2022.

H. Wang, Z. Fang, H. Wang, Y. Li, Y. Geng, L Chen, et al., “A Novel Time-Frequency Analysis Method for Fault Diagnosis Based on Generalized S-Transform and Synchroextracting Transform,” Measurement Science and Technology, vol. 35, no. 3, article no. 036101, 2023.

W. Liu, Y. Liu, S. Li, and Z. Zhai, “Demodulated Synchrosqueezing S-Transform and Its Application to Machine-Fault Diagnosis,” Measurement Science and Technology, vol. 34, no. 6, article no. 065004, 2023.

C. Yi, J. Qin, H. Xiao, and T. Zhou, “Second-Order Synchrosqueezing Modified S Transform for Wind Turbine Fault Diagnosis,” Applied Acoustics, vol. 189, article no. 108614, 2022.

X. Lv and H. Li, “Rolling Bearing Fault Diagnosis Based on GWVD and Convolutional Neural Network,” 19th International Conference on Advanced Intelligent Computing Technology and Applications, pp. 514-523, 2023.

H. Li, Q. Zhang, X. Qin, and Y. Sun, “K-SVD-Based WVD Enhancement Algorithm for Planetary Gearbox Fault Diagnosis under a CNN Framework,” Measurement Science and Technology, vol. 31, no. 2, article no. 025003, 2019.

D. Quan, F. Ren, X. Wang, M. Xing, N. Jin, and D. Zhang, “WVD-GAN: A Wigner-Ville Distribution Enhancement Method Based on Generative Adversarial Network,” IET Radar, Sonar & Navigation, vol. 18, no. 6, pp. 849-865, 2024.

Z. Li, S. Sun, Y. Wang, W. Dai, J. Qiu, and Z. Liu, “Time-Frequency Analysis of Non-stationary Signal Based on Sliding Mode Singular Spectrum Analysis and Wigner-Ville Distribution,” 3rd International Conference on Information Science and Education, pp. 218-222, 2022.

E. Elbouchikhi, V. Choqueuse, Y. Amirat, M. E. H. Benbouzid, and S. Turri, “An Efficient Hilbert–Huang Transform-Based Bearing Faults Detection in Induction Machines,” IEEE Transactions on Energy Conversion, vol. 32, no. 2, pp. 401-413, 2017.

G. Cheng, Y. L. Cheng, L. H. Shen, J. B. Qiu, and S. Zhang, “Gear Fault Identification Based on Hilbert–Huang Transform and SOM Neural Network,” Measurement, vol. 46, no. 3, pp. 1137-1146, 2013.

M. Zabin, H. J. Choi, J. Uddin, M. H. Furhad, and A. B. Ullah, “Industrial Fault Diagnosis Using Hilbert Transform and Texture Features,” IEEE International Conference on Big Data and Smart Computing (BigComp), pp. 121-128, 2021.