Loading [MathJax]/extensions/MathMenu.js
SA-STT: A Structure-Aware Spatio-Temporal Transformer for Traffic Prediction | IEEE Conference Publication | IEEE Xplore

SA-STT: A Structure-Aware Spatio-Temporal Transformer for Traffic Prediction


Abstract:

Predicting traffic flow is central to alleviating congestion, optimizing routes, and supporting traffic planning and urban management. However, this task remains challeng...Show More

Abstract:

Predicting traffic flow is central to alleviating congestion, optimizing routes, and supporting traffic planning and urban management. However, this task remains challenging, particularly given the vast and intricate topologies of urban traffic networks. This paper presents a novel model-Structure Aware Spatio-Temporal Transformer (SASTT)-that leverages graph convolutional networks, attention mechanisms, and traffic transfer embedding to effectively capture complex spatio-temporal dependencies in traffic data. SASTT employs a local information sampling approach before computing global relevance, ensuring a robust representation of traffic propagation features. Furthermore, we develop a traffic transfer embedding method that aligns with the spatial structure of traffic data by incorporating road network information. Experiments conducted on four real-world datasets from different cities and compared with several state-of-the-art baselines show that our model achieves significant improvements in terms of mean absolute error (MAE) and root mean squared error (RMSE), confirming the effectiveness of our method.
Date of Conference: 17-19 November 2023
Date Added to IEEE Xplore: 08 April 2024
ISBN Information:
Conference Location: Fuzhou, China

Contact IEEE to Subscribe

References

References is not available for this document.