A Fault Diagnosis Method of Rolling Bearing Based on Attention Entropy and Adaptive Deep Kernel Extreme Learning Machine

Abstract

To address the difficulty of early fault diagnosis of rolling bearings, this paper proposes a rolling bearing diagnosis method by combining the attention entropy and adaptive deep kernel extreme learning machine (ADKELM). Firstly, the wavelet threshold denoising method is employed to eliminate the noise in the vibration signal. Then, the empirical wavelet transform (EWT) is utilized to decompose the denoised signal, and extract the attention entropy of the intrinsic mode function (IMF) as the feature vector. Next, the hyperparameters of the deep kernel extreme learning machine (DKELM) are optimized using the marine predators algorithm (MPA), so as to achieve the adaptive changes in the DKELM parameters. By analyzing the fault diagnosis performances of the ADKELM model with different kernel functions and hidden layers, the optimal ADKELM model is determined. Compared with conventional machine learning models such as extreme learning machine (ELM), back propagation neural network (BPNN) and probabilistic neural network (PNN), the high efficiency of the method proposed in this paper is verified.