Cutting-edge algorithms for reliable failure prediction in metro train systems

Abstract

This study investigated various machine learning algorithms on dataset for failure prediction within metro train systems. The data indicated strong linear relationships within the dataset, making linear models such as support vector machines (SVMs) viable, as well as logistic regression analysis. For example, the least absolute shrinkage and selection operator (LASSO) regularization method used in feature selection had profound implications, leading to enhanced performance through the identification of pertinent attributes. Some advanced models like gradient boosting machines (GBMs), convolutional neural networks (CNNs), and kernel SVMs were found to outperform the conventional methods because they are capable of recognizing any complicated trends or non-linear relationships present in data sets. Combining strong learners can produce an ensemble model that improves forecast performance, while top-performing models are used in the ensemble method to enhance prediction accuracy. These findings would help professionals in the metro train industry choose appropriate machine learning methods to support preventive maintenance strategies, minimizing costs while enhancing operational effectiveness and safety.