Intelligent classification model for railway signal equipment fault based on SMOTE and ensemble learning

2018 International Joint Conference on Materials Science and Mechanical Engineering

Yang, Lianbao^1 ; Li, Ping^1 ; Xue, Rui^1 ; Ma, Xiaoning^1 ; Li, Xinqin^1 ; Wang, Zhe^1

China Academy of Railway Sciences, Beijing

100081, China^1

关键词: Ensemble classifiers;    Experiment analysis;    Fault classification accuracy;    Imbalanced faults;    Intelligent classification;    Logistic regressions;    Multinomial naive bayes;    Multiple classifiers;   

Others  :  https://iopscience.iop.org/article/10.1088/1757-899X/383/1/012042/pdf
DOI  :  10.1088/1757-899X/383/1/012042

来源: IOP

【 摘 要 】

In this paper, we propose a novel intelligent classification model to classify the railway signal equipment fault based on SMOTE and ensemble learning. To tackle the imbalanced fault text data, the model uses SMOTE algorithm to generate the minority railway signal equipment fault class data randomly, making the data balanced. Then the model adopts the base classifier, such as Logistic Regression, Multinomial Naive Bayes, SVM and the ensemble classifier, such as GBDT, Random Forests to classify the data processed by SMOTE. To combine the advantages of various classifiers, the model integrates multiple classifiers by way of voting. Based on the experiment analysis of railway signal equipment fault text data from 2012 to 2016, the result shows that the model has a significant improvement in fault classification accuracy, recall rate and f-score.