Rolling Bearing Fault Diagnosis Based on Convolutional Neural Network and Support Vector Machine

Rolling bearings are one of the essential components in rotating machinery. Efficient bearing fault diagnosis is necessary to ensure the regular operation of the mechanical system. Traditional fault diagnosis methods usually rely on a complex artificial feature extraction process, which requires a lot of human expertise. Emerging deep learning methods can reduce the dependence of the feature extraction process on manual intervention effectively. However, its training requires a large number of fault signals, which is difficult to obtain in actual engineering. In this paper, a rolling bearing fault diagnosis method based on Convolutional Neural Network and Support Vector Machine is proposed to solve the above problems. Firstly, the Continuous Wavelet Transform is used to convert one-dimensional original vibration signals into two-dimensional time-frequency images. Secondly, the obtained time-frequency images are input for training the constructed model. Finally, the diagnosis of the fault location and severity is completed. The method is verified on the CWRU data set and the MFPT data set. The results demonstrate that the proposed method achieves higher diagnostic accuracy and stability than other advanced techniques.

[1]  Jun Wu,et al.  Intelligent fault diagnosis of rotating machinery via wavelet transform, generative adversarial nets and convolutional neural network , 2020 .

[2]  Tao Zhang,et al.  Deep Model Based Domain Adaptation for Fault Diagnosis , 2017, IEEE Transactions on Industrial Electronics.

[3]  Shuhui Wang,et al.  Convolutional neural network-based hidden Markov models for rolling element bearing fault identification , 2017, Knowl. Based Syst..

[4]  Mohammad Modarres,et al.  Deep Learning Enabled Fault Diagnosis Using Time-Frequency Image Analysis of Rolling Element Bearings , 2017 .

[5]  Minqiang Xu,et al.  A fault diagnosis scheme for rotating machinery using hierarchical symbolic analysis and convolutional neural network. , 2019, ISA transactions.

[6]  Konstantinos Gryllias,et al.  A deep learning method for bearing fault diagnosis based on Cyclic Spectral Coherence and Convolutional Neural Networks , 2020 .

[7]  V. Sugumaran,et al.  A comparative study of Naïve Bayes classifier and Bayes net classifier for fault diagnosis of monoblock centrifugal pump using wavelet analysis , 2012, Appl. Soft Comput..

[8]  Hee-Jun Kang,et al.  Rolling element bearing fault diagnosis using convolutional neural network and vibration image , 2019, Cognitive Systems Research.

[9]  Buket D. Barkana,et al.  Utilizing CNNs and transfer learning of pre-trained models for age range classification from unconstrained face images , 2019, Image Vis. Comput..

[10]  Haidong Shao,et al.  Rolling bearing fault detection using continuous deep belief network with locally linear embedding , 2018, Comput. Ind..

[11]  Robert X. Gao,et al.  Wavelets for fault diagnosis of rotary machines: A review with applications , 2014, Signal Process..

[12]  Xiaojiang Du,et al.  A Novel Deep Learning Strategy for Classifying Different Attack Patterns for Deep Brain Implants , 2019, IEEE Access.

[13]  Meiying Qiao,et al.  Deep Convolutional and LSTM Recurrent Neural Networks for Rolling Bearing Fault Diagnosis Under Strong Noises and Variable Loads , 2020, IEEE Access.

[14]  Haidong Shao,et al.  Rolling bearing health prognosis using a modified health index based hierarchical gated recurrent unit network , 2019, Mechanism and Machine Theory.

[15]  Minping Jia,et al.  A novel optimized SVM classification algorithm with multi-domain feature and its application to fault diagnosis of rolling bearing , 2018, Neurocomputing.

[16]  Yitao Liang,et al.  A novel bearing fault diagnosis model integrated permutation entropy, ensemble empirical mode decomposition and optimized SVM , 2015 .

[17]  Myeongsu Kang,et al.  A Hybrid Feature Selection Scheme for Reducing Diagnostic Performance Deterioration Caused by Outliers in Data-Driven Diagnostics , 2016, IEEE Transactions on Industrial Electronics.

[18]  Sukhendu Das,et al.  Mutual variation of information on transfer-CNN for face recognition with degraded probe samples , 2018, Neurocomputing.

[19]  Dongfeng Yuan,et al.  Deep Transfer Learning for Intelligent Cellular Traffic Prediction Based on Cross-Domain Big Data , 2019, IEEE Journal on Selected Areas in Communications.

[20]  Qin Hu,et al.  Machinery Fault Diagnosis Scheme Using Redefined Dimensionless Indicators and mRMR Feature Selection , 2020, IEEE Access.

[21]  Haidong Shao,et al.  Intelligent fault diagnosis of rolling bearing using deep wavelet auto-encoder with extreme learning machine , 2018, Knowl. Based Syst..

[22]  Xin Huang,et al.  Intelligent fault diagnosis method of planetary gearboxes based on convolution neural network and discrete wavelet transform , 2019, Comput. Ind..

[23]  Baoping Tang,et al.  A Novel Method for Mechanical Fault Diagnosis Based on Variational Mode Decomposition and Multikernel Support Vector Machine , 2016 .

[24]  Yu Li,et al.  Accelerating Flash Calculation through Deep Learning Methods , 2018, J. Comput. Phys..

[25]  Brigitte Chebel-Morello,et al.  Linear feature selection and classification using PNN and SFAM neural networks for a nearly online diagnosis of bearing naturally progressing degradations , 2015, Eng. Appl. Artif. Intell..

[26]  Fei Dong,et al.  Rolling Bearing Fault Diagnosis Using Modified LFDA and EMD With Sensitive Feature Selection , 2018, IEEE Access.

[27]  Jürgen Schmidhuber,et al.  Deep learning in neural networks: An overview , 2014, Neural Networks.

[28]  Jun Wang,et al.  Forecasting energy fluctuation model by wavelet decomposition and stochastic recurrent wavelet neural network , 2018, Neurocomputing.

[29]  Dawei Zhong,et al.  An Intelligent Fault Diagnosis Method based on STFT and Convolutional Neural Network for Bearings Under Variable Working Conditions , 2019, 2019 Prognostics and System Health Management Conference (PHM-Qingdao).

[30]  Hee-Jun Kang,et al.  A survey on Deep Learning based bearing fault diagnosis , 2019, Neurocomputing.

[31]  Zihan Zhang,et al.  Compound Fault Diagnosis of Gearboxes via Multi-label Convolutional Neural Network and Wavelet Transform , 2019, Comput. Ind..

[32]  Yueli Cui,et al.  Learning Affective Video Features for Facial Expression Recognition via Hybrid Deep Learning , 2019, IEEE Access.

[33]  Yu Zhang,et al.  A New Bearing Fault Diagnosis Method Based on Fine-to-Coarse Multiscale Permutation Entropy, Laplacian Score and SVM , 2019, IEEE Access.

[34]  Xianzhi Wang,et al.  Early fault diagnosis of rolling bearings based on hierarchical symbol dynamic entropy and binary tree support vector machine , 2018, Journal of Sound and Vibration.

[35]  Jian Sun,et al.  Deep Residual Learning for Image Recognition , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[36]  Chunbo Xiu,et al.  Target Detection Method Based on Improved Particle Search and Convolution Neural Network , 2019, IEEE Access.