Shortening passengers' travel time: A dynamic metro train scheduling approach using deep reinforcement learning

Abstract

As travel efficiency matters to the work productivity of cities, shortening passengers' travel time for metros is therefore a pressing need. To this end, we study a strategy by dynamically scheduling dwell time for trains. Developing such a strategy is challenging because of three aspects: 1) Optimizing the average travel time of passengers needs to properly balance passengers' waiting time at platforms and journey time on trains, as well as considering long-term impacts; 2) Capturing dynamic spatio-temporal (ST) correlations of incoming passengers for metro stations is difficult; and 3) For each train, the dwell time scheduling is affected by other trains, which is hard to measure. To tackle these challenges, we propose a novel deep neural network, entitled AutoDwell. Specifically, AutoDwell optimizes the long-term rewards of dwell time settings in terms of passengers' waiting and journey time by a reinforcement learning framework. Next, AutoDwell employs gated recurrent units and graph attention networks to extract the ST correlations of the passenger flows among metro stations. Moreover, attention mechanisms are leveraged in AutoDwell for capturing the interactions between the trains. Extensive experiments on two real-world datasets demonstrate the superior performance of AutoDwell over several baselines, capable of saving passengers' travel time significantly.