Multi-channel spatial-temporal traffic flow prediction based on hybrid static-dynamic graph convolution

Abstract

Aiming at the problem that the traffic flow prediction model did not consider the correlation of road context and the dynamics of spatial dependency, a multi-channel spatial-temporal traffic flow prediction based on hybrid static-dynamic graph convolution (MHGCN) was proposed.A sandwich structure (i.e.multi-channel spatial module in the middle and temporal module on both sides) was used in the model to extract spatial-temporal features, and the multi-channel spatial module was divided into static graph convolution module and dynamic graph convolution module.The static graph convolution module simultaneously extracted specific and common features from topological spatial structures, semantic spatial structures, and their combinations.The dynamic graph convolution module assigned different weights to different features and extracts dynamic spatial features from unknown graph structures.In the temporal module, the multi-head attention mechanism was used to extract the global temporal features, and the temporal gating mechanism extracted the local temporal features.The model extracted spatial information from different spatial structures and temporal information from different time intervals to establish a global and comprehensive spatial-temporal relationship.The experimental results show that the MHGCN performs better than the existing traffic flow prediction models on four real world traffic flow datasets.

Keywords

• The National Natural Science Foundation of China;The “Pioneer” and “Leading Goose” Research ＆ Development Program of Zhejiang Province;intelligent transportation;dynamic graph convolution;multi-head attention