Simulation Data Driven Time-Frequency Fusion 1D Convolutional Neural Network with Multiscale Attention for Bearing Fault Diagnosis

Abstract This paper presents a bearing fault diagnosis model based on a simulated data-driven time-domain and frequency-domain feature fusion using a one-dimensional convolutional neural network (1D CNN). The proposed simulation data driven time and frequency feature fusion
1D CNN with Multi-Scale attention (TFF-MSA) model aims to accurately classify bearing faults in strong noise environments by training with simulated data from bearing dynamic models. Traditional 1D CNN-based bearing fault diagnosis models often focused on either time-domain signals or frequency-domain spectra. The proposed model incorporates a parallel 1D CNN structure that simultaneously extracts features from both the time and frequency domains. These multi-domain features are then fused to capture comprehensive information from the bearing vibration signals. Specifically, to enhance the extraction of time-domain features, we introduce the Pyramid Attention Module (PAM) to mine multi-scale features. Meanwhile, in the frequency-domain feature extraction, a specific fault feature frequency weighting method, allowing the proposed model to extract specific fault-related frequencies from the signals, is also proposed. Experimental results demonstrate that the proposed model which is only driven by simulated signals from dynamic model achieves an average diagnostic accuracy of 94%.