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

Gaozhong Tang, Bo Li*, Hong Ning Dai, Xi Zheng

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

1 Citation (Scopus)

Abstract

The capability of predicting the future trends of crowds has rendered crowd flow prediction 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 topography. Therefore, extracting and leveraging both spatial features and temporal features are key gradients for effectively predicting crowd flows. Many previous works extract spatial 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.

Original languageEnglish
Pages (from-to)19-34
Number of pages16
JournalInformation Sciences
Volume614
DOIs
Publication statusPublished - Oct 2022

Scopus Subject Areas

  • Theoretical Computer Science
  • Software
  • Control and Systems Engineering
  • Computer Science Applications
  • Information Systems and Management
  • Artificial Intelligence

User-Defined Keywords

  • Crowd flow prediction
  • Gated recurrent unit
  • Road structural information
  • Spatial feature
  • Temporal feature

Fingerprint

Dive into the research topics of 'SPRNN: A spatial–temporal recurrent neural network for crowd flow prediction'. Together they form a unique fingerprint.

Cite this