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 Forecasting Based on Adaptive Graph Structure for Spatio-Temporal Attention Mechanism Network

Adaptive Spatio-temporal Graph Neural Network for traffic forecasting

A Short-term Traffic Flow Forecasting Model Based on Spatial-temporal Attention Neural Network

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

Adaptive spatial-temporal graph attention networks for traffic flow forecasting

Learning Dynamics and Heterogeneity of Spatial-Temporal Graph Data for Traffic Forecasting

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

Spatiotemporal Synchronous Dynamic Graph Attention Network for Traffic Flow Forecasting

Dynamic Multi-Granularity Spatial-Temporal Graph Attention Network for Traffic Forecasting

Attention based spatiotemporal graph attention networks for traffic flow forecasting

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

Make More Connections: Urban Traffic Flow Forecasting with Spatiotemporal Adaptive Gated Graph Convolution Network

Spatial-temporal graph neural network based on gated convolution and topological attention for traffic flow prediction

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

A Novel Attention-Based Dynamic Multi-Graph Spatial-Temporal Graph Neural Network Model for Traffic Prediction

Multi-stage attention spatial-temporal graph networks for traffic prediction

Spatiotemporal Adaptive Fusion Graph Network for Short-Term Traffic Flow Forecasting

Graph Attention Network with Spatial-Temporal Clustering for Traffic Flow Forecasting in Intelligent Transportation System

Spatial-Temporal Graph Neural Network Framework with Multi-source Local and Global Information Fusion for Traffic Flow Forecasting

STAGNN: A Spatial-Temporal Attention Graph Neural Network for Network Traffic Prediction

Spatial-Temporal Fusion Graph Neural Networks for Traffic Flow Forecasting