Fault diagnosis of spur gearbox based on random forest and wavelet packet decomposition

Diego Cabrera, Fernando Sancho, René Vinicio Sánchez, Grover Zurita, Mariela Cerrada, Chuan Li, Rafael E. Vásquez

Research output: Contribution to journalArticlepeer-review

41 Scopus citations


This paper addresses the development of a random forest classifier for the multi-class fault diagnosis in spur gearboxes. The vibration signal’s condition parameters are first extracted by applying the wavelet packet decomposition with multiple mother wavelets, and the coefficients’ energy content for terminal nodes is used as the input feature for the classification problem. Then, a study through the parameters’ space to find the best values for the number of trees and the number of random features is performed. In this way, the best set of mother wavelets for the application is identified and the best features are selected through the internal ranking of the random forest classifier. The results show that the proposed method reached 98.68% in classification accuracy, and high efficiency and robustness in the models.

Original languageEnglish
Pages (from-to)277-286
Number of pages10
JournalFrontiers of Mechanical Engineering
Issue number3
StatePublished - 11 Sep 2015

Bibliographical note

Publisher Copyright:
© 2015, Higher Education Press and Springer-Verlag Berlin Heidelberg.


  • fault diagnosis
  • random forest
  • spur gearbox
  • wavelet packet decomposition


Dive into the research topics of 'Fault diagnosis of spur gearbox based on random forest and wavelet packet decomposition'. Together they form a unique fingerprint.

Cite this