Rolling Bearing Fault Diagnosis Based on Support Vector Machine Optimized by Improved Grey Wolf Algorithm

Abstract

This study targets the low accuracy and efficiency of the support vector machine (SVM) algorithm in rolling bearing fault diagnosis. An improved grey wolf optimizer (IGWO) algorithm was proposed based on deep learning and a swarm intelligence optimization algorithm to optimize the structural parameters of SVM and improve the rolling bearing fault diagnosis. A nonlinear contraction factor update strategy was also proposed. The variable coefficient changes with the shrinkage factor α. Thus, the search ability was balanced at different early and late stages by controlling the dynamic changes of the variable coefficient. In the early stages of optimization, its speed is low to avoid falling into local optimization. In the later stages of optimization, the speed is higher, and finding the optimal solution is easier, balancing the two different global and local optimization capabilities to complete efficient convergence. The dynamic weight update strategy was adopted to perform position updates based on adaptive dynamic weights. First, the dataset of Case Western Reserve University was used for simulation, and the results showed that the diagnosis accuracy of IGWO-SVM was 98.75%. Then, the IGWO-SVM model was trained and tested using data obtained from the full-life-cycle test platform of mechanical transmission bearings independently researched and developed by Nanjing Agricultural University. The fault diagnosis accuracy and convergence value of the adaptation curve were compared with those of PSO-SVM (particle swarm optimization) and GWO-SVM diagnosis models. Results showed that the IGWO-SVM model had the highest rolling bearing fault diagnosis accuracy and the best diagnosis convergence.