STP-TrellisNets+: Spatial-Temporal Parallel TrellisNets for Multi-Step Metro Station Passenger Flow Prediction

Junjie Ou,Jiahui Sun,Yichen Zhu,Haiming Jin,Yijuan Liu,Fan Zhang,Jianqiang Huang,Xinbing Wang
DOI: https://doi.org/10.1109/tkde.2022.3187690
IF: 9.235
2022-01-01
IEEE Transactions on Knowledge and Data Engineering
Abstract:The drastic increase of metro passengers in recent years inevitably causes the overcrowdedness in the metro systems. Accurately predicting passenger flows at metro stations is critical for efficient metro system management, which helps alleviate such overcrowdedness. Compared to the prevalent next-step prediction, multi-step passenger flow prediction could prominently increase the prediction duration and reveal finer-grained passenger flow variations, which better helps metro system management. Thus, in this paper, we address the problem of multi-step metro station passenger (MSP) flow prediction. In light of MSP flows’ unique spatial-temporal characteristics, we propose STP-TrellisNets+, which for the first time augments the newly-emerged temporal convolutional framework TrellisNet for multi-step MSP flow prediction. The temporal module of STP-TrellisNets+ (named CP-TrellisNetsED) employs a Closeness TrellisNet followed by a Periodicity TrellisNets-based Encoder-Decoder (P-TrellisNetsED) to jointly capture the short- and long-term temporal correlation of MSP flows. In parallel to CP-TrellisNetsED, its spatial module (named GC-TrellisNetsED) adopts a novel transfer flow-based metric to characterize the spatial correlation among MSP flows, and implements another TrellisNetsED on multiple diffusion graph convolutional networks (DGCNs) in time-series order to capture the dynamics of such spatial correlation. Extensive experiments with two large-scale real-world automated fare collection datasets demonstrate that STP-TrellisNets+ outperforms the state-of-the-art baselines.
computer science, information systems, artificial intelligence,engineering, electrical & electronic
What problem does this paper attempt to address?