Gearbox fault detection based on pre-trained GoogleNet modal and transfer learning
In order to address the issues of cumbersome feature processing algorithms and the significant human factors in the diagnosis of gearbox,a fault diagnosis method based on a pre-training model and transfer learning(TL)was proposed.Firstly,the continuous wavelet transform(CWT)was used to convert discrete time series into two-dimensional wavelet scale images to construct a sample set.Then,the pre-trained model was structurally fine-tuned and parameter-tuned to meet the task requirements.The fine-tuned model was further trained with processed training samples until the desired accuracy was achieved and the corresponding model was saved and applied to fault classification tasks.Finally,the fine-tuned model was validated by using parallel gearbox data from key laboratory of control and optimization of Kunming University of Science and Technology and the planetary gearbox data of Southeast University.The research results demonstrate that comparing to traditional convolutional neural networks(CNN)and non-pretrained GoogleNet models,the proposed model achieves an average classification accuracy of 97.40%with limited training samples.Additionally,the proposed model exhibits the characteristics of faster convergence speed,lower dependence on computational power since upper layers of model is modified.The method of fine-tuning the upper level of the proposed model can personalize output according to the task classification,so that the proposed model can be applied in different scenarios.
万方数据
维普
