Gearbox Fault Diagnosis Method Based on Improved MobileNetV3 and Transfer Learning

Abstract

Under different working conditions of gearbox, the feature extraction of fault signals is difficult, and large difference in data distribution affects the fault diagnosis results. Based on the problems, the research proposes a method based on improved MobileNetV3 network and transfer learning (TL-Pro-MobilenetV3 network). Three time-frequency analysis methods are used to obtain time-frequency distribution. Among them, short time Fourier transform (STFT) combined with Pro-MobilenetV3 network takes the shortest time and has the highest accuracy. Furthermore, transfer learning is introduced into the model, and the optimal training parameters are selected training the network. Using the dataset from Southeast University, the TL-Pro-MobilenetV3 model is compared with four classical fault diagnosis models. The experimental results show the accuracy of the method proposed can reach 100% and the training time is the shortest in two working conditions, proving the proposed model has a good performance in generalization ability, recognition accuracy and training time