Composite Fault Diagnosis of Rolling Bearing Based on Chaotic Honey Badger Algorithm Optimizing VMD and ELM

Abstract

In order to effectively extract the characteristic information of bearing vibration signals and improve the classification accuracy, a composite fault diagnosis method of rolling bearing based on the chaotic honey badger algorithm (CHBA), which optimizes variational mode decomposition (VMD) and extreme learning machine (ELM), is proposed in this paper. Firstly, aiming to solve the problem that the HBA optimization process can easily fall into local optimization and slow convergence speed, sinusoidal chaotic mapping is introduced to improve HBA, and the advantages of CHBA are verified by 23 benchmark functions. Then, taking the Gini index of the square envelope (GISE) as the fitness function, the VMD is optimized with CHBA to obtain the optimal number of modes K and the quadratic penalty factor. Secondly, the first four IMF components with the largest GISE values are selected, and the IMF components are grouped by the “Systematic Sampling Method (SSM)” to calculate the signal energy to form the fault feature vector. Finally, taking the classification error rate as the fitness function, the feature vector is input into the ELM model optimized by CHBA to classify and identify different types of faults. Through experimental analysis, and compared with BP, ELM, GWO-ELM, and HBA-ELM, this method has better diagnosis results for composite faults, and the accuracy of fault classification can reach 100%, which provides a new way to solve the problem of composite fault diagnosis.