Abstract
Traditional rolling bearing fault diagnosis methods struggle to adaptively extract features under complex industrial environments, and obtaining large and rich fault data under real operating conditions is difficult and expensive. Aiming at these issues, a bearing fault diagnosis method based on Self-Attention Generative Adversarial Networks (SAGAN) and Improved Deep Residual Networks (IResNet) was proposed (SAGAN_IResNet). Firstly, the original vibration signals are transformed into two-dimensional time-frequency images using continuous wavelet transform, providing both time domain and frequency domain information. Secondly, SAGAN is used to generate new samples similar to the original sample distribution, thereby expanding the data. Furthermore, a bearing fault diagnosis model is constructed using an improved residual network that incorporates the Multi-head Self-Attention (MHA) to adaptively obtain the global feature information, alleviate the problem of gradient dispersion and network degradation, and enhance the model's diagnostic performance in the presence of strong noise and variable load conditions. Experimental verification is conducted using bearing datasets from Case Western Reserve University, Southeast University and Jiangnan University. The results show that the method proposed in this paper has strong bearing fault diagnosis performance under the condition of few samples, strong noise and variable load.