Research on traffic flow prediction of dynamic graph convolution network based on spatio-temporal attention mechanism

Accurate traffic flow prediction is the basis of urban traffic guidance and control, which is of great significance to intelligent traffic management and control. The complex spatial and temporal dependencies of traffic flow, as well as the periodicity and spatial heterogeneity of capturing traffic data, make accurate traffic flow prediction still a major challenge. To address the above problems, this article proposes DGCN-STA, a dynamic graph convolutional network for traffic flow prediction based on a spatio-temporal attention mechanism. Specifically, in the time dimension, this article constructs a new self-attention mechanism that can utilize local contexts, specifically for numerical sequence representation transformation, so that the prediction model is able to capture the temporal correlation of traffic flow, which is conducive to long-term prediction. In the spatial dimension, a dynamic graph convolution method is constructed to capture spatial correlations in a dynamic manner using self-attention, modeling for the periodicity of traffic flows and capturing spatial heterogeneity through embedding modules. The spatio-temporal correlation of long time series is then modeled using the spatio-temporal attention mechanism. Many experiments are conducted on four datasets, and the experimental results show that the DGCN-STA model proposed in this article has better prediction performance compared with the baseline method.

• Publication year

  2026

• Venue

  Proceedings of the Institution of mechanical engineers. Part D, journal of automobile engineering

• Publication date

  2026-03-01

• Fields of study

  Not labeled

• Identifiers
  DOI 10.1177/09544070261416597
• External record

  Open on Semantic Scholar

• Source metadata

  Semantic Scholar

• No claims are published for this paper.

• No concepts are published for this paper.

• Adaptive Decision Spatio-temporal neural ODE for traffic flow forecasting with Multi-Kernel Temporal Dynamic Dilation Convolution

  2024cited by this paper
• Spatial-Temporal Complex Graph Convolution Network for Traffic Flow Prediction

  2023cited by this paper
• DiffSTG: Probabilistic Spatio-Temporal Graph Forecasting with Denoising Diffusion Models

  2023cited by this paper
• A dynamic spatial-temporal deep learning framework for traffic speed prediction on large-scale road networks

  2022cited by this paper
• Graph Neural Controlled Differential Equations for Traffic Forecasting

  2021cited by this paper
• Spatial-Temporal Synchronous Graph Convolutional Networks: A New Framework for Spatial-Temporal Network Data Forecasting

  2020cited by this paper
• Graph WaveNet for Deep Spatial-Temporal Graph Modeling

  2019cited by this paper
• Attention Based Spatial-Temporal Graph Convolutional Networks for Traffic Flow Forecasting

  2019cited by this paper
• Road traffic state prediction based on a graph embedding recurrent neural network under the SCATS.

  2019cited by this paper
• GMAN: A Graph Multi-Attention Network for Traffic Prediction

  2019cited by this paper
• Deeper Insights into Graph Convolutional Networks for Semi-Supervised Learning

  2018cited by this paper
• Spatio-Temporal Graph Convolution for Skeleton Based Action Recognition

  2018cited by this paper
• Spatio-temporal Graph Convolutional Neural Network: A Deep Learning Framework for Traffic Forecasting

  2017cited by this paper
• Short-Term Forecasting of Passenger Demand under On-Demand Ride Services: A Spatio-Temporal Deep Learning Approach

  2017cited by this paper
• Sequence to Sequence Learning with Neural Networks

  2014cited by this paper
• A Summary of Traffic Flow Forecasting Methods

  2004cited by this paper
• Freeway Performance Measurement System: Mining Loop Detector Data

  2001cited by this paper
• Long Short-Term Memory

  1997cited by this paper

• No citing papers are available for this paper.