论文标题

用于运输需求预测的耦合图层图卷积

Coupled Layer-wise Graph Convolution for Transportation Demand Prediction

论文作者

Ye, Junchen, Sun, Leilei, Du, Bowen, Fu, Yanjie, Xiong, Hui

论文摘要

图形卷积网络(GCN)由于其出色的能力捕获了站点级别或区域运输需求之间的非欧盟空间依赖性,因此已广泛应用于运输需求预测。但是,在大多数现有研究中,图形卷积是在启发式生成的邻接矩阵上实施的,该矩阵既不能准确地反映电台的真实空间关系,也不能自适应地捕获需求的多级空间依赖性。为了应对上述问题,本文提供了一个新颖的图形卷积网络,用于运输需求预测。首先,提出了一种新型的图形卷积体系结构,该体系结构具有不同的层次矩阵,并且在训练过程中所有邻接矩阵都是自学的。其次,提供了层的耦合机构,该机构将高级邻接矩阵与下层相关联。它还减少了我们模型中参数的规模。最后,构建了一个统一网络,以通过将隐藏的空间状态与门控复发单元集成在一起,从而给出最终的预测结果,该单位可以同时捕获多级空间依赖性和时间动态。已经在两个现实世界数据集(NYC Citi Bike和NYC出租车)上进行了实验,结果证明了我们的模型优于最先进的模型。

Graph Convolutional Network (GCN) has been widely applied in transportation demand prediction due to its excellent ability to capture non-Euclidean spatial dependence among station-level or regional transportation demands. However, in most of the existing research, the graph convolution was implemented on a heuristically generated adjacency matrix, which could neither reflect the real spatial relationships of stations accurately, nor capture the multi-level spatial dependence of demands adaptively. To cope with the above problems, this paper provides a novel graph convolutional network for transportation demand prediction. Firstly, a novel graph convolution architecture is proposed, which has different adjacency matrices in different layers and all the adjacency matrices are self-learned during the training process. Secondly, a layer-wise coupling mechanism is provided, which associates the upper-level adjacency matrix with the lower-level one. It also reduces the scale of parameters in our model. Lastly, a unitary network is constructed to give the final prediction result by integrating the hidden spatial states with gated recurrent unit, which could capture the multi-level spatial dependence and temporal dynamics simultaneously. Experiments have been conducted on two real-world datasets, NYC Citi Bike and NYC Taxi, and the results demonstrate the superiority of our model over the state-of-the-art ones.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源