Time-Frequency Fusion Features-Based GSWOA-KELM Model for Gear Fault Diagnosis

Abstract

To improve the accuracy of gear fault diagnosis and overcome the low diagnostic accuracy of the model caused by manual parameter selection, a combined diagnostic model based on time-frequency fusion features is combined with the improved global search whale optimization algorithm (GSWOA) to optimize the fault diagnosis capability of the kernel extreme learning machine (KELM). First, the time-domain and frequency-domain features of the gear fault state are extracted separately, and feature vectors are constructed through feature fusion, which overcomes the limitations of single features. Second, the GSWOA based on three strategies is used to optimize the regularization coefficient C and kernel function parameter γ of KELM, and a GSWOA-KELM fault diagnosis model is built to avoid the problem of low fault diagnosis accuracy caused by the manual selection of KELM parameters. Finally, the public dataset from Southeast University is taken to verify the performance of the proposed model by comparing it with KELM, SSA-KELM, and WOA-KELM models. The experimental results demonstrate that the improved time-frequency fusion features-based GSWOA-KELM model shows faster convergence speed and stronger global search ability. Compared with KELM, SSA-KELM, and WOA-KELM models, the performance of the proposed model has been improved by 11.33%, 8.67%, and 1.33%, respectively.