Rolling Bearing Fault Diagnosis Based on SABO–VMD and WMH–KNN

Abstract

To improve the performance of roller bearing fault diagnosis, this paper proposes an algorithm based on subtraction average-based optimizer (SABO), variational mode decomposition (VMD), and weighted Manhattan-K nearest neighbor (WMH–KNN). Initially, the SABO algorithm uses a composite objective function, including permutation entropy and mutual information entropy, to optimize the input parameters of VMD. Subsequently, the optimized VMD is used to decompose the signal to obtain the optimal decomposition characteristics and the corresponding intrinsic mode function (IMF). Finally, the weighted Manhattan function (WMH) is used to enhance the classification distance of the KNN algorithm, and WMH–KNN is used for fault diagnosis based on the optimized IMF features. The performance of the SABO–VMD and WMH–KNN models is verified through two experimental cases and compared with traditional methods. The results show that the accuracy of motor-bearing fault diagnosis is significantly improved, reaching 97.22% in Dataset 1, 98.33% in Dataset 2, and 99.2% in Dataset 3. Compared with traditional methods, the proposed method significantly reduces the false positive rate.