A bearing fault diagnosis method based on improved convolution neural network and transfer learning

Fan Jiang; Xi Shen; Feng Jiang; ZiShan Zhao; ShuMan Cheng

doi:10.1117/12.2625555

12 December 2021 A bearing fault diagnosis method based on improved convolution neural network and transfer learning

Fan Jiang, Xi Shen, Feng Jiang, ZiShan Zhao, ShuMan Cheng

Proceedings Volume 12127, International Conference on Intelligent Equipment and Special Robots (ICIESR 2021); 121270K (2021) https://doi.org/10.1117/12.2625555
Event: International Conference on Intelligent Equipment and Special Robots (ICIESR 2021), 2021, Qingdao, China

Abstract

Aiming at these issues of complex working conditions and small sample dataset in practical bearing fault diagnosis, a novel method based on improved convolution neural network and transfer learning is proposed. In this method, the global average pooling layer is firstly used to replace the traditional full connection layer as output layer, which can reduce the training parameters of whole network. Meanwhile, the original ReLu activation function is removed and then LeakyReLu activation function is introduced to make the network model be hard to produce Dead ReLu. The improved network is pretrained with the source domain data to learn the fault diagnosis feature. Here, the bottom network parameters are frozen, and the top network is also fine-tuned with a small amount of target domain data to obtain the transfer model. Finally, the target domain data is input to the transfer model for bearing fault diagnosis. The proposed method is verified by considering the scenario of small sample and cross working conditions with the bearing dataset of Case Western Reserve University. The results show that the average classification accuracy of the proposed method can reach 95.49%, and is higher than that of normal convolution neural network.

Citation Download Citation

Fan Jiang, Xi Shen, Feng Jiang, ZiShan Zhao, and ShuMan Cheng "A bearing fault diagnosis method based on improved convolution neural network and transfer learning", Proc. SPIE 12127, International Conference on Intelligent Equipment and Special Robots (ICIESR 2021), 121270K (12 December 2021); https://doi.org/10.1117/12.2625555

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available

Members: $17.00

Non-members: $21.00 ADD TO CART

PROCEEDINGS
7 PAGES

DOWNLOAD PAPER SAVE TO MY LIBRARY

GET CITATION

RIGHTS & PERMISSIONS

Get copyright permission Get copyright permission on Copyright Marketplace

KEYWORDS

Convolution

Neural networks

Data modeling

Mining

Signal processing

Convolutional neural networks

Failure analysis

Show All Keywords

Keywords/Phrases

Search In:

Publication Years