Keras implementation of ResNet

Keras implementation of Attention LSTM

Keras implementation of GCN

We propose a deep-learning architecture combined residual network (ResNet), graph convolutional network (GCN) and long short-term memory (LSTM) (called “ResLSTM”) to forecast short-term passenger flow in urban rail transit on a network scale. First, improved methodologies of ResNet, GCN, and attention LSTM models are presented. Then, model architecture is proposed, wherein ResNet is used to capture deep abstract spatial correlations between subway stations, GCN is applied to extract network-topology information, and attention LSTM is used to extract temporal correlations. Model architecture includes four branches for inflow, outflow, graph-network topology, as well as weather conditions and air quality. To the best of our knowledge, this is the first time that air-quality indicators have been taken into account, and their influences on prediction precision have been quantified. Finally, ResLSTM is applied to Beijing subway. Three time granularities (10, 15, and 30 min) are chosen to conduct short-term passenger flow forecasting. Comparison of prediction performance of ResLSTM with those of many state-of-the-art models shows the advancement and robustness of ResLSTM. Moreover, comparison of prediction precisions obtained from time granularities of 10, 15, and 30 min indicates that prediction precision increases with increasing time granularity.

The dimension of inflow data is n*time steps, where n represents number of stations and time steps denote time steps in 25 weekdays.

The structure of outflow data is the same with inflow data.

The dimension of meteorology data is n*time steps, where n represents the 11 meteorology indicators such as temperature, wind speed.

Keras == 2.2.4
tensorflow-gpu == 1.10.0
numpy == 1.14.5
scipy == 1.3.3
scikit-learn == 0.20.2
protobuf == 3.6.0

Just download this repository and using PyCharm to open it. Then run ResLSTM.py.

J. Zhang, F. Chen, Z. Cui, Y. Guo and Y. Zhu, "Deep Learning Architecture for Short-Term Passenger Flow Forecasting in Urban Rail Transit," in IEEE Transactions on Intelligent Transportation Systems, doi: 10.1109/TITS.2020.3000761.