Multi-fault diagnosis of rolling bearing using two-dimensional feature vector of WP-VMD and PSO-KELM algorithm

Abstract

In order to achieve accurate fault diagnosis of rolling bearing under random noise, a new fault diagnosis method based on wavelet packet-variational mode decomposition (WP-VMD) and kernel extreme learning machine (KELM) optimized by particle swarm optimization (PSO) is proposed in this paper. Firstly, the time–frequency domain feature vectors of the original rolling bearing fault signals are effectively obtained by preprocessing of WMD and decomposition and reconstruction of VMD. Then, the extracted two-dimensional feature vector is input into the KELM neural network for fault identification, and combined with PSO, KELM parameters were optimized. The experimental results show that the proposed method can effectively diagnose the rolling bearing under random noise, with the features of fast speed, stable performance and high accuracy. By comparison, this paper obtains better accuracy and real-time performance with fewer features, which provides a simple and efficient solution for fault diagnosis of rolling bearings.