基于时空卷积注意力网络的道路速度预测研究--以宁波主要路网为例

Research on Road Speed Prediction Based on Spatiotemporal Convolution Attention Network-A Case Study of Ningbo's Main Road Network

针对交通流数据建模时空特性挖掘不足的问题,提出了STGAN网络。运用时空图卷积和注意力机制挖掘道路网络时空规律。注意力机制使得网络对相邻道路和历史时间数据的关注度不同,其分组注意力卷积的机制能够使得网络训练摆脱路网空间拓扑规模的限制,并使模型可运用在较大规模的路网上。实验表明,STGAN模型在宁波高、快速路和主干路上速度预测误差比DCRNN小,在宁波数据集上预测速度表现出良好的精度。

STGAN network is proposed to solve the problem of insufficient mining of spatiotemporal characteristics of traffic flow data modeling.The spatiotemporal graph convolution and attention mechanism is used to mine the spatiotemporal laws of road network.The attention mechanism makes the network pay different attention to adjacent roads and historical time data.Its grouping attention convolution mechanism can make the network training get rid of the restriction of the spatial topological scale of the road network,and make the model can be applied to a large scale road network.The experiment shows that the STGAN model has a smaller speed prediction error than DCRNN in Ningbo high-speed,expressway and trunk roads,and the prediction speed shows good accuracy on Ningbo data set.