Sparse Temporal Data-Driven SSA-CNN-LSTM-Based Fault Prediction of Electromechanical Equipment in Rail Transit Stations

Abstract

Mechanical and electrical equipment is an important component of urban rail transit stations, and the service capacity of stations is affected by its reliability. To solve the problem of predicting faults in station mechanical and electrical equipment with sparse data, this study proposes a fault prediction framework based on SSA-CNN-LSTM. Firstly, this article proposes a fault enhancement method for station electromechanical equipment based on TimeGAN, which expands and generates data that conform to the temporal characteristics of the original dataset, to solve the problem of sparse data in the original fault dataset. An SSA-CNN-LSTM model is then established to extract effective data features from low-dimensional data with insufficient feature depth through structures such as convolutional layers and pooling layers in a CNN, determine the optimal hyperparameters, automatically optimize the model network size, solve the problem of the difficult determination of the neural network model size, and achieve accurate prediction of the fault rate of station electromechanical equipment. Finally, an engineering verification was conducted on the platform screen door (PSD) systems in stations on Shanghai Metro Lines 1, 5, 9, and 10. The experiments showed that the proposed prediction method improved the RMSE by 0.000699, the MAE by 0.00042, and the R2 index by 0.109779 when predicting the fault rate data of platform screen doors on all of the lines. When predicting the fault rate data of the screen doors on a single line, the performance of the model was better than that of the CNN-LSTM model optimized with the PSO algorithm.