Fault Prediction of Centrifugal Pump Based on Improved KNN

Abstract

To effectively predict the faults of centrifugal pumps, the idea of machine learning k-nearest neighbor algorithm (KNN) was introduced into the traditional Mahalanobis distance fault discrimination, and an improved centrifugal pump fault prediction model of KNN based on the Mahalanobis distance is proposed. In this method, the Mahalanobis distance is used to replace the distance function in the conventional KNN algorithm. Grid search and cross-validation are used to determine the optimal K value of the prediction model. A centrifugal pump test rig was established to solve three common faults of centrifugal pumps: cavitation, impeller damage, and machine seal damage, and the method was verified. The results show that this method can effectively distinguish the specific fault types of centrifugal pumps based on vibration signals, and the fault prediction accuracy of the off-balance condition is up to 82%. This study provides a novel idea and method for centrifugal pump fault prediction and diagnosis and avoids the interaction between parameters when monitoring multiple parameters.