HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Multi-Head Spatiotemporal Attention Graph Convolutional Network for Traffic Prediction

{Rauf HT Yu Y Sarfraz MS Qin Z Aftab MU Oluwasanmi A}

Abstract

Intelligent transportation systems (ITSs) have become an indispensable component of modern global technological development, as they play a massive role in the accurate statistical estimation of vehicles or individuals commuting to a particular transportation facility at a given time. This provides the perfect backdrop for designing and engineering an adequate infrastructural capacity for transportation analyses. However, traffic prediction remains a daunting task due to the non-Euclidean and complex distribution of road networks and the topological constraints of urbanized road networks. To solve this challenge, this paper presents a traffic forecasting model which combines a graph convolutional network, a gated recurrent unit, and a multi-head attention mechanism to simultaneously capture and incorporate the spatio-temporal dependence and dynamic variation in the topological sequence of traffic data effectively. By achieving 91.8% accuracy on the Los Angeles highway traffic (Los-loop) test data for 15-min traffic prediction and an R2 score of 85% on the Shenzhen City (SZ-taxi) test dataset for 15- and 30-min predictions, the proposed model demonstrated that it can learn the global spatial variation and the dynamic temporal sequence of traffic data over time. This has resulted in state-of-the-art traffic forecasting for the SZ-taxi and Los-loop datasets.

Benchmarks

BenchmarkMethodologyMetrics
traffic-prediction-on-sz-taxiMHAST-GCN
MAE @ 15min: 2.6624
MAE @ 30min: 2.7101
MAE @ 45min: 2.7221
MAE @ 60min: 2.7361

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
Multi-Head Spatiotemporal Attention Graph Convolutional Network for Traffic Prediction | Papers | HyperAI