Mechanical Engineering Science

Application of Xgboost Feature Extraction in Fault Diagnosis of Rolling Bearing

WANGXingang, WANGChao


Due to the difficulty that excessive feature dimension in fault diagnosis of rolling bearing will lead to the decrease of classification accuracy, a fault diagnosis method based on Xgboost algorithm feature extraction is proposed. When the Xgboost algorithm classifies features, it generates an order of importance of the input features. The time domain features were extracted from the vibration signal of the rolling bearing, the time-frequency features were formed by the singular value of the modal components that were decomposed by the variational mode decomposition. Firstly, the extracted time domain and time-frequency domain features were input into the support vector machine respectively to observe the fault diagnosis accuracy. Then, Xgboost algorithm was used to rank the importance of features and got the accuracy of fault diagnosis. Finally, important features were extracted and the extracted features were input into the support vector machine to observe the fault diagnosis accuracy. The result shows that the fault diagnosis accuracy of rolling bearing is improved after important feature extraction in time domain and time-frequency domain by Xgboost.


fault diagnosis;rolling bearing;xgboost;feature extraction;support vector machine

Full Text:



WANG Xingang, YAN Mingming. Research on tool change time and dynamic reliability of machining process based on sensitivity analysis. Journal of Ordance Equipment, Engineering, 2017, 38(1), pp.1-6. (in chinese)

ZHANG Yu, CHEN Jun, WANG Xiaofeng, et al. Application of Xgboost to fault diagnosis of rolling bearings. Noise and Vibration Control, 2017, 37(4), pp.166-170. (in chinese)

WANG Lei, LIU Zhiwen, MIAO Qiang, et al. Time-frequency analysis based on ensemble local mean decomposition and fast kurtogram for Rotating machinery fault diagnosis. Mechanical Systems and Signal Processing, 2018, 103, pp.60-75.

Dragomiretskiy K, Zosso D. Variational mode decomposition. IEEE. Transactions on Signal Processing, 2014, 62(3), pp.531-544.

TANG Guiji, WANG Xiaolong. Parameter optimized variational mode decomposition method with application to incipient fault diagnosis. Journal of Xi’an Jiaotong University. 2015, 49(5), pp.73-81. (in chinese)

WANG Xianbo, YANG Zhixin, YAN Xiaoan. Novel particle swarm optimization based variational mode decomposition method for the fault diagnosis of complex rotating machinery. Energy, 2018, 23(1), pp.68-79.

LIU Jianchang, QUAN He, YU Xia, et al. Rolling bearing fault diagnosis based on parameter optimization VMD and sample entropy. Acta Automatica Sinica (in press). (in chinese)

PAN Jun, ZI Yanyang, CHEN Jinglong, et al. Lifting Net: A novel deep learning network with layer wise feature learning from noisy mechanical data for fault classification. IEEE. Transactions on Industrial Electronics, 2018, 65(6), pp.4973-4982.

FU Yunxiao, JIA Limin, QIN Yong, et al. Rolling bearing fault diagnosis method based on LMD-CM-PCA. Journal of Vibration, Measurement &Diagnosis, 2017, 37(2), pp.249-255. (in chinese)

ZHANG Xining, ZHANG Wenwen, ZHOU Rongtong, et al. Bearing fault diagnosis method based on multiple dimensional scaling and random forest. Journal of Xi’an Jiaotong University, 2019, 53(8), pp.1-7. (in chinese)

Vapnik V. The nature of statistical learning theory, 2nd ed. Springer:New York, America; 1995; pp.163-167.

Zidi S, Moulahi T, Alaya B. Fault detection in wireless sensor networks through SVM classifier. IEEE. Sensors Journal, 2018, 18(1), pp.211−244.

ZHAN Chunfu, WANG Song, WU Yadong, et al. Diabetes risk prediction based on GA_Xgboost model. Computer Engineering (in press).(in chinese)

CHEN Tianqi, Guestrin C. Xgboost: A Scalable Tree Boosting System. ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, ACM:California, America, 2016, pp.785-794.

Bearing Data Center Website. Available online: (accessed on 10.5.2019)

LI Jimeng, YAO Xifeng, WANG Hui, et al. Periodic impulses extraction based on improved adaptive VMD and sparse code shrinkage denoising and its application in rotating machinery fault Diagnosis. Mechanical Systems and Signal Processing. 2019, 126, pp.568-589.

MA Hongbin, TONG Qingbin, ZHANG Yanan. Applications of optimization parameters VMD to fault diagnosis of rolling bearings. China Mechanical Engineering, 2018, 29(4), pp.390-397. (in chinese)

LIU Ruonan, YANG Boyuan, Zio E, et al. Artificial intelligence for fault diagnosis of rotating machinery:A Review. Mechanical Systems and Signal Processing. 2018, 108, pp.33-47.



  • There are currently no refbacks.

Creative Commons License
This work is licensed under a Creative Commons Attribution-NonCommercial-NoDerivatives 4.0 International License.