Rolling bearing fault diagnosis based on bacterial foraging algorithm and deep belief network

Jie TAO,Yilun LIU,Dalian YANG,Guangfu BIN
DOI: https://doi.org/10.13465/j.cnki.jvs.2017.23.011
2017-01-01
Abstract:When studying rolling bearing fault diagnosis with the deep belief network method,parameters in the deep belief network have a great effect on fault diagnosis results and it is hard to obtain suitable parameters.Here,the fault diagnosis method based on the bacterial foraging algorithm and the deep belief network was proposed to improve the correct rate of bearing fault diagnosis.The parallel search ability of the bacterial foraging algorithm was adopted to effectively choose the number of hidden layer,the number of hidden nodes,the learning rate in a deep belief network.The deep belief network's training data classification error was used to calculate the fitness function of the bacterial foraging algorithm to build an appropriate fault classifier and finish rolling bearing fault diagnosis.The test results showed that the correct rate of the proposed method for rolling bearing fault diagnosis reaches 98.5%;compared with BPNN,SVM and KNN methods,the proposed method can more stably and more accurately identify rolling bearing faults.
What problem does this paper attempt to address?