Founded in 1971 >
Chinese Sci-tech Core Periodicals >
British Science Abstracts (SA, INSPEC) Indexed Journals >
United States, Cambridge Scientific Abstract: Technology (CSA: T) Indexed Journals >
United States, Ulrich's Periodicals Directory(UPD)Indexed Journals >
United States, Cambridge Scientific Abstract: Natural Science (CSA: NS) Indexed Journals >
Poland ,Index of Copernicus(IC) Indexed Journals >
International Standard Serial Number:
ISSN 1001-4551
Sponsor:
Zhejiang University;
Zhejiang Machinery and Electrical Group
Edited by:
Editorial of Journal of Mechanical & Electrical Engineering
Chief Editor:
ZHAO Qun
Vice Chief Editor:
TANG ren-zhong,
LUO Xiang-yang
Tel:
86-571-87041360,87239525
Fax:
86-571-87239571
Add:
No.9 Gaoguannong,Daxue Road,Hangzhou,China
P.C:
310009
E-mail:
meem_contribute@163.com
Abstract: 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 nonpretrained 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.
Key words: speed transmission; pretrained networks; transfer learning(TL); continuous wavelet transform(CWT); scalogram; convolutional neural network(CNN)