Maximal Information Coefficient-Based Two-Stage Feature Selection Method for Railway Condition Monitoring

Abstract

In railway condition monitoring, feature classification is a very critical step, and the extracted features are used to classify the types and levels of the faults. To achieve better accuracy and efficiency in the classification, the extracted features must be properly selected. In this paper, maximal information coefficient is employed in two different stages to establish a new feature selection method. By using this proposed two-stage feature selection method, strong features with low redundancy are reserved as the optimal feature subset, which results in the classification process having a more moderate computational cost and good overall performance. To evaluate this proposed two-stage selection method and prove its advantages over others, a case study focusing on the rolling bearing is carried out. The result shows that the proposed selection method can achieve a satisfactory overall classification performance with low-computational cost.