Recent works demonstrate that capturing correlations between road network nodes is crucial to improving traffic flow forecasting accuracy. In general, there are spatial, temporal, and joint spatial-temporal correlations between two nodes, whose strength is related to spatial and temporal position factors. For example, traffic congestion that occurs at a traffic hub has a wider and stronger impact than that at a branch road. Moreover, the above impacts can vary with temporal position. Although spatial-temporal graph convolution networks have become a popular paradigm for modeling those correlations, there are still three problems with existing models: (i) failing to effectively model joint spatial-temporal correlations; (ii) ignoring spatial and temporal position factors when modeling the aforementioned correlations; and (iii) failing to capture distinct spatial-temporal patterns of each node. To cope with the above issues, this paper proposes a novel Spatial-Temporal Position-aware Graph Convolution Network (STPGCN) for traffic flow forecasting. Specifically, a trainable embedding module is constructed to represent the spatial and temporal positions of the nodes. Subsequently, a spatial-temporal position-aware relation inference module is proposed to adaptively infer the correlation weights of the three important spatial-temporal relations. Based on this, the generated spatial-temporal relations are integrated into a graph convolution layer for aggregating and updating node features. Finally, we design a spatial-temporal position-aware gated activation unit in the graph convolution, to capture the node-specific pattern features under the guidance of position embedding. Extensive experiments on six real-world datasets demonstrate the superiority of our model in terms of prediction performance and computational efficiency.
Spatial-Temporal Position-Aware Graph Convolution Networks for Traffic Flow Forecasting
IEEE Transactions on Intelligent Transportation Systems ; 24 , 8 ; 8650-8666
2023-08-01
3009163 byte
Article (Journal)
Electronic Resource
English
Spatio-Temporal Graph Attention Convolution Network for Traffic Flow Forecasting
Transportation Research Record | 2024
|Graph transformer based dynamic multiple graph convolution networks for traffic flow forecasting
DOAJ | 2023
|Graph transformer based dynamic multiple graph convolution networks for traffic flow forecasting
Wiley | 2023
|Spatial‐temporal correlation graph convolutional networks for traffic forecasting
Wiley | 2023
|