Abstract:Traffic flow prediction is considered to be one of the fundamental technologies in intelligent transportation systems (ITSs) with a tremendous application prospect. Unlike traditional time series analysis tasks, the key challenge in traffic flow prediction lies in effectively modelling the highly complex and dynamic spatiotemporal dependencies within the traffic data. In recent years, researchers have proposed various methods to enhance the accuracy of traffic flow prediction, but certain issues still persist. For instance, some methods rely on specific static assumptions, failing to adequately simulate the dynamic changes in the data, thus limiting their modelling capacity. On the other hand, some approaches inadequately capture the spatiotemporal dependencies, resulting in the omission of crucial information and leading to unsatisfactory prediction outcomes. To address these challenges, this paper proposes a model called the Dynamic Spatial–Temporal Self-Attention Network (DSTSAN). Firstly, this research enhances the interaction between different dimension features in the traffic data through a feature augmentation module, thereby improving the model’s representational capacity. Subsequently, the current investigation introduces two masking matrices: one captures local spatial dependencies and the other captures global spatial dependencies, based on the spatial self-attention module. Finally, the methodology employs a temporal self-attention module to capture and integrate the dynamic temporal dependencies of traffic data. We designed experiments using historical data from the previous hour to predict traffic flow conditions in the hour ahead, and the experiments were extensively compared to the DSTSAN model, with 11 baseline methods using four real-world datasets. The results demonstrate the effectiveness and superiority of the proposed approach.

GraphSANet: A Graph Neural Network and Self Attention Based Approach for Spatial Temporal Prediction in Sensor Network

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

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

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

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

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

Traffic Flow Prediction via Spatial Temporal Graph Neural Network

Graph Attention Spatial-Temporal Network for Deep Learning Based Mobile Traffic Prediction.

Spatial-Temporal Graph Attention Model on Traffic Forecasting.

Sampling Spatial-Temporal Attention Network for Traffic Forecasting.

Forecasting traffic flow with spatial–temporal convolutional graph attention networks

Spatial-Temporal Self-Attention Network for Flow Prediction

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

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

A mobility aware network traffic prediction model based on Dynamic Graph Attention Spatio-temporal Network

STNN: A Spatial-Temporal Graph Neural Network for Traffic Prediction.

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

A Spatial–Temporal Attention Approach for Traffic Prediction

Dynamic Spatial–Temporal Self-Attention Network for Traffic Flow Prediction

Spatio-Temporal Attention-based Graph Convolution Networks for Traffic Prediction

STA-GCN: Spatial-Temporal Self-Attention Graph Convolutional Networks for Traffic-Flow Prediction