ST-Attn: Spatial-Temporal Attention Mechanism for Multi-step Citywide Crowd Flow Prediction

Yirong Zhou,Hao Chen,Jun Li,Ye Wu,Jiangjiang Wu,Luo Chen
DOI: https://doi.org/10.1109/ICDMW.2019.00092
2019-01-01
Abstract:Multi-step citywide crowd flow prediction (MsCCFP) is to predict the in/out flow of each region in a city in the given multiple consecutive periods. For traffic control and public safety protection, it can provide a long term view for taking measures. However, the spatial and temporal correlations in crowd movements and the lack of information make MsCCFP challenging. In this paper, a deep-learning based prediction model with spatial-temporal attention mechanism is proposed for MsCCFP. The model, called ST-Attn for short, follows the general encoder-decoder framework for modeling sequential data but adopts a multiple-output strategy to preserve the correlations characterizing between each predicted step. The spatial-temporal attention mechanism learns to globally determine the focus on those parts of the city at certain periods that are more relevant to the predicted region and time period. Besides, a pre-predicted result calculated by spatiotemporal kernel density estimation is fed to ST-Attn, which provides a reference for further accurate predicting. Experiments on three real-world datasets are carried out to verify ST-Attn's performance and the results show that ST-Attn outperforms the baselines in terms of MsCCFP.
What problem does this paper attempt to address?