End-edge-cloud collaborative learning-aided prediction for high-speed train operation using LSTM.

This paper aims to incorporate the throttle handle level prediction in high speed train(HST) operation prediction problem to enable the prediction of HST drivers' activities, in which the key instructions available to HST driver are difficult to determine. Specifically, we consider an end-edge-cloud orchestration system to capture the real-time responses for driver state changes. By adding edge computing nodes, the real-time performance of data collection, transmission, and processing is improved. Our ultimate goal is to guide and regulate train drivers' activities in the same way, regardless of uncertain factors affecting HST dynamic or kinematic performance. We formulate the problem as a physical-based and data-driven deep learning-aided prediction model and solve it using a novel long short-term memory (LSTM) deep neural network which combines: (i) an off-line approximate training model to learn the time series data in the cloud layer, and (ii) an online prediction process to determine driving strategies in the real-time windows, more in general expressed as driving skill level constraints. To evaluate the performance of our approach, some case studies using the real-world railway infrastructure and HST data have been conducted. The results show that the proposed models produce higher prediction accuracy for both speed and throttle handle level prediction tasks. Compared to the conventional HST operation prediction problem, which considers speed sequences only without throttle handle level consideration, this study finds that jointly modeling speed and throttle handle level actually improves the next operation prediction performance itself, potentially because throttle handle level observations capture the information on HST control dynamics, which may affect operators' driving choices. • Structure. Edge computing and prediction scheme are innovatively combined with HST data collection to meet the delay-sensitive and sequential-aware operation requirements, and a novel HST operation structure for end-edge cloud orchestration system is built. • Modeling. A combination of the physical-based and the data-driven deep learning model is formulated for HST operation prediction under uncertainty in throttle handle level and speed, which is a new model in the literature. As discussed, the existing research has only considered speed dynamics in real-time DAS limitedly. • Practice. From the field collected data in Beijingxi–Zhengzhoudong HSR (as shown in Figure 5), we test the effectiveness of our proposed model with seven cases. Since the real-time driving advice is calculated in a few milliseconds, our two-phase solution method can be practically relevant when designing DAS or ATO systems. More importantly, most time-consuming calculations are executed off-line in the cloud layer and the amount of on-line calculations in edge layer is limited to looking up a table. [ABSTRACT FROM AUTHOR]