• 基于动态时空图网络的交通流量预测模型

    Traffic flow prediction model based on dynamic spatio-temporal graph network

    • 传统的交通流量预测模型使用静态邻接矩阵进行时空建模,忽略了路网节点间潜在的空间依赖关系以及交通流量数据所呈现出的周期相似性。为此,提出一种基于多头注意力机制的动态时空图网络模型ADSTGN。首先,将交通流量数据处理成三通道周期性时间序列,并对序列数据经过时间位置编码建模时间特征确保序列顺序性。其次,在空间维度方面提出一种动态图生成器模块。该模块利用相似性空间特征与距离特征表示静态路网结构信息,并结合自适应邻接矩阵实时捕捉路网节点之间隐藏的空间依赖。最后,在时间维度方面提出了对多头注意力机制运用一维卷积整合局部上下文的办法,充分捕获三通道周期数据下的时间相关性,并过滤历史冗余信息。在4个高速公路公开数据集(PEMS03、PEMS04、PEMS07和PEMS08)上进行了实验。结果表明:相比现有的基线模型,ADSTGN模型具有更优的性能指标。证明了该模型在交通流量预测中的有效性。

       

      Abstract: Traditional traffic flow prediction models use static adjacency matrices for spatiotemporal modeling, ignoring the potential spatial dependencies between road network nodes and the periodic similarity presented by Traffic flow data. Therefore, a dynamic spatio-temporal graph network model ADSTGN based on multi-head attention mechanism is proposed. Firstly, the traffic flow data are processed into three channel periodic time series, and the temporal features of sequence data are modeled by temporal and positional encoding to ensure sequence order. Secondly, a dynamic graph generator module is proposed in terms of spatial dimension, which uses similarity spatial features and distance features to represent road network structure information, and combines adaptive adjacency matrix to capture hidden spatial dependencies between road network nodes. Finally, a method of using one-dimensional convolution to integrate local context for multi head attention mechanism is proposed in terms of time dimension, which fully captures the time correlation under three channel periodic data and filters historical redundant information. Experiments are conducted on four publicly available datasets on highways(PEMS03, PEMS04, PEMS07 and PEMS08), and the results show that the ADSTGN model has better performance indicators compared to existing baseline models, demonstrating the effectiveness of the model in traffic flow prediction.

       

    /

    返回文章
    返回