Spatiotemporal Attention-Based Graph Convolution Network for Segment-Level Traffic Prediction | IEEE Journals & Magazine | IEEE Xplore

Spatiotemporal Attention-Based Graph Convolution Network for Segment-Level Traffic Prediction


Abstract:

Traffic prediction, as a core component of intelligent transportation systems (ITS), has been investigated thoroughly in the literature. Nevertheless, timely accurate tra...Show More

Abstract:

Traffic prediction, as a core component of intelligent transportation systems (ITS), has been investigated thoroughly in the literature. Nevertheless, timely accurate traffic prediction still remains an open challenge due to the nonlinearities and complex patterns of traffic flows. In addition, most of the existing traffic prediction methods focus on grid-based computing problems (e.g., crowd in-out flow prediction) and point-based computing problems (e.g., traffic detector data prediction), ignoring the segment-based traffic prediction tasks. In this study, we propose an attention-based spatiotemporal graph attention network (AST-GAT) for segment-level traffic speed prediction. In particular, a multi-head graph attention block is designed to capture the spatial dependencies among road segments. Then, a component fusion block is built for speed, volume, and weather information integration. Finally, an attention-based Long short-term memory (LSTM) block is constructed for temporal dependency learning as well as segment-based speed prediction. Experiments on a real-world dataset from Highways England demonstrate that the proposed AST-GAT model outperforms the state-of-the-art baselines, providing an efficient tool for segment-based traffic prediction and therefore filling the gap between point-based and grid-based predictions.
Published in: IEEE Transactions on Intelligent Transportation Systems ( Volume: 23, Issue: 7, July 2022)
Page(s): 8337 - 8345
Date of Publication: 26 May 2021

ISSN Information:

Funding Agency:


References

References is not available for this document.