Skip to main navigation Skip to search Skip to main content

A pagerank self-attention network for traffic flow prediction

  • Ting Kang
  • , Huaizhi Wang*
  • , Ting Wu*
  • , Jianchun Peng
  • , Hui Jiang
  • *Corresponding author for this work
  • Shenzhen University
  • Harbin Institute of Technology Shenzhen

Research output: Contribution to journalArticlepeer-review

Abstract

Traffic information is collected from sensors in the urban road network, and traffic information can be said to be a mapping of people’s activities, which are difficult to model as a linear function, so this makes traffic information difficult to be predicted. In other words, traffic information is difficult to build effective models to predict traffic information because of its non-linear characteristics that are difficult to capture. As researchers go deeper, researchers have been able to extract good spatio-temporal features for modern urban road networks. However, it is worth mentioning that most researchers have neglected the importance of models for global potential features under the topology map of urban road networks, yet this global potential feature is very important for traffic prediction. In this paper, we propose a new spatio-temporal graph convolutional network model A Pagerank Self-attention Network (hereafter we abbreviate as PSN) in order to solve this problem based on a full consideration of the urban road network topology features, in which we employ a global spatio-temporal self-attention module to capture the global spatio-temporal features well. and the graph wandering module is used to propagate the spatio-temporal feature information effectively and widely. It is worth mentioning that experiments on two well-known datasets show that our proposed method achieves better prediction results compared to existing baseline methods.

Original languageEnglish
Article number948954
JournalFrontiers in Energy Research
Volume10
DOIs
StatePublished - 7 Sep 2022
Externally publishedYes

UN SDGs

This output contributes to the following UN Sustainable Development Goals (SDGs)

  1. SDG 7 - Affordable and Clean Energy
    SDG 7 Affordable and Clean Energy

Keywords

  • global spatio-temporal self-attention module
  • graph wandering module
  • spatio-temporal graph convolutional network
  • traffic forecast
  • urban road network

Fingerprint

Dive into the research topics of 'A pagerank self-attention network for traffic flow prediction'. Together they form a unique fingerprint.

Cite this