Spatial-temporal encoder-decoder model for traffic flow prediction
CSTR:
Author:
Affiliation:

1.College of Information Science and Engineering, Hunan Normal University, Changsha 410081 , China ; 2.School of Computer Science, Changsha University of Science and Technology, Changsha 410114 , China

Clc Number:

TP391

Fund Project:

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • |
  • Materials
  • |
  • Comments
    Abstract:

    In order to solve the problem that many traffic flow prediction research methods are unable to comprehensively explore the dynamic hidden correlations in traffic data, the dynamic spatio-temporal variation characteristics were studied and an encoder-decoder-based traffic prediction model was proposed. In the model, both encoder and decoder mainly consisted of multi-head spatio-temporal attention mechanism modules, and a connection attention mechanism was added in between to analyze the spatio-temporal correlations of the road network. The model also used a dynamic embedding module consisting of a combination of both spatio-temporal embedding coding and adaptive graph convolution to analyze the dynamic and static information of nodes. Experiments on two real datasets demonstrate that the spatio-temporal model outperform other models for long-and short-term traffic prediction. Thus, the spatio-temporal encoder-decoder model can effectively handle complex spatio-temporal sequences and improve the traffic flow prediction accuracy.

    Reference
    Related
    Cited by
Get Citation

张锦, 皮煜, 孙程, 等. 面向交通流预测的时空编码器-解码器模型[J]. 国防科技大学学报, 2025, 47(3): 173-182.

Copy
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:February 13,2023
  • Revised:
  • Adopted:
  • Online: June 03,2025
  • Published:
Article QR Code