Predicting Cargo Train Failures: A Machine Learning Approach for a Lightweight Prototype

In cargo transportation, reliability is a crucial issue. In the case of railway traffic, the consequences of locomotive failure are not limited to the affected machine, but are propagated through the railway network and may affect public transport as well. Therefore it is desirable to predict and avoid failures. In order to do this, constant monitoring of the trains’ systems and measurement of the relevant variables is required, but often not implemented. In this paper we leverage the existing technology of the 185 locomotive series and build a layered model for power converter failure prediction that can be applied without additional technology. We train instance anomaly detectors based on the pattern structure of the locomotives’ diagnostic messages from historical data records. For this purpose we selected rule and decision tree learning because they can be easily implemented in the existing software, whereas more complex classifiers would require costly software adaptations. In order to predict a time series of instances, we construct a meta classification layer. We then evaluate our model on the data of 180 locomotive tours by leave one out classification. The results show that the meta classifier improves classification accuracy, which will allow us to use this technology in a fielded prototype installation without disturbing daily operations.