Multi-resolution short-time Fourier transform providing deep features for 3D CNN to classify rolling bearing fault vibration signals

Abstract

The time-frequency domain features of vibration signals provide valuable information for deep learning-based rolling bearing fault diagnosis methods, where fault signal classification aiding in the identification of nominal fault types during diagnosis. The Short-Time Fourier Transform (STFT) is a widely used time-frequency transformation method, and its window length is the key parameter that determines the trade-off between time and frequency resolution. The primary motivation of this study is to address the limitation in traditional STFT-based 2D CNN methods: the inability to adapt the window length to different types of signals. To achieve accurate classification of bearing fault types, this study proposes a method based on three-dimensional convolutional neural networks (3D CNNs) to deeply explore the time-frequency domain information of one-dimensional vibration signals from faulty bearings. This method first applies STFT with multiple window sizes to perform multi-resolution time-frequency transformations on the time-domain vibration signals, yielding three-dimensional data. Subsequently, a classifier is trained based on the proposed 3D CNN. Experimental results on public datasets show that, without any sophisticated techniques, the proposed method achieves an average classification accuracy of 99.2% for six types of bearing faults using a relatively simple CNN structure. Compared to 1D CNN and 2D CNN methods that use fixed window sizes for STFT, the proposed method significantly enhances classification performance. Furthermore, it demonstrates robust classification results even on small-scaled bearing datasets.