Window Spatio-temporal Attention Modeling Approach for Traffic Prediction
Author:
  • Article
  • | |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • | |
  • Comments
    Abstract:

    Currently, in traffic prediction, deep learning-based spatio temporal separation modeling methods have difficulty in expressing spatio-temporal coupling correlations in data effectively. Although spatio-temporal joint modeling methods can compensate for the shortcomings of spatio-temporal separation modeling to some extent, there are deficiencies such as insufficient express ability and high computational complexity in constructing spatio-temporal hypergraphs. To address these issues, this study proposes an improved spatio-temporal joint modeling method, window spatial-temporal attention network (W-STANet). W-STANet mainly comprises three parts: a data embedding layer, a spatio-temporal correlation modeling layer, and a prediction head. The spatio-temporal correlation modeling layer learns spatio-temporal correlation features of traffic data by stacking multiple spatio-temporal attention blocks. Meanwhile, by introducing the local window calculation method, data shifting and permutation operations, the computational complexity in the modeling process is greatly reduced, and the modeling from both local and global perspectives within the spatio temporal graph is achieved. Experimental results on five real traffic public datasets demonstrate superior prediction performance compared to other spatio-temporal joint modeling methods. Compared with spatio-temporal separation modeling methods, it has superior prediction performance on large-scale road network datasets.

    Reference
    Related
    Cited by
Get Citation

王嘉俊,李勇,欧丽云.面向交通预测的窗口时空注意力建模方法.计算机系统应用,,():1-9

Copy
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:October 17,2024
  • Revised:November 08,2024
  • Online: February 26,2025
Article QR Code
You are the first991371Visitors
Copyright: Institute of Software, Chinese Academy of Sciences Beijing ICP No. 05046678-3
Address:4# South Fourth Street, Zhongguancun,Haidian, Beijing,Postal Code:100190
Phone:010-62661041 Fax: Email:csa (a) iscas.ac.cn
Technical Support:Beijing Qinyun Technology Development Co., Ltd.

Beijing Public Network Security No. 11040202500063