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.

Graph Attention LSTM: A Spatiotemporal Approach for Traffic Flow Forecasting

Directed Hypergraph Attention Network for Traffic Forecasting

Attention-based Spatial-Temporal Graph Convolutional Recurrent Networks for Traffic Forecasting

Attention based spatiotemporal graph attention networks for traffic flow forecasting

A Graph-Based Temporal Attention Framework for Multi-Sensor Traffic Flow Forecasting

Adaptive Graph Attention and Long Short-Term Memory-Based Networks for Traffic Prediction

Adaptive spatial-temporal graph attention networks for traffic flow forecasting

Forecasting traffic flow with spatial–temporal convolutional graph attention networks

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

Short-Term Traffic Congestion Forecasting Using Attention-Based Long Short-Term Memory Recurrent Neural Network.

GT-LSTM: A spatio-temporal ensemble network for traffic flow prediction

Reinforced Spatiotemporal Attentive Graph Neural Networks for Traffic Forecasting.

Traffic Flow Prediction via Spatial Temporal Graph Neural Network

LST-GCN: Long Short-Term Memory Embedded Graph Convolution Network for Traffic Flow Forecasting

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

Traffic Graph Convolutional Recurrent Neural Network: A Deep Learning Framework for Network-Scale Traffic Learning and Forecasting

Attention-Based Multiple Graph Convolutional Recurrent Network for Traffic Forecasting

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

Spatial-Temporal Adaptive Graph Convolution with Attention Network for Traffic Forecasting

Multistep traffic speed prediction from multiple time-scale spatiotemporal features using graph attention network

T-LSTM: A Long Short-Term Memory Neural Network Enhanced by Temporal Information for Traffic Flow Prediction