ST-ABC: Spatio-Temporal Attention-Based Convolutional Network for Multi-Scale Lane-Level Traffic Prediction

Abstract

With the widespread application of intelligent transportation systems and navigation software, traffic prediction should be modeled in finer granularity to facilitate lane-changing guidance and congestion mitigation. However, existing studies divide the road network into continuous segments which assumes different lanes share the same spatio-temporal patterns. This paper proposes a novel lightweight, attention-based, fully convolutional model, named the Spatio-Temporal Attention- Based Convolutional network (ST-ABC), where lane segments are treated as graph nodes and dynamically models the adjacent spatial dependencies using local attention graph convolution. The attention-based dilated convolutions can process longer sequence periods in parallel, and a global attention layer allows individual nodes to be associated with the global context. By setting a target window, it can further reduce unnecessary computations and improve the prediction effect for the targeted area. Further-more, the ST-ABC model facilitates the simultaneous integration of spatio-temporal information and relational distance metrics among lane segments, enriching the granularity of multi-scaled spatial prediction. Empirical evaluations conducted on two real-world datasets substantiate the augmented efficacy of the STABC model in comparison to established models, with a marked prominence in long-term prediction scenarios.

Topics & Concepts