SPRNN: A spatial-temporal recurrent neural network for crowd flow prediction

被引:7
|
作者
Tang, Gaozhong [1 ]
Li, Bo [1 ]
Dai, Hong-Ning [2 ]
Zheng, Xi [3 ]
机构
[1] South China Univ Technol, Sch Elect & Informat Engn, Guangzhou, Peoples R China
[2] Hong Kong Baptist Univ, Dept Comp Sci, Hong Kong, Peoples R China
[3] Macquarie Univ, Dept Comp, Sydney, Australia
基金
中国国家自然科学基金; 澳大利亚研究理事会;
关键词
Crowd flow prediction; Spatial feature; Temporal feature; Road structural information; Gated recurrent unit; MODEL; DEEP;
D O I
10.1016/j.ins.2022.09.053
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
The capability of predicting the future trends of crowds has rendered crowd flow predic-tion more critical in building intelligent transportation systems, and attracted substantial research efforts. The trend of crowd flows is closely related to time and the urban topog-raphy. Therefore, extracting and leveraging both spatial features and temporal features are key gradients for effectively predicting crowd flows. Many previous works extract spa-tial features from crowd-flow data in an iteration way. As a result, models suffer from a heavy computation cost while ignoring details of road topology and structure information. Meanwhile, temporal features, including short-term features and long-term features, are separately extracted. The fusion of all features at the last stage before accomplishing the prediction also neglects the underlying associativity between various features. To address the limitations, we leverage spatial features by extracting structural information of road structures, such as road connection, road density, road width, etc. Rather than extracting spatial features from crowd-flow data, we capture them from images of city maps by adopting convolutional neural networks. Moreover, we implement a new sequence feature fusion mechanism to merge both spatial features and temporal features from various time scales so as to predict crowd flows. We conduct extensive experiments to evaluate our model on three benchmark datasets. The experimental results demonstrate that the model outperforms 15 state-of-the-art methods. The source code is available at: https:// github.com/CVisionProcessing/SPRNN.(c) 2022 Elsevier Inc. All rights reserved.
引用
收藏
页码:19 / 34
页数:16
相关论文
共 50 条
  • [1] A Spatial-Temporal Recurrent Neural Network for Video Saliency Prediction
    Zhang, Kao
    Chen, Zhenzhong
    Liu, Shan
    IEEE TRANSACTIONS ON IMAGE PROCESSING, 2021, 30 : 572 - 587
  • [2] An effective spatial-temporal attention based neural network for traffic flow prediction
    Do, Loan N. N.
    Vu, Hai L.
    Vo, Bao Q.
    Liu, Zhiyuan
    Dinh Phung
    TRANSPORTATION RESEARCH PART C-EMERGING TECHNOLOGIES, 2019, 108 : 12 - 28
  • [3] Multi-task Adversarial Spatial-Temporal Networks for Crowd Flow Prediction
    Wang, Senzhang
    Miao, Hao
    Chen, Hao
    Huang, Zhiqiu
    CIKM '20: PROCEEDINGS OF THE 29TH ACM INTERNATIONAL CONFERENCE ON INFORMATION & KNOWLEDGE MANAGEMENT, 2020, : 1555 - 1564
  • [4] Spatial-temporal Prediction of Air Quality based on Recurrent Neural Networks
    Sun, Xiaotong
    Xu, Wei
    Jiang, Hongxun
    PROCEEDINGS OF THE 52ND ANNUAL HAWAII INTERNATIONAL CONFERENCE ON SYSTEM SCIENCES, 2019, : 1265 - 1274
  • [5] Multisize Patched Spatial-Temporal Transformer Network for Short-and Long-Term Crowd Flow Prediction
    Xie, Yulai
    Niu, Jingjing
    Zhang, Yang
    Ren, Fang
    IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2022, 23 (11) : 21548 - 21568
  • [6] STAGNN: a spatial-temporal attention graph neural network for network traffic prediction
    Luo, Yonghua
    Ning, Qian
    Chen, Bingcai
    Zhou, Xinzhi
    Huang, Linyu
    INTERNATIONAL JOURNAL OF COMMUNICATION NETWORKS AND DISTRIBUTED SYSTEMS, 2024, 30 (04) : 413 - 432
  • [7] Spatial-Temporal Traffic Flow Prediction With Fusion Graph Convolution Network and Enhanced Gated Recurrent Units
    Cai, Chuang
    Qu, Zhijian
    Ma, Liqun
    Yu, Lianfei
    Liu, Wenbo
    Ren, Chongguang
    IEEE ACCESS, 2024, 12 : 56477 - 56491
  • [8] STTD: spatial-temporal transformer with double recurrent graph convolutional cooperative network for traffic flow prediction
    Zeng, Hui
    Cui, Qiang
    Huang, XiaoHui
    Duan, XueWei
    CLUSTER COMPUTING-THE JOURNAL OF NETWORKS SOFTWARE TOOLS AND APPLICATIONS, 2024, 27 (09): : 12069 - 12089
  • [9] Landslide Displacement Prediction With Gated Recurrent Unit and Spatial-Temporal Correlation
    Ma, Wenli
    Dong, Jianhui
    Wei, Zhanxi
    Peng, Liang
    Wu, Qihong
    Chen, Chunxia
    Wu, Yuanzao
    Xie, Feihong
    FRONTIERS IN EARTH SCIENCE, 2022, 10
  • [10] Traffic Network Speed Prediction via Multi-periodic-component Spatial-temporal Neural Network
    Yang J.-X.
    Yu C.-S.
    Li R.
    Du L.-F.
    Jiang S.-X.
    Wang D.
    Jiaotong Yunshu Xitong Gongcheng Yu Xinxi/Journal of Transportation Systems Engineering and Information Technology, 2021, 21 (03): : 112 - 119and139