Journal Home Online First Current Issue Archive For Authors Journal Information 中文版

Frontiers of Information Technology & Electronic Engineering >> 2021, Volume 22, Issue 9 doi: 10.1631/FITEE.2000243

Forecasting traffic flows in irregular regions with multi-graph convolutional network and gated recurrent unit

杭州电子科技大学计算机学院,中国杭州市,310018

Received: 2020-05-21 Accepted: 2021-09-10 Available online: 2021-09-10

Next Previous

Abstract

The prediction of regional traffic flows is important for traffic control and management in an intelligent traffic system. With the help of deep neural networks, the convolutional neural network or residual neural network, which can be applied only to regular grids, is adopted to capture the spatial dependence for flow prediction. However, the obtained regions are always irregular considering the road network and administrative boundaries; thus, dividing the city into grids is inaccurate for prediction. In this paper, we propose a new model based on and (MGCN-GRU) to predict traffic flows for . Specifically, we first construct heterogeneous inter-region graphs for a city to reflect the relationships among regions. In each graph, nodes represent the and edges represent the relationship types between regions. Then, we propose a to fuse different inter-region graphs and additional attributes. The GRU is further used to capture the temporal dependence and to predict future traffic flows. Experimental results based on three real-world large-scale datasets (public bicycle system dataset, taxi dataset, and dockless bike-sharing dataset) show that our MGCN-GRU model outperforms a variety of existing methods.

Related Research