Enhanced road information representation in graph recurrent network for traffic speed prediction

Abstract

Correctly capturing the spatial-temporal correlation of traffic sequences will benefit to make accurate predictions of the future traffic states. In the paper, the methods of enhancing road spatial and temporal information representation are proposed. Firstly, the parameter matrix of each road is constructed to represent the road-specific traffic patterns for the graph convolution neural network and the recurrent neural network. Then, the node embedding, and matrix factorization are used to reduce the scale of the parameter matrix. Secondly, the node embedding-based Data Adaptive Graph Generation model was introduced to infer the indirect relationship of each node, and the gating mechanism is designed to control the weights of the direct spatial information and the indirect spatial information. Thirdly, to enhance the traffic sequence representation, the time tag and peak tag for the sequences are designed at each sampling moment. At last, the Enhanced Road Information Representation in Graph Recurrent Network (En-GRN) is proposed to predict traffic speed, and the prediction performance is tested on SZ-taxi and Los-loop dataset. The experimental results show that the presented works are effective for improving traffic prediction accuracy.