Gearbox Fault Diagnosis Based on ICEEMDAN-MPE-AWT and SE-ResNeXt50 Transfer Learning Model

Because the gearbox in transmission systems is prone to failure and the fault signal is not obvious, the fault end cannot be located. In this paper, a gearbox fault diagnosis method grounded on improved complete ensemble empirical mode decomposition with adaptive noise, a multiscale permutation entropy and adaptive wavelet thresholding (ICEEMDAN-MPE-AWT) denoising method and an SE-ResNeXt50 transfer learning model are proposed. Initially, the vibration signal is denoised by ICEEMDAN-MPE-AWT, the denoised vibration signal is then converted into a Gram angle field (GAF) diagram, and then the parameters are transferred by the fine-tuning transfer learning strategy. Finally, a GAF diagram is input into the model for training to achieve fault extraction and classification. In this paper, the open gear dataset of Southeast University is used for experimental research. The experimental results show that when using the ICEEMDAN-MPE-AWT and when the signal-to-noise ratio (SNR) of the experimental data is −4 dB, the average accuracy of the GASF+TSE-ResNeXt50 and the GASF+TSE-ResNeXt18 can reach 98.8% and 97.5%, respectively. When the SNR is 6 dB, the accuracy of the above two models reaches 100% and 99.3%, respectively. Moreover, when compared to alternative approaches, the noise reduction method in this paper can better remove noise interference so that the model can better extract fault features. Therefore, the method proposed in this article shows significant improvement in noise reduction and fault classification accuracy compared to other methods.