Fault Diagnosis of Rolling Bearing Based on Spectrum-Adaptive Convolution and Interactive Attention Mechanism

With the development of artificial intelligence technology, intelligent fault diagnosis methods based on deep learning have received extensive attention. Among them, convolutional neural network (CNN) has been widely applied in the fault diagnosis of rolling bearings due to its strong feature extraction ability. However, traditional CNN models still have deficiencies in the extraction of early weak fault features and the suppression of high noise. In response to these problems, this paper proposes a convolutional neural network (SAWCA-net) that integrates spectrum-guided dynamic variable-width convolutional kernels and dynamic interactive time-domain–channel attention mechanisms. In this model, the spectrum-adaptive wide convolution is introduced. Combined with the time-domain and frequency-domain statistical characteristics of the input signal, the receptive field of the convolution kernel is adaptively adjusted, and the sampling position is dynamically adjusted, thereby enhancing the model’s modeling ability for periodic weak faults in complex non-stationary vibration signals and improving its anti-noise performance. Meanwhile, the dynamic time–channel attention module was designed to achieve the collaborative modeling of the time-domain periodic structure and the feature dependency between channels, improve the feature utilization efficiency, and suppress redundant interference. The experimental results show that the fault diagnosis accuracy rates of SAWCA-Net on the bearing datasets of Case Western Reserve University (CWRU) and Xi’an Jiaotong University (XJTU-SY) reach 99.15% and 99.64%, respectively, which are superior to the comparison models and have strong generalization and robustness. The visualization results of t-distributed random neighbor embedding (t-SNE) further verified its good feature separability and classification ability.