Fault Early Warning Model for High-Speed Railway Train Based on Feature Contribution and Causal Inference.
Sensors (Basel)
; 22(23)2022 Nov 25.
Article
em En
| MEDLINE
| ID: mdl-36501884
ABSTRACT
The demands for model accuracy and computing efficiency in fault warning scenarios are increasing as high-speed railway train technology continues to advance. The black box model is difficult to interpret, making it impossible for this technology to be widely adopted in the railway industry, which has strict safety regulations. This paper proposes a fault early warning machine learning model based on feature contribution and causal inference. First, the contributions of the features are calculated through the Shapley additive explanations model. Then, causal relationships are discovered through causal inference models. Finally, data from causal and high-contribution time series are applied to the model. Ablation tests are conducted with the Naïve Bayes, Gradient Boosting Decision Tree, eXtreme Gradient Boosting, and other models in order to confirm the efficiency of the method based on early warning data regarding the on-site high-speed train traction equipment circuit board failure. The findings indicate that the strategy improves the evaluation markers, including the early warning accuracy, precision, recall, and F1 score, by an average of more than 10%. There is a 35% improvement in the computing efficiency, and the model can provide feature causal graph verification for expert product decision-making.
Palavras-chave
Texto completo:
1
Coleções:
01-internacional
Base de dados:
MEDLINE
Assunto principal:
Aprendizado de Máquina
/
Indústrias
Tipo de estudo:
Prognostic_studies
Idioma:
En
Ano de publicação:
2022
Tipo de documento:
Article