Fault diagnosis of rolling element bearing using Naïve Bayes classifier

Xiao Jian Yi1, Yue Feng Chen2, Peng Hou3

1Department of Overall Technology, China North Vehicle Research Institute, Beijing, China

2The Fourth Research Laboratory, Beijing Special Vehicle Institute, Beijing, China

3School of Mechatronical Engineering, Beijing Institute of Technology, Beijing, China

3Corresponding author

E-mail: 1yixiaojianbit@sina.cn, 2yuefengch@163.com, 3apolloar@163.com

Received 11 September 2017; accepted 18 September 2017

DOI https://doi.org/10.21595/vp.2017.19153


Abstract. The development of machine learning brings a new way for diagnosing the fault of rolling element bearings. However, the method in machine learning with high accuracy often has the poor ability of generalization due to the overuse of feature engineering. To address this challenge, Naïve Bayes classifier is applied in this paper. As the one of the cluster of Bayes classifiers, its ability of classification is very outstanding. In this paper, the method is provided with a detailed description for why and how to diagnose the fault of bearing. Finally, an evaluation of the performance of Naïve Bayes classifier is presented with real world data. The evaluation indicates that Naïve Bayes classifier can achieve a high level of accuracy without any feature engineering.

Keywords: Naïve Bayes classifier, machine learning, fault diagnosis, rolling element bearing.


[1]        Liu T., Chen J., Dong G. Singular spectrum analysis and continuous hidden Markov model for rolling element bearing fault diagnosis. Journal of Vibration and Control, Vol. 21, 8, p. 1506‑1521.

[2]        Weihua L. I. A firefly neural network and its application in bearing fault diagnosis. Journal of Mechanical Engineering, Vol. 51, Issue 7, 2015, https://doi.org/10.3901/JME.2015.07.099.

[3]        Wu S. D., Wu C. W., Wu P. H., et al. Bearing fault diagnosis based on multiscale permutation entropy and support vector machine. Third International Conference on Mechanic Automation and Control Engineering, IEEE Computer Society, 2012, p. 2650‑2654.

[4]        Guo X., Chen L., Shen C. Hierarchical adaptive deep convolution neural network and its application to bearing fault diagnosis. Measurement, Vol. 93, 2016, p. 490‑502.

[5]        Mao W., He J., Li Y., et al. Bearing fault diagnosis with auto-encoder extreme learning machine: A comparative study. Proceedings of the Institution of Mechanical Engineers, Part C: Journal of Mechanical Engineering Science, Vol. 231, 8, p. 1560‑1578.

[6]        Lin J., Qu Liangsheng Feature extraction based on Morlet wavelet and its application for mechanical fault diagnosis. Journal of Sound and Vibration, Vol. 234, Issue 1, 2000, p. 135‑148.

[7]        Mallat S. A Wavelet Tour of Signal Processing. Elsevier, 1999, p. 83‑85.

[8]        Venkatasubramanian V., Rengaswamy R., Yin K., et al. A review of process fault detection and diagnosis: Part I: Quantitative model-based methods. Computers and Chemical Engineering, Vol. 27, Issue 3, 2003, p. 293‑311.

[9]        Muro H., Tsushima N., Nunome K. Failure analysis of rolling bearings by X-ray measurement of residual stress. Wear, Vol. 25, Issue 3, 1973, p. 345‑356.

[10]     Robert C. Machine learning, a probabilistic perspective. Mathematics Education Library, Vol. 58, Issue 8, 2012, p. 27‑71.

[11]     Yang J., Zhang Y., Zhu Y. Intelligent fault diagnosis of rolling element bearing based on SVMs and fractal dimension. Mechanical Systems and Signal Processing, Vol. 21, Issue 5, 2007, p. 2012‑2024.

[12]     Xing Z., Pei J., Keogh E. A brief survey on sequence classification. ACM SIGKDD Explorations Newsletter, Vol. 12, Issue 1, 2010, p. 40‑48.

[13]     Zhou P. Y., Chan K. C. C. A feature extraction method for multivariate time series classification using temporal patterns. Pacific-Asia Conference on Knowledge Discovery and Data Mining. Springer International Publishing, 2015, p. 409‑421.

[14]     Bishop C. M. Pattern Recognition and Machine Learning (Information Science and Statistics). Springer-Verlag, New York, 2006.

[15]     Loparo K. A. Bearings Vibration Data Set. Case Western Reserve University, hhttp://www.eecs.cwru.edu/laboratory/bearing/download.htmi.

[16]     Lei Y., He Z., Zi Y. A new approach to intelligent fault diagnosis of rotating machinery. Expert Systems with Applications, Vol. 35, Issue 4, 2008, p. 1593‑1600.

[17]     Hu Q., He Z., Zhang Z., et al. Fault diagnosis of rotating machinery based on improved wavelet package transform and SVMs ensemble. Mechanical Systems and Signal Processing, Vol. 21, Issue 2, 2007, p. 688‑705.

[18]     Huang Y., Liu C., Zha X. F., et al. Research Article: A lean model for performance assessment of machinery using second generation wavelet packet transform and Fisher criterion. Expert Systems with Applications, Vol. 37, Issue 5, 2010, p. 3815‑3822.

[19]     Wan S., Tong H., Dong B. Bearing fault diagnosis using wavelet packet transform and least square support vector machines. Journal of Vibration Measurement and Diagnosis, Vol. 30, Issue 2, 2010, p. 149‑152.

Cite this article

Yi Xiao Jian, Chen Yue Feng, Hou Peng Fault diagnosis of rolling element bearing using Naïve Bayes classifier. Vibroengineering PROCEDIA, Vol. 14, 2017, p. 64‑69.


JVE International Ltd. Vibroengineering PROCEDIA. Oct 2017, Vol. 14. ISSN 2345-0533