Multi-Scale Spatio-Temporal Attention Networks for Network-Scale Traffic Learning and Forecasting.

Cong Wu, Hui Ding, Zhongwang Fu, Ning Sun
Author Information
  1. Cong Wu: Engineering Research Center of Wideband Wireless Communication Technology, Ministry of Education, Nanjing University of Posts and Telecommunications, Nanjing 210003, China. ORCID
  2. Hui Ding: State Key Laboratory of Air Traffic Management System, Nanjing 210014, China.
  3. Zhongwang Fu: Engineering Research Center of Wideband Wireless Communication Technology, Ministry of Education, Nanjing University of Posts and Telecommunications, Nanjing 210003, China.
  4. Ning Sun: Engineering Research Center of Wideband Wireless Communication Technology, Ministry of Education, Nanjing University of Posts and Telecommunications, Nanjing 210003, China. ORCID

Abstract

Accurate and timely forecasting of traffic on local road networks is crucial for deploying effective dynamic traffic control, advanced route planning, and navigation services. This task is particularly challenging due to complex spatio-temporal dependencies arising from non-Euclidean spatial relations in road networks and non-linear temporal dynamics influenced by changing road conditions. This paper introduces the spatio-temporal network embedding (STNE) model, a novel deep learning framework tailored for learning and forecasting graph-structured traffic data over extended input sequences. Unlike traditional convolutional neural networks (CNNs), the model employs graph convolutional networks (GCNs) to capture the spatial characteristics of local road network topologies. Moreover, the segmentation of very long input traffic data into multiple sub-sequences, based on significant temporal properties such as closeness, periodicity, and trend, is performed. Multi-dimensional long short-term memory neural networks (MDLSTM) are utilized to flexibly access multi-dimensional context. Experimental results demonstrate that the STNE model surpasses state-of-the-art traffic forecasting benchmarks on two large-scale real-world traffic datasets.

Keywords

References

  1. IEEE Trans Neural Netw Learn Syst. 2017 Oct;28(10):2371-2381 [PMID: 27448371]
  2. Sensors (Basel). 2017 Jun 26;17(7): [PMID: 28672867]
  3. Sensors (Basel). 2024 Apr 07;24(7): [PMID: 38610556]
  4. Sensors (Basel). 2017 Apr 10;17(4): [PMID: 28394270]
  5. Sensors (Basel). 2024 Apr 03;24(7): [PMID: 38610486]
  6. Sensors (Basel). 2023 Jan 11;23(2): [PMID: 36679639]

Grants

  1. SKLATM202004/State Key Laboratory of Air Traffic Management System, China