Optimized Graph Convolution Recurrent Neural Network for Traffic Prediction

Kan Guo,Yongli Hu,Zhen Qian,Hao Liu,Ke Zhang,Yanfeng Sun,Junbin Gao,Baocai Yin
DOI: https://doi.org/10.1109/tits.2019.2963722
IF: 8.5
2021-02-01
IEEE Transactions on Intelligent Transportation Systems
Abstract:Traffic prediction is a core problem in the intelligent transportation system and has broad applications in the transportation management and planning, and the main challenge of this field is how to efficiently explore the spatial and temporal information of traffic data. Recently, various deep learning methods, such as convolution neural network (CNN), have shown promising performance in traffic prediction. However, it samples traffic data in regular grids as the input of CNN, thus it destroys the spatial structure of the road network. In this paper, we introduce a graph network and propose an optimized graph convolution recurrent neural network for traffic prediction, in which the spatial information of the road network is represented as a graph. Additionally, distinguishing with most current methods using a simple and empirical spatial graph, the proposed method learns an optimized graph through a data-driven way in the training phase, which reveals the latent relationship among the road segments from the traffic data. Lastly, the proposed method is evaluated on three real-world case studies, and the experimental results show that the proposed method outperforms state-of-the-art traffic prediction methods.
engineering, electrical & electronic,transportation science & technology, civil
What problem does this paper attempt to address?
This paper mainly explores the problem of traffic flow prediction, which is an important issue in intelligent transportation systems and has wide applications in traffic management and planning. The current challenge is how to effectively mine the spatial and temporal information of traffic data. The paper proposes an optimized graph convolution recurrent neural network (OGCRNN) to solve this problem. Traditional deep learning methods, such as convolutional neural networks (CNN), have shown good performance in traffic prediction. However, they sample traffic data into regular grids, which disrupts the natural spatial structure of road networks. The paper introduces the concept of graph networks, constructs a graph to represent the spatial information of road networks, and the proposed method learns to optimize the graph in a data-driven manner during the training phase, revealing the potential relationships between road segments in the time series data of road segments. Experimental results show that the proposed OGCRNN method outperforms existing traffic prediction methods in three real case studies, demonstrating its superiority in predicting future 15-minute and 30-minute traffic flows. The main contributions of this method include constructing a graph-based spatio-temporal neural network, proposing a graph updating strategy, and constructing an optimized graph matrix. The paper also reviews related work, including statistical, machine learning, and deep learning-based traffic prediction methods, and provides a detailed introduction to the components of the proposed method, such as data representation, graph convolution, GRU network, and the working principle of OGCRNN. Finally, the paper provides experimental settings and comparison methods to validate the effectiveness of OGCRNN.