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

Bearing fault diagnosis algorithm based on CNN and CatBoost
Published:2023-07-18 author:LU Xi-yao, ZHANG Cheng-bin, GAO Jun, et al. Browse: 1170 Check PDF documents
Bearing fault diagnosis algorithm based on CNN and CatBoost


LU Xi-yao1, 2, ZHANG Cheng-bin1, GAO Jun1, XU Yan-ping1, SHAO Xing1

(1.College of Information Engineering, Yancheng Institute of Technology, Yancheng 224051, China; 
2.College of Mechanical Engineering, Yancheng Institute of Technology, Yancheng 224051, China)


Abstract:  When using general diagnostic algorithms for fault ranking of rolling bearings, feature extraction of the data was required, and there were defects in feature extraction in terms of large amount of data and limited manual extraction and selection. Aiming at this problem, a hybrid classification model based on convolutional neural network (CNN) and CatBoost was proposed. Firstly, the pre-processed data extracted by CNN was fed into this model with the features extracted as the input quantity to extract the model parameters output after training. Then, the rolling bearing dataset was analyzed using CatBoost method to further investigate the effect of different learning models on the classification accuracy under the same dataset. Finally, by reducing the risk of overfitting and applying four correlation coefficient indicators, a comparative experiment was conducted to study the classification effect of CNN-CatBoost hybrid classification model on rolling bearing data. The results indicate that the average accuracy of the proposed method is over 98%, which validates the effectiveness of the proposed method; a small number of data training samples can produce better classification results for bearing fault data, and has better performance in the Case Western Reserve University(CWRU) public bearing dataset compared with a single deep learning model and some typical machine learning models.

Key words:  convolutional neural network (CNN); CatBoost algorithm; fault feature extraction; fault classification accuracy; deep learning model; training time
Advanced Search
Online Office System
  • 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