A Hybrid Model Integrating Local and Global Spatial Correlation for Traffic Prediction

Abstract

Accurate traffic prediction can effectively alleviate traffic congestion problems. The complex spatial correlation of traffic flow contributes to the challenging prediction problem. Most current prediction methods focus on learning local spatial correlation, ignoring the spatial correlation of long-distance traffic flow. In this paper, we combine the improved Graph Convolutional Network (GCN) with Gated Recurrent Unit (GRU) to propose a hybrid model integrating local and global spatial correlation (T-LGGCN) for traffic prediction. The model consists of two parts: global spatial-temporal component and local spatial-temporal component. For the global spatial-temporal component, we construct the global correlation matrix to improve the GCN for obtaining the global spatial correlation. And GRU is stacked to obtain the global spatial-temporal correlation. For the local spatial-temporal component, we utilize the strategy of combining Fully Connected Layer (FCL) and GCN to analyze the local spatial correlation. Similarly, GRU is used to perform the output of local spatial-temporal correlation. The output of the two components is finally summed, and the prediction results are generated with the dense network. Experiments were conducted using the highway datasets PEMS04 and PEMS08 from the Caltrans Performance Measurement System, and the results show that our model significantly outperforms state-of-the-art baselines.

Keywords

• Traffic prediction
• graph convolutional network
• global spatial correlation
• local spatial correlation