A feature fusion method using WPD-SVD and t-SNE for gearbox fault diagnosis

Jinwen Sun1, Chen Lu2, Jian Ma3

School of Reliability and Systems Engineering, Beihang University, Beijing, 100191, China

Science and Technology on Reliability and Environmental Engineering Laboratory, Beijing, 100191, China

3Corresponding author

E-mail: 1sunjinwen@buaa.edu.cn, 2luchen@buaa.edu.cn, 309977@buaa.edu.cn

Received 18 September 2017; accepted 26 September 2017

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


Abstract. The vibration signals of a gearbox always contain the dynamic operation information, which are important for the feature extraction and further work. However, the low signal-to-noise ratio and combined multi-mode faults make it difficult to extract discriminable features of gearboxes. In this study, a feature fusion method based on wavelet packet decomposition (WPD), singular value decomposition (SVD) and -Distributed stochastic neighbor embedding (-SNE) for gearbox fault diagnosis is proposed. First, time-frequency analysis method of WPT-SVD as well as time-domain analysis methods are utilized to extract robust feature vectors of gearboxes with different conditions. As an effective method for the visualization of high-dimensional datasets, -SNE is then introduced to realize the dimensionality reduction of feature vectors. Finally, with the fused features, a radial basis function (RBF) neural network is trained to realize the classification of gearbox fault modes. Sufficient experiments have been implemented to validate the effectiveness and superiority of the proposed method by analyzing the vibration signals of gearboxes.

Keywords: gearbox, fault diagnosis, wavelet packet decomposition, t-distributed stochastic neighbor embedding.


[1]        Chaari F., Fakhfakh T., Haddar M. Dynamic analysis of a planetary gear failure caused by tooth pitting and cracking. Journal of Failure Analysis and Prevention, Vol. 6, Issue 2, 2006, p. 73‑78.

[2]        Lin J., Zuo M. J. Gearbox fault diagnosis using adaptive wavelet filter. Mechanical Systems and Signal Processing, Vol. 17, Issue 6, 2003, p. 1259‑1269.

[3]        Wang J., Jiang C., Han Z., et al. Network association strategies for an energy harvesting aided super-wifi network relying on measured solar activity. IEEE Journal on Selected Areas in Communications, Vol. 12, 2016, p. 3785‑3797.

[4]        Sreejith B., Verma A. K., Srividya A. Fault diagnosis of rolling element bearing using time-domain features and neural networks. 3rd international Conference on Industrial and Information Systems, 2008, p. 1‑6.

[5]        Liu B., Riemenschneider S., Xu Y. Gearbox fault diagnosis using empirical mode decomposition and Hilbert spectrum. Mechanical Systems and Signal Processing, Vol. 20, Issue 3, 2006, p. 718‑734.

[6]        Zhang Z., Zha H. Principal manifolds and nonlinear dimensionality reduction via tangent space alignment. Society for Industrial and Applied Mathematics, 2005.

[7]        Hinton G. E. Visualizing high-dimensional data using t-SNE. Vigiliae Christianae, Vol. 9, Issue 2, 2008, p. 2579‑2605.

Cite this article

Sun Jinwen, Lu Chen, Ma Jian A feature fusion method using WPD‑SVD and t‑SNE for gearbox fault diagnosis. Vibroengineering PROCEDIA, Vol. 14, 2017, p. 91‑96.


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