Rolling Bearing Fault Diagnosis Method Based on Parallel QPSO-BPNN Under Spark-GPU Platform

Facing the massive rolling bearing vibration data, how to improve the training efficiency, diagnosis efficiency, and diagnosis accuracy of the rolling bearing fault diagnosis model is a challenge. Considering that the Spark-GPU platform provides powerful distributed parallel computing capabilities and back propagation neural network (BPNN) optimized by quantum particle swarm optimization (QPSO) algorithm has the characteristics of low computational complexity and high diagnosis accuracy, a rolling bearing fault diagnosis method based on parallel QPSO-BPNN under Spark-GPU platform is proposed. First, the distributed parallelization of QPSO-BPNN model based on Spark-GPU platform is realized, which can improve the training efficiency and diagnosis efficiency of rolling bearing fault diagnosis model in the big data environment. Second, in order to improve the convergence speed of fault diagnosis model, a parameter update strategy suitable for the distributed parallel training of QPSO-BPNN model is designed. At each iteration during training, the local parameters of each worker node are collected to the master node, and the global parameters are updated according to the weights and synchronized to each worker node. Third, a combination strategy of multiple QPSO-BPNN models based on ensemble learning is proposed. The weighted voting method is adopted to combine the output results of different QPSO-BPNN models to obtain the best fault diagnosis result of a sample, which can improve the fault diagnosis accuracy to a certain extent. Experimental results show that the proposed method can quickly perform model training and fault diagnosis for large-scale rolling bearing vibration data, and the fault diagnosis accuracy reaches 98.73%.