A comparative study of rolling bearing fault diagnosis based on tree models

Hanyu Zhang; Chengcheng Zhong; Zitong Zhang; Yanan Jiang

doi:10.1117/12.2673391

3 April 2023 A comparative study of rolling bearing fault diagnosis based on tree models

Hanyu Zhang, Chengcheng Zhong, Zitong Zhang, Yanan Jiang

Proceedings Volume 12599, Second International Conference on Digital Society and Intelligent Systems (DSInS 2022); 125992D (2023) https://doi.org/10.1117/12.2673391
Event: 2nd International Conference on Digital Society and Intelligent Systems (DSInS 2022), 2022, Chendgu, China

Abstract

We systematically carried out a comparative study of 12 kinds of tree-based models for the task of rolling bearing fault diagnosis, using the publicly available XJTU-SY bearing dataset as an example. The results show that the ensemble tree models including random forest (RF), extremely randomized trees (ETs), and deep learning tree model (multi-Grained Cascade Forest, i.e. gcForest) are high-precision and strong robust models suiting industrial application of this task, which have better-performing detecting accuracy and stability than conventional machine learning and single tree models (decision tree and extremely randomized tree). gcForest achieves 99.37% test accuracy using only 3% of the training samples, while RF and ETs also exceed 98%, which outperform eXtreme Gradient Boosting (XGB), Light Gradient Boosting Machine (LGBM), categorical boosting (CatBoost), and neural tree models, i.e. Neural Network with Random Forest (NNRF) and TabNet. RF and ETs are better suited for real-time industrial detection tasks in terms of time consumption. This study provides a scientific basis for the rational selection of rolling bearing fault diagnosis methods.

Citation Download Citation

Hanyu Zhang, Chengcheng Zhong, Zitong Zhang, and Yanan Jiang "A comparative study of rolling bearing fault diagnosis based on tree models", Proc. SPIE 12599, Second International Conference on Digital Society and Intelligent Systems (DSInS 2022), 125992D (3 April 2023); https://doi.org/10.1117/12.2673391

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