Fault Diagnosis of Rolling Bearing Based on Improved VMD and KNN

Variational modal decomposition (VMD) has the end effect, which makes it difficult to efficiently obtain fault eigenvalues from rolling bearing fault signals. Inspired by the mirror extension, an improved VMD is proposed. This method combines VMD and mirror extension. The mirror extension is a basic algorithm to inhibit the end effect. A comparison is made with empirical mode decomposition (EMD) for fault diagnosis. Experiments show that the improved VMD outperforms EMD in extracting the fault eigenvalues. The performance of the new algorithm is proven to be effective in real-life mechanical fault diagnosis. Furthermore, in this article, combining with singular value decomposition (SVD), fault eigenvalues are extracted. In this way, fault classification is realized by K-nearest neighbor (KNN). Compared with EMD, the proposed approach has advantages in the recognition rate, which can accurately identify fault types.