Fault Diagnosis of Rolling Bearing Based on SDAE and PSO-DBN

A new fault diagnosis method for rolling bearing based on two-step cascaded system with two deep network is presented in this paper. In view of the low accuracy of traditional diagnosis algorithm, Stacked Denoising Auto-Encoder (SDAE) model as the first network is used to extract the basic and shallow feature of the fault signal; in order to acquire more robust and deep feature representation, Deep Belief Network (DBN) is configured as the second network. However, as for specific fault diagnosis problems, the number of hidden layer nodes, learning rate and momentum factor will directly affect the diagnosis result of DBN model. Therefore, this paper adopts particle swarm optimization (PSO) algorithm to adaptively select the hyper-parameters of DBN to determine the optimal structure of network, finally realizes the classification of multiple faults. Rolling bearing fault simulation and experiments have been conducted under single load condition to verify the effectiveness of the proposed algorithm. Experimental results obviously demonstrate that, from the aspects of generalization capability and classification performance, this algorithm is superior to support vector machine (SVM), back propagation neural network (BPNN) and grey relational analysis (GRA).

[1]  Honglak Lee,et al.  An Analysis of Single-Layer Networks in Unsupervised Feature Learning , 2011, AISTATS.

[2]  Yoshua Bengio,et al.  Exploring Strategies for Training Deep Neural Networks , 2009, J. Mach. Learn. Res..

[3]  Yoshua Bengio,et al.  Extracting and composing robust features with denoising autoencoders , 2008, ICML '08.

[4]  Shuilong He,et al.  A novel intelligent method for bearing fault diagnosis based on affinity propagation clustering and adaptive feature selection , 2017, Knowl. Based Syst..

[5]  Geoffrey E. Hinton.  Training Products of Experts by Minimizing Contrastive Divergence , 2002, Neural Computation.

[6]  Peter Glöckner,et al.  Why Does Unsupervised Pre-training Help Deep Learning? , 2013 .

[7]  Russell C. Eberhart,et al.  A new optimizer using particle swarm theory , 1995, MHS'95. Proceedings of the Sixth International Symposium on Micro Machine and Human Science.

[8]  周东华,et al.  Review of multiple fault diagnosis methods , 2015 .

[9]  Yoshua Bengio,et al.  Gradient-based learning applied to document recognition , 1998, Proc. IEEE.

[10]  Chukwudi Anyakoha,et al.  A review of particle swarm optimization. Part I: background and development , 2007, Natural Computing.

[11]  Ioan Cristian Trelea,et al.  The particle swarm optimization algorithm: convergence analysis and parameter selection , 2003, Inf. Process. Lett..

[12]  Geoffrey E. Hinton.  A Practical Guide to Training Restricted Boltzmann Machines , 2012, Neural Networks: Tricks of the Trade.

[13]  Yee Whye Teh,et al.  A Fast Learning Algorithm for Deep Belief Nets , 2006, Neural Computation.

[14]  Robert X. Gao,et al.  Learning features from vibration signals for induction motor fault diagnosis , 2016, 2016 International Symposium on Flexible Automation (ISFA).

[15]  Geoffrey E. Hinton,et al.  Reducing the Dimensionality of Data with Neural Networks , 2006, Science.

[16]  Chikkannan Eswaran,et al.  Reconstruction and recognition of face and digit images using autoencoders , 2010, Neural Computing and Applications.

[17]  Marc'Aurelio Ranzato,et al.  Sparse Feature Learning for Deep Belief Networks , 2007, NIPS.

[18]  Pascal Vincent,et al.  Stacked Denoising Autoencoders: Learning Useful Representations in a Deep Network with a Local Denoising Criterion , 2010, J. Mach. Learn. Res..