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.

Network traffic prediction with Attention-based Spatial-Temporal Graph Network

A Spatial-Temporal Gated Hypergraph Convolution Network for Traffic Prediction

Attention based spatiotemporal graph attention networks for traffic flow forecasting

A Spatiotemporal Graph Neural Network with Graph Adaptive and Attention Mechanisms for Traffic Flow Prediction

Adaptive spatial-temporal graph attention networks for traffic flow forecasting

Attention Based Spatial-Temporal Graph Convolutional Networks for Traffic Flow Forecasting

Spatio-Temporal Graph Neural Network for Traffic Prediction Based on Adaptive Neighborhood Selection

Adaptive Spatio-Temporal Convolutional Network for Traffic Prediction.

SAST-GNN: A Self-Attention Based Spatio-Temporal Graph Neural Network for Traffic Prediction

A Spatial–Temporal Attention Approach for Traffic Prediction

Spatio-temporal causal graph attention network for traffic flow prediction in intelligent transportation systems

A Self-Adaptive Spatio-Temporal Graph Convolutional Network for Traffic Prediction

Channel attention-based spatial-temporal graph neural networks for traffic prediction

Traffic Flow Prediction via Spatial Temporal Graph Neural Network

Traffic Prediction Model Based on Spatio-temporal Graph Attention Network

The Prediction of Multistep Traffic Flow Based on AST-GCN-LSTM

Attention Mechanism Based on Temporal Graph Convolutional Neural Network for Traffic Flow Prediction

Attention-based spatial-temporal graph transformer for traffic flow forecasting

A Spatio-Temporal Graph Neural Network Approach for Traffic Flow Prediction

STGIN: A Spatial Temporal Graph-Informer Network for Long Sequence Traffic Speed Forecasting

Road traffic flow prediction based on dynamic spatiotemporal graph attention network