HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

PDG2Seq: Periodic Dynamic Graph to Sequence Model for Traffic Flow Prediction

{Jia Wu Huifeng Wu Qikai Chen Wenchao Weng Jin Fan}

Abstract

Traffic flow prediction is the foundation of intelligent traffic management systems. Current methods prioritize the development of intricate models to capture spatio-temporal correlations, yet they often neglect the exploitation of latent features within traffic flow. Firstly, the correlation among different road nodes exhibits dynamism rather than remaining static. Secondly, traffic data exhibits evident periodicity, yet current research lacks the exploration and utilization of periodic features. Lastly, current models typically rely solely on historical data for modeling, resulting in the limitation of accurately capturing future trend changes in traffic flow. To address these findings, this paper proposes a Periodic Dynamic Graph to Sequence Model (PDG2Seq) for traffic flow prediction. PDG2Seq consists of the Periodic Feature Selection Module (PFSM) and the Periodic Dynamic Graph Convolutional Gated Recurrent Unit (PDCGRU) to further extract the spatio-temporal features of the dynamic real-time traffic. The PFSM extracts learned periodic features using time points as indices, while the PDCGRU leverages the extracted periodic features from the PFSM and dynamic features from traffic flow to generate a Periodic Dynamic Graph for extracting spatio-temporal features. In the decoding phase, PDG2Seq utilizes periodic features corresponding to the prediction target to capture future trend changes, leading to more accurate predictions. Comprehensive experiments conducted on four large-scale datasets substantiate the superiority of PDG2Seq over existing state-of-the-art baselines. Related codes are available at https://github.com/wengwenchao123/PDG2Seq.

Benchmarks

BenchmarkMethodologyMetrics
traffic-prediction-on-pems04PDG2Seq
12 Steps MAE: 18.24
traffic-prediction-on-pems07PDG2Seq
MAE@1h: 19.28
traffic-prediction-on-pems08PDG2Seq
MAE@1h: 13.60
traffic-prediction-on-pemsd3PDG2Seq
12 steps MAE: 14.62
12 steps MAPE: 14.88
12 steps RMSE: 25.47
traffic-prediction-on-pemsd4PDG2Seq
12 steps MAE: 18.24
12 steps MAPE: 12.09
12 steps RMSE: 30.08
traffic-prediction-on-pemsd8PDG2Seq
12 steps MAE: 13.60
12 steps MAPE: 8.99
12 steps RMSE: 23.37

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
PDG2Seq: Periodic Dynamic Graph to Sequence Model for Traffic Flow Prediction | Papers | HyperAI