Induction Motor Fault Classification Based on ROC Curve and t-SNE

Abstract

This paper proposes a novel fault classification method with application to induction motors, which is based on integrating and combining with receiver operating characteristic (ROC) curve and t-distribution stochastic neighbor embedding (t-SNE). According to the feature selection methods of ReliefF, symmetrical uncertainty (SU), and fast correlation-based filter (FCBF), the significant features were verified. Additionally, support vector machine (SVM), k-nearest neighbor (KNN), and decision tree (DT) are also considered as classifiers to identify the simulation results. To begin with, the current signals obtained from distinctive four topologies of working conditions of the motor, which includes healthy, bearing damage, broken rotor bar, and short circuit in stator windings, respectively. The potential feature set is extracted by using Hilbert-Huang transform (HHT) technique. Then, three feature selection methods are adopted to select three optimal feature subsets from the original feature set. Finally, the classification accuracy (ACC) and ROC curve are used to demonstrate the capability of classifiers’ recognition. The results showed that the optimal feature subsets significantly reduce the number of selected features and improve the classification ACC and area under the curve (AUC) compared with the original feature set. In conclusion, the proposed method can downgrade the data, demonstrate the scatter plot more intuitively, and identify various types of faults, unlike with other fault diagnosis literature.