Abstract:Traffic flow forecasting is a crucial task in transportation and necessary for congestion mitigation, traffic control, and intelligent traffic management. Deep learning models can aid in high-accuracy traffic flow forecasting; however, the current research focuses only the ability of the model to capture dynamic spatiotemporal features, and studies on the effect of deeper network layers on spatiotemporal features—a critical factor affecting traffic flow forecasting accuracy—are limited. In this paper, we propose an attention-based spatiotemporal graph attention network (ASTGAT) model designed for network degradation and over-smoothing problems to investigate in-depth spatiotemporal information. Compared to other networks, ASTGAT can capture dynamic spatiotemporal correlations in data and deepen the network to improve prediction accuracy through multiple residual convolution and high-low feature concat. ASTGAT comprises three components that separately model the temporal relationships of the recent, daily, and weekly periods. Each component stacks multiple spatiotemporal blocks constructed using the attention mechanism, dilated gated convolution, and graph attention network. The graph and temporal attention layers capture spatiotemporal information dynamically, and the graph attention layer alleviates the over-smoothing phenomenon to deepen the network. The combined utilization of the attention mechanism and dilated gated convolution layer improves the medium and long temporal span prediction ability. We validated ASTGAT using two open highway datasets, and the results demonstrated that our ASTGAT model effectively extracts in-depth spatiotemporal information and the prediction results outperform those predicted by the current eight baselines. Our research is dedicated to establishing a better scientific basis for intelligent traffic management that can assist in decision making.

Traffic Flow Prediction Based on Spatio-temporal Attention Block

Short-Term Traffic Flow Prediction using Attention-Based Long Short-Term Memory Network

Deep spatio-temporal neural network based on interactive attention for traffic flow prediction

Road traffic flow prediction based on dynamic spatiotemporal graph attention network

Spatial–temporal combination and multi-head flow-attention network for traffic flow prediction

Unified Spatial-Temporal Neighbor Attention Network for Dynamic Traffic Prediction

ST-D3DDARN: Urban traffic flow prediction based on spatio-temporal decoupled 3D DenseNet with attention ResNet

Attention based spatiotemporal graph attention networks for traffic flow forecasting

Spatio-Temporal Traffic Flow Prediction Based on Coordinated Attention

Spatial-Temporal Multi-Head Attention Networks for Traffic Flow Forecasting.

Weather Interaction-Aware Spatio-Temporal Attention Networks for Urban Traffic Flow Prediction

Traffic Flow Prediction via Spatial Temporal Graph Neural Network

Traffic Prediction Model Based on Spatio-temporal Graph Attention Network

A traffic flow-forecasting model based on multi-head spatio–temporal attention and adaptive graph convolutional networks

A Spatial–Temporal Attention Approach for Traffic Prediction

Dynamic multi-scale spatial-temporal graph convolutional network for traffic flow prediction

Forecasting traffic flow with spatial–temporal convolutional graph attention networks

A Hybrid Deep Learning Model With Attention-Based Conv-LSTM Networks for Short-Term Traffic Flow Prediction

Interpretable Crowd Flow Prediction with Spatial-Temporal Self-Attention

Dynamic Spatial-Temporal Representation Learning for Traffic Flow Prediction

Efficient Adaptive Spatial-Temporal Attention Network for Traffic Flow Forecasting