《机电工程》杂志社科技期刊核心期刊论文投稿机电一体化机械、仪表电工自动化、计算机浙江大学主办

Introduction

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: 156 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

Advanced Search

Online Office System

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