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.
Spatiotemporal Attention-Based Graph Convolution Network for Segment-Level Traffic Prediction
IEEE Transactions on Intelligent Transportation Systems ; 23 , 7 ; 8337-8345
2022-07-01
1608373 byte
Article (Journal)
Electronic Resource
English
Traffic Flow Prediction Based on Multi-Spatiotemporal Attention Gated Graph Convolution Network
DOAJ | 2022
|Traffic flow prediction method based on graph attention convolution network
European Patent Office | 2020
|Road network traffic flow prediction method based on gated loop graph convolution attention network
European Patent Office | 2023
|