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.

Traffic Flow Prediction Based on Self-attention Mechanism and Deep Packet Residual Network.

Short-Term Urban Traffic Flow Prediction Using Deep Spatio-Temporal Residual Networks

A Hybrid Deep Learning Based Traffic Flow Prediction Method and Its Understanding

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

Short-Term Traffic Flow Prediction via Improved Mode Decomposition and Self-Attention Mechanism Based Deep Learning Approach

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

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

Urban Road Traffic Flow Prediction with Attention-Based Convolutional Bidirectional Long Short-Term Memory Networks

A deep learning method based on an attention mechanism for wireless network traffic prediction

Traffic Flow Prediction with Attention Mechanism Based on TS-NAS

Attention Based Convolutional Networks for Traffic Flow Prediction

A pagerank self-attention network for traffic flow prediction

Attention Mechanism With Spatial-Temporal Joint Model for Traffic Flow Speed Prediction

Traffic Flow Prediction Based on Spatio-temporal Attention Block

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

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

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

An Efficient Traffic Prediction Model Using Deep Spatial-Temporal Network

Predicting Citywide Crowd Flows Using Deep Spatio-Temporal Residual Networks

A Deep Learning Framework Based on Spatio-Temporal Attention Mechanism for Traffic Prediction.

An effective spatial-temporal attention based neural network for traffic flow prediction