Multi-Dimensional Attention Based Spatial-Temporal Networks for Traffic Forecasting

Traffic flow prediction is the key problem of intelligent transportation system. Accurate prediction results are indispensable for traffic management and road planning. However, due to the complex spatial-temporal correlation of traffic flow data, including the spatial correlation and temporal corre...

Full description

Saved in:
Bibliographic Details
Published in:Wireless communications and mobile computing 2022-09, Vol.2022, p.1-13
Main Authors: Xu, Guangxia, Hu, Xinting
Format: Article
Language:English
Subjects:
Ablation
Accuracy
Correlation
Deep learning
Dynamic models
Effectiveness
Engineering
Feature extraction
Forecasting
Intelligent transportation systems
Machine learning
Natural language
Neural networks
Research methodology
Roads & highways
Spatial data
Time series
Traffic control
Traffic flow
Traffic management
Traffic models
Traffic planning
Transportation networks
Citations: Items that this one cites
Items that cite this one
Online Access:Get full text
Tags: Add Tag
No Tags, Be the first to tag this record!
Description
Summary:Traffic flow prediction is the key problem of intelligent transportation system. Accurate prediction results are indispensable for traffic management and road planning. However, due to the complex spatial-temporal correlation of traffic flow data, including the spatial correlation and temporal correlation of adjacency, periodicity, and trend that exist between different roads. The existing forecasting methods consider the spatial-temporal correlation but lack the dynamic modeling of spatial-temporal correlation. To deal with this dynamic feature, this paper proposes a multi-dimensional attention-based spatial-temporal network (MA-STN). It mainly contains three parts, the spatial-temporal attention unit, the spatial-temporal feature extraction unit based on Graph Convolutional Network (GCN) and the fusion prediction unit, and the residual connection is also added to the model to avoid the gradient disappearance problem. Meanwhile, this paper divides the dataset into three subsets to deal with the three features in the temporal dimension separately. To verify the effectiveness of the proposed model, two real-world road traffic flow data collected by PeMS system are used for validation. By comparing six different models, the proposed network in this paper has a 7% accuracy improvement compared to the baseline model. To verify the effectiveness of the attention mechanism, ablation experiments are used in this paper for validation, and the results show that the attention mechanism can achieve a 5% accuracy improvement.
ISSN:1530-8669
1530-8677
DOI:10.1155/2022/1358535