Optimization-based improved kernel extreme learning machine for rolling bearing fault diagnosis

  • Longkui Zheng
  • Yang XiangEmail author
  • Chenxing Sheng
Technical Paper


Rolling bearing is one of the key components in rotating machinery. The working condition of rolling bearing is complex and non-stationary with shock and noise. Thus, fault diagnosis of rolling bearing is of great significance in rotating machinery. In this paper, a novel method called optimization-based improved kernel extreme learning machine is proposed for fault diagnosis of rolling bearing. Firstly, different signal processing methods and data analysis methods are used as the second layer for feature extraction and vibration data dimension reduction, and the extracted data are sorted by experience pool. Secondly, kernel extreme learning machine (K-ELM) is used as the hidden layer and the output layer to enhance feature learning and classification of the extracted data. Finally, particle swarm optimization is employed to optimize the key parameters of the improved kernel extreme learning machine. The proposed method and five other methods are applied to analyze the raw vibration data of rolling bearing, and the results confirm that the proposed method is more effective than ELM, K-ELM, optimized K-ELM, support vector machine and back-propagation neural network.


Multi-angle features Kernel extreme learning machine Fault diagnosis Particle swarm optimization Roller bearing 



The research is founded in part under the NSFC-Zhejiang Joint Found for the Integration of Industrialization and informatization (Project No. U1709215), and the National Natural Science Foundation of China (Project Nos. 51079118 and 51279148).


  1. 1.
    Shao HD, Jiang HK, Zhang HZ, Duan WJ, Liang TC, Wu SP (2018) Rolling bearing fault feature learning using improved convolutional deep belief network with compressed sensing. Mech Syst Signal Process 100:743–765CrossRefGoogle Scholar
  2. 2.
    Gao JX, Yang JH, Huang DW, Liu HG, Liu SY (2019) Experimental application of vibrational resonance on bearing fault diagnosis. J Braz Soc Mech Sci Eng 6:1–13Google Scholar
  3. 3.
    Yan XA, Jia MP (2019) Improved singular spectrum decomposition-based 1.5-dimensional energy spectrum for rotating machinery fault diagnosis. J Braz Soc Mech Sci Eng 50:1–20Google Scholar
  4. 4.
    Lei YG, Lin J, He ZJ, Zuo MJ (2013) A review on empirical mode decomposition in fault diagnosis of rotating machinery. Mech Syst Signal Process 35:108–126CrossRefGoogle Scholar
  5. 5.
    Lei YG, He ZJ, Zi YY (2009) Application of an intelligent classification method to mechanical fault diagnosis. Expert Syst Appl 36:9941–9948CrossRefGoogle Scholar
  6. 6.
    Prieto MD, Cirrincione G, Espinosa AG (2013) Bearing fault detection by a novel condition-monitoring scheme based on statistical-time features and neural networks. IEEE Trans Industr Electron 8:3398–3407CrossRefGoogle Scholar
  7. 7.
    Liu TY, Li GZ (2010) The imbalanced data problem in the fault diagnosis of rolling bearing. Comput Eng Sci 32:150–153Google Scholar
  8. 8.
    Prieto MD, Cirrincione G, Espinosa AG, Ortega JA, Henao H (2013) Bearing fault detection by a novel condition-monitoring scheme based on statistical-time features and neural networks. IEEE Trans Ind Electron 30:3398–3407CrossRefGoogle Scholar
  9. 9.
    Chen GZ, Wu YF, Fu L, Bai N (2019) Fault diagnosis of full-hydraulic drilling rig based on RS–SVM data fusion method. J Braz Soc Mech Sci Eng 140:1–13Google Scholar
  10. 10.
    Pan HH, He XX, Tang S, Meng FM (2018) An improved bearing fault diagnosis method using one-dimensional CNN and LSTM. J Mech Eng 64:443–452Google Scholar
  11. 11.
    Cai ZY, Xu YB, Duan ZS (2018) An alternative demodulation method using envelope-derivative operator for bearing fault diagnosis of the vibrating screen. J Vib Control 24:3249–3261MathSciNetCrossRefGoogle Scholar
  12. 12.
    Li W, Qiu MQ, Zhu ZC, Wu B, Zhou GB (2016) Bearing fault diagnosis based on spectrum images of vibration signals. Meas Sci Technol 27:1–10Google Scholar
  13. 13.
    Boukra T, Lebaroud A, Clerc G (2013) Statistical and neural-network approaches for the classification of induction machine faults using the ambiguity plane representation. IEEE Trans Ind Electron 60:4034–4042CrossRefGoogle Scholar
  14. 14.
    Liu HH, Han MH (2014) A fault diagnosis method based on local mean decomposition and multi-scale entropy for roller bearings. Mech Mach Theory 75:67–78CrossRefGoogle Scholar
  15. 15.
    Rauber TW, Boldt FDA, Varejao FM (2015) Heterogeneous feature models and feature selection applied to bearing fault diagnosis. IEEE Trans Ind Electron 62:637–646CrossRefGoogle Scholar
  16. 16.
    Liu RN, Yang BY, Zhang XL, Wang SB, Chen XF (2016) Time–frequency atoms-driven support vector machine method for bearings incipient fault diagnosis. Mech Syst Signal Process 75:345–370CrossRefGoogle Scholar
  17. 17.
    Xia X, Zhou JZ, Xiao J (2016) A novel identification method of Volterra series in rotor bearing system for fault diagnosis. Mech Syst Signal Process 66–67:557–567CrossRefGoogle Scholar
  18. 18.
    Keskes H, Braham A, Lachiri Z (2013) Broken rotor bar diagnosis in induction machines through stationary wavelet packet transform and multiclass wavelet SVM. Electr Power Syst Res 97:151–157CrossRefGoogle Scholar
  19. 19.
    Muruganatham B, Sanjith MA, Krishnakumar B (2013) Roller element bearing fault diagnosis using singular spectrum analysis. Mech Syst Signal Process 35:150–166CrossRefGoogle Scholar
  20. 20.
    Kang M, Kim J, Kim JM (2014) An FPGA-based multicore system for real-time bearing fault diagnosis using ultrasampling rate AE signals. IEEE Trans Ind Electron 61:2041–2052Google Scholar
  21. 21.
    Zimroz R, Bartelmus W, Barszcz T, Urbanek J (2014) Diagnostics of bearing in presence of strong operating conditions non-stationarity: a procedure of load-dependent features processing with application to wind turbine bearings. Mech Syst Signal Process 46:16–27CrossRefGoogle Scholar
  22. 22.
    Jia F, Lei YG, Lin J, Zhou X, Lu N (2016) Deep neural networks: a promising tool for fault characteristic mining and intelligent diagnosis of rotating machinery with massive data. Mech Syst Signal Process 72–73:303–315CrossRefGoogle Scholar
  23. 23.
    Shao HD, Jiang HK, Zhang HZ, Liang TC (2018) Electric locomotive bearing fault diagnosis using a novel convolutional deep belief network. IEEE Trans Ind Electron 65:2727–2736CrossRefGoogle Scholar
  24. 24.
    Tran VT, AlThobiani F, Ball A (2014) An approach to fault diagnosis of reciprocating compressor valves using Teager–Kaiser energy operator and deep belief networks. Expert Syst Appl 41:4113–4122CrossRefGoogle Scholar
  25. 25.
    Shao HD, Jiang HK, Zhao HW, Wang F (2017) A novel deep autoencoder feature learning method for rotating machinery fault diagnosis. Mech Syst Signal Process 95:187–204CrossRefGoogle Scholar
  26. 26.
    Shao HD, Jiang HK, Wang F, Zhao HW (2017) An enhancement deep feature fusion method for rotating machinery fault diagnosis. Knowl-Based Syst 119:200–220CrossRefGoogle Scholar
  27. 27.
    Li MB, Huang GB, Saratchandram P, Sundararajan N (2005) Fully complex extreme learning machine. Neurocomputing 68:306–314CrossRefGoogle Scholar
  28. 28.
    Huang GB, Zhu QY, Siew CK (2006) Extreme learning machine: a new learning scheme of feedforward neural networks. Neurocomputing 70:489–501CrossRefGoogle Scholar
  29. 29.
    Huang GB, Li MB, Chen L, Siew CK (2008) Incremental extreme learning machine with fully complex hidden nodes. Neurocomputing 71:576–583CrossRefGoogle Scholar
  30. 30.
    Zhang YS, Wu J, Cai ZH, Zhang P, Chen L (2016) Memetic extreme learning machine. Pattern Recogn 58:135–148CrossRefGoogle Scholar
  31. 31.
    Huang GB, Wang DH, Lan Y (2011) Extreme learning machines: a survey. Int J Mach Learn Cybern 2:107–122CrossRefGoogle Scholar
  32. 32.
    de Assis Boldt F, Rauber TW, Varejao FM (2017) Cascade feature selection and ELM for automatic fault diagnosis of the Tennessee Eastman process. Neurocomputing 239:238–248CrossRefGoogle Scholar
  33. 33.
    Li K, Su L, Wu JJ, Wang HQ, Chen P (2017) A rolling bearing fault diagnosis method based on variational mode decomposition and an improved kernel extreme learning machine. Appl Sci 7:1–20CrossRefGoogle Scholar
  34. 34.
    Wang Y, Li B, Weise T, Wang J-Y et al (2011) Self-adaptive learning based particle swarm optimization. Inf Sci 181:4515–4538MathSciNetCrossRefGoogle Scholar
  35. 35.

Copyright information

© The Brazilian Society of Mechanical Sciences and Engineering 2019

Authors and Affiliations

  1. 1.School of Energy and Power EngineeringWuhan University of TechnologyWuhanChina
  2. 2.Key Laboratory of Marine Power Engineering and TechnologyMinistry of CommunicationsWuhanChina

Personalised recommendations