arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2009.08087cs.AI
收藏

基于FastGCRNN的城市交通流预测

Urban Traffic Flow Forecast Based on FastGCRNN

School of Computer Science and Engineering, Central South University(中南大学计算机科学与工程学院);School of Geosciences and Info-Physics, Central South University(中南大学地球科学与信息物理学院)

Ya Zhang, Mingming Lu, Haifeng Li

摘要:

交通预测是智能交通系统在城市交通网络中应用的重要前提。已有的研究工作采用RNN和CNN/GCN来刻画交通流的时空相关性,其中GCRN是最新的研究成果。然而,由于高计算复杂性,GCRN很难应用于大规模道路网络。为了解决这个问题,我们建议将道路网络抽象成一个几何图,并建立一个快速图卷积递归神经网络(FastGCRNN)来模拟交通流的时空依赖性。具体而言,利用FastGCN单元有效捕捉图中道路与周围道路的拓扑关系,通过重要性采样降低计算复杂性,结合GRU单元捕捉交通流的时间依赖性,并基于Encoder-Decoder框架将时空特征嵌入到Seq 2Seq中.在大规模交通数据集上的实验表明,该方法在保持较高准确率的同时,大大降低了计算复杂性和内存消耗。

英文摘要:

Traffic forecasting is an important prerequisite for the application of intelligent transportation systems in urban traffic networks. The existing works adopted RNN and CNN/GCN, among which GCRN is the state of art work, to characterize the temporal and spatial correlation of traffic flows. However, it is hard to apply GCRN to the large scale road networks due to high computational complexity. To address this problem, we propose to abstract the road network into a geometric graph and build a Fast Graph Convolution Recurrent Neural Network (FastGCRNN) to model the spatial-temporal dependencies of traffic flow. Specifically, We use FastGCN unit to efficiently capture the topological relationship between the roads and the surrounding roads in the graph with reducing the computational complexity through importance sampling, combine GRU unit to capture the temporal dependency of traffic flow, and embed the spatiotemporal features into Seq2Seq based on the Encoder-Decoder framework. Experiments on large-scale traffic data sets illustrate that the proposed method can greatly reduce computational complexity and memory consumption while maintaining relatively high accuracy.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑