A Lightweight CNN Using Smoothed Pseudo-Wigner–Ville Distribution for Bearing Fault Severity Classification
摘要
Accurate and computationally efficient prediction of bearing fault severity is essential for predictive maintenance and the reliability of rotating machinery. This study aims to present a lightweight diagnostic framework that enhances the extraction and discrimination of transient fault signatures.
MethodsThe framework integrates the Smoothed Pseudo Wigner–Ville Distribution (SPWVD) for high-resolution time–frequency preprocessing with a lightweight Convolutional Neural Network (CNN, ~0.42M parameters, 0.1 GFLOPs). Comparative experiments evaluate SPWVD against the Short-Time Fourier Transform (STFT) and the Continuous Wavelet Transform (CWT) for extracting transient fault signatures.
ResultsSPWVD outperforms STFT and CWT, delivering higher resolution and improved cross-term suppression. The proposed SPWVD-CNN model achieves 99.3% accuracy for 14-class fault severity on the Case Western Reserve University (CWRU) dataset and generalizes robustly with 94.4% accuracy on the Intelligent Maintenance System (IMS) dataset. The confusion matrix and per-class precision analyses confirm high reliability across fault types and severity levels.
ConclusionBenchmarking against conventional classifiers and deep learning models demonstrates favourable accuracy, indicating that the SPWVD-CNN framework offers an efficient and reliable solution for bearing fault severity prediction.