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

Gear fault diagnosis method based on Transformer and convolutional neural network
Published:2024-03-26 author:YAN Huiyu, ZHANG Chao. Browse: 1001 Check PDF documents
Gear fault diagnosis method based on Transformer and 
convolutional neural network


YAN Huiyu1,2, ZHANG Chao1,2

(1.School of Mechanical Engineering, Inner Mongolia University of Science & Technology, Baotou 014017, China; 
2.Inner 
Mongolia Autonomous Region Key Laboratory of Intelligent Diagnosis and Control for Electromechanical System, Baotou 014010, China)


Abstract: Aiming at complex operating environment of partial gears which leading to the lack of sample data, a method for diagnosis transfer learning gear fault based on Transformer and convolutional neural network(CNN) was present. First, Gaussian filter was employed to preprocess the original vibration signal. It was able to smooth the signal and reduce interference from noisy signals. Then, the signal as an input signal to the Transformer was transformed into a patch sequence with position information. It enhanced the Transformer‘’s feature extraction capabilities, and it improved model‘’s diagnostic accuracy. Besides, the Transformer output sequence was input into a one-dimensional CNN to keep extracting fault information, and a residual block was added to the model to prevent network degradation. What‘’s more, the gear dataset collected in the laboratory and the gearbox dataset of Southeast University were divided into source and target domains, the model with source domain data was pretrained, and 100 samples of each type of gear were selected as the target domain. Finally, four sets of ten replicates were performed with different datasets as the source and target domains in order to test the accuracy of the model. The experimental results show that the accuracy of gear fault diagnosis with the method of Transformer-CNN transfer learning was more than 90%. Among them, the highest fault diagnosis accuracy can reach 100%. Transformer-CNN also compares the gear fault diagnosis accuracy of other convolutional neural networks, multi-scale convolutional neural networks and two-dimensional convolutional neural networks without Transformer, with an average accuracy of 99.64%, which is higher than that of the above networks. Therefore, the transfer learning method based on Transformer-CNN is able to diagnose gear faults under small samples.

Key words:  gear box; signal smoothing processing; transfer learning; Transformer; convolutional neural network(CNN); feature extraction ability


  • Chinese Core Periodicals
  • Chinese Sci-tech Core Periodicals
  • SA, INSPEC Indexed
  • CSA: T Indexed
  • UPD:Indexed


2010 Zhejiang Information Institute of Mechinery Industry

Technical Support:Hangzhou Bory science and technology

You are 1895221 visit this site