Integrated Gradient-Based Continuous Wavelet Transform for Bearing Fault Diagnosis

Bearing fault diagnosis is important to ensure safe operation and reduce loss for most rotating machinery. In recent years, deep learning (DL) has been widely used for bearing fault diagnosis and has achieved excellent results. Continuous wavelet transform (CWT), which can convert original sensor data to time–frequency images, is often used to preprocess vibration data for the DL model. However, in time–frequency images, some frequency components may be important, and some may be unimportant for DL models for fault diagnosis. So, how to choose a frequency range of important frequency components is needed for CWT. In this paper, an Integrated Gradient-based continuous wavelet transform (IG-CWT) method is proposed to address this issue. Through IG-CWT, the important frequency components and the component frequency range can be detected and used for data preprocessing. To verify our method, experiments are conducted on four famous bearing datasets using 3 DL models, separately, and compared with CWT, and the results are compared with the original CWT. The comparisons show that the proposed IG-CWT can achieve higher fault diagnosis accuracy.