Time-Frequency Multi-Domain 1D Convolutional Neural Network with Channel-Spatial Attention for Noise-Robust Bearing Fault Diagnosis

Abstract

This paper proposes a noise-robust and accurate bearing fault diagnosis model based on time-frequency multi-domain 1D convolutional neural networks (CNNs) with attention modules. The proposed model, referred to as the TF-MDA model, is designed for an accurate bearing fault classification model based on vibration sensor signals that can be implemented at industry sites under a high-noise environment. Previous 1D CNN-based bearing diagnosis models are mostly based on either time domain vibration signals or frequency domain spectral signals. In contrast, our model has parallel 1D CNN modules that simultaneously extract features from both the time and frequency domains. These multi-domain features are then fused to capture comprehensive information on bearing fault signals. Additionally, physics-informed preprocessings are incorporated into the frequency-spectral signals to further improve the classification accuracy. Furthermore, a channel and spatial attention module is added to effectively enhance the noise-robustness by focusing more on the fault characteristic features. Experiments were conducted using public bearing datasets, and the results indicated that the proposed model outperformed similar diagnosis models on a range of noise levels ranging from −6 to 6 dB signal-to-noise ratio (SNR).