Effects of vibration direction, feature selection, and the svm kernel on unbalance fault classification

Abstract

In this study, the combined influence of vibration direction, feature selection strategy, and the support vector machine (SVM) kernel on the classification accuracy of unbalance faults was investigated. Experiments were carried out on a Jeffcott rotor test rig at a constant speed and under three operating conditions. The overlapping sliding window method was used for raw sample expansion. Features extracted from time domain signals and from the order and power spectra obtained in the frequency domain were ranked using the Kruskal-Wallis algorithm. Based on the feature-ranking results, the three most discriminative features for each domain-axis combination, as well as all nine most discriminative features for each axis in a hybrid manner, were fed into SVM classifiers with different kernels, and their performance was evaluated using ten-fold cross-validation. Classification using vibration signals in the vertical direction had higher accuracy rates than those using signals in the horizontal direction for the feature sets obtained in the same domains. According to the statistical results, feature set selection had a much greater impact on classification accuracy than SVM kernel choice. Power spectrum-based features allowed higher classification accuracies in all SVM algorithms compared to both the time domain features and the order spectrum-based features for detecting unbalance faults. Increasing the number of features or employing hybrid feature selection did not result in a consistent or significant enhancement in overall classification performance. Selecting the right SVM kernel shapes both the model's flexibility and its fit to the chosen feature space; when this fit is inadequate, classification accuracy may decrease. Consequently, by selecting the appropriate vibration direction, feature set, and SVM kernel, an improvement of up to 67% in unbalance fault classification accuracy was achieved.