Trasnet : A Lightweighting Spatio-temporal Attention Network for Traffic Flow Prediction

Minghao Li,Xuxiang Ta,Chao Chen
DOI: https://doi.org/10.1109/ijcnn60899.2024.10651231
2024-01-01
Abstract:Accurate traffic prediction is a significant challenge in the development of intelligent transportation systems. The ability to capture crucial spatio-temporal traffic information plays a crucial role in the accuracy of predictions. In recent years, many complex neural networks have been proposed to address this issue. However, intricate network architectures have resulted in lower performance. In this paper, we introduce a spatio-temporal graph neural network, Trasnet, which employs spatio-temporal attention convolution to capture complex spatio-temporal information. Additionally, we propose a graph learning module to learn spatio-temporal dependencies from both global and local perspectives. We conduct extensive experiments on real-world datasets to validate the effectiveness of our approach.
What problem does this paper attempt to address?