基于ASTG-CRNN模型的多步长交通流预测.pdf

针对交通流预测模型中路网表征结构难以进行刻画和交通流数据中动态时空相关性难以进行建模以及其中时间特征捕获不充分的问题,提出一种基于注意力机制和时空图卷积循环神经网络的交通流预测模型(ASTG-CRNN)。首先,通过定义节点相对邻近度来确定路网表征结构的关系权重;其次,通过在时空维度上引入注意力机制对动态时空相关性进行建模,再采用图卷积捕获交通流数据中的空间特征;最后,采用卷积神经网络和双向门控循环神经单元的组合模块共同捕捉时间特征,从而能更好地表达交通流的时空特性。在两个公开交通流数据集 PeMS04 和 PeMS08 上对模型预测效果进行验证,其结果表明,ASTG-CRNN 模型的预测结果均优于其它模型,与时空同步图卷积网络模型(STSGCN)相比,在未来 1h 内预测结果的 MAE、RMSE 和 MAPE 在数据集 PeMS08 上分别降低了 2.71、2.69 和 0.87%。关键词:交通流预测;注意力机制;相对邻近度;时空相关性;图卷积网络;循环神经网络中图分类号:TP183;U491.1+4 摇 摇 摇 文献标识码:A摇 摇摇 摇 摇 文章编号:1673-629X(2023)09-0141-08doi:10.3969/j.issn.1673-629X.2023.09.021Multi-step Traffic Flow Prediction Based on ASTG-CRNNGUI Xiang-quan1,XIONG Jia-chang1,LI Li1,GUO Sha-sha2(1.School of Computer and Communication,Lanzhou University of Technology,Lanzhou 730050,China;2.Digitization and Information Center,China National Petroleum Corporation Changqing Oilfield Branch,Xi爷an 710000,China)Abstract:Aiming at the problems that it is difficult to describe the road network representation structure in the traffic flow predictionmodel,to model the dynamic spatio-temporal correlation,and to capture sufficiently the temporal features in the traffic flow data,wepropose a traffic flow prediction model based on attention mechanism and spatio-temporal graph convolutional recurrent neural network(ASTG-CRNN).Firstly,the relationship weight of the road network representation structure is determined by defining the relativeproximity of nodes.Secondly,the dynamic spatio-temporal correlation is modeled by introducing an attention mechanism in the spatio-temporal dimension,and then the spatial features in the traffic flow data are captured by the graph convolutional network.Finally,thecombination module of the convolutional neural network and the bidirectional gated recurrent neural unit are used to capture temporalfeatures,so that the spatio-temporal characteristics of traffic flow can be better expressed.The prediction effect of the model is verifiedon two public traffic flow datasets PeMS04 and PeMS08.It is showed that the prediction results of the ASTG-CRNN model are betterthan that of other models.Compared with the spatio-temporal synchronous graph convolutional network model(STSGCN),the MAE,RMSE,and MAPE of the predicting results in the next 1h are reduced by 2.71,2.69 and 0.87%respectively on the dataset PeMS08.Key words:traffic flow prediction;attention mechanism;relative proximity;spatio-temporal correlations;graph convolutional network;recurrent neural network0摇 引摇 言随着城市化水平不断提升,大量人口涌向城市,城市中的出行车辆增多,导致城市居民所面临的交通拥堵问题日益突出。通过交通流预测,实现对未来交通状况的预测评估,进而完成车辆调度分流、拥堵风险评估等工作,这对于缓解城市交通压力具有重要作用。文中研究的交通流特指交通流量,是在特定时间段内通过道路上某一断面的车流数。交通流量是量化第 33 卷摇 第 9 期2023 年 9 月摇 摇 摇 摇 摇 摇 摇 摇 摇 摇计 算 机 技 术 与 发 展COMPUTER TECHNOLOGY AND DEVELOPMENT摇 摇 摇 摇 摇 摇 摇 摇 摇 摇Vol.33摇 No.9Sep.摇 2023交通通行能力的一种有效指标,准确的交通流预测信息可以为交通管理者提供有力的交通决策依据。目前交通流预测方法主要有两类:模型驱动方法和数据驱动方法。模型驱动方法根据数学理论假设预先确定模型,并用少量数据去拟合模型来实现交通流预测。这种方法存在模型结构单一、假设情况与现实情况较大差距等问题,导致预测模型在实际应用中预测效果不好。常用的预测模型有卡尔曼滤波模型1、自回归整合移动平均模型(ARIMA)2等。与模型驱动方法不同,数据驱动方法可以很好地克服交通流数据的不确定性。数据驱动方法分为传统机器学习方法和基于深度学习的方法。传统机器学习方法中被应用于交通流预测的模型有支持向量机回归(SVR)3、贝叶斯模型4、K 最近邻(KNN)5等。尽管传统机器学习方法与模型驱动方法相比可以从交通流数据中学习统计规律,但其处理高维的交通流数据能力有限。随着深度学习方法的兴起,具有高维数据处理和非线性数据特征挖掘能力的深度神经网络模型受到越来越多的青睐。Wang 等人6提出一种长短记忆网络(LSTM)编码器-解码器结构的交通流预测模型,该模型能够准确地学习局部特征和长期依赖关系。Chen等人7构建了一种卷积 LSTM 模型来估计短期交通流,该模型能同时挖掘交通流数据中的时空特征。袁华等人8在卷积神经网络(CNN)的基础上提出扩张-因果卷积神经网络,分别用扩张卷积和因果卷积来增加感受野的大小和解决信息泄露问题。桂智明等人9利用卷积神经网络和门控循环单元(GRU)来提取交通流的时空特征,同时利用交通流的周相似性提取周期特征。凤少伟等人10提出一种 K-means 与 GRU 神经网络结合的交通流预测方法,该方法能够很好地挖掘交通流时间序列的关联性。传统的 CNN、LSTM、GRU 等模型能够很好地处理不平稳非线性序列问题,但其大多只能提取到交通流数据的局部信息,且无法应用于标准的网格数据中,而利用图卷积可以将神经网络应用于图结构的数据中。Lv 等人11提出了一种时间多图卷积网络,用于联合建模道路网络中与各种全局特征的空间、时间、语义相关性。谷振宇等人12提出了一种基于时空图卷积循环神经网络的交通流预测模型,采用图卷积神经网络捕获交通流数据中的空间依赖关系,采用门控循环神经网络捕获交通流数据中的时空依赖关系。然而交通流数据在时空维度上不仅具有依赖性,也存在着较强的相关性,且相关性是动态变化的。为了对交通流数据中的动态相关性进行建模,Guo 等人13提出了一种新的基于注意力机制的图卷积门控循环神经网络模型(ASTGCN),该模型使用利用图卷积网络捕获用于建模空间依赖的拓扑特征,同时利用注意力机制根据速度数据的重要程度为特征分配权重,进一步提高了模型的预测精度。Song 等人14提 出 了 一 种 时 空 同 步 图 卷 积 网 络 模 型(STSGCN),该模型通过精心设计的时空同步建模机制,能够有效地解决复杂的局部时空相关性以及局部时空图中的异质性问题。车辆在道路网络系统上聚集成为交通流,交通流的产生、扩展以及状况演变过程与路网表征结构密切相关,因此路网表征结构的关系权重确定是影响道路交通流预测精度的主要因素。目前在路网表征结构的关系权重设计研究工作方面,一般采用单纯的逻辑关联(0 或 1)或以距离为运算准则的高斯核来表示,相连关系则表示为无向连接的,这种权重和连通关系的设计存在对路网的空间结构刻画能力不足的问题。同时,交通系统具有时空强耦合的特性,不同地点的交通状况相互影响,不同时间切片内的交通状况之间存在时间相关性,现有的交通流预测方法大多缺乏对交通流数据中的动态时空相关性进行建模的能力。并且现有研究在交通流数据中时间依赖性的挖掘上往往采用的是单一结构的模块,这存在对交通流数据中的时间特性挖掘不充分的问题。针对上述问题,该文的研究工作情况如下:(1)构建一种权重邻接矩阵,用节点的有向带权连通关系来刻画路网结构的有向性,通过定义节点相对邻近度来确定路网表征结构的关系权重,解决了对路网空间结构刻画不足的问题。(2)在时空维度上引入注意力机制,自适应地赋予数据不同的重要性和捕获节点之间的动态相关性,解决了交通流数据中动态时空相关性难以进行建模的问题。(3)采用卷积神经网络和双向门控循环神经单元的组合模块对已经提取过空间特征的交通流数据进行时间特征的捕捉,解决了对交通流数据中时间特性挖掘不充分的问题。1摇 ASTG-CRNN 模型1.1摇 交通路网时空图构建文中 t 时刻有向空间图 Gt表述如式(1)所示。Gt=(V,E,A)V=vi|i=1,2,NE=(vi,vj)|vi,vj沂 VA=Aij|(vi,vj)沂 E(1)其中,V 是图 Gt中检测节点(即为路网结构中的传感器)vi的集合,E 表示图 Gt中有向边的集合,(vi,vj)表示从节点 vi到节点 vj的交通流向,且(vi,vj)屹(vj,241摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 计算机技术与发展摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 第 33 卷vi);A 为权重连接矩阵,Aij表示节点 vi与节点 vj间的相对邻近度,如式(2)式(4)所示。wij=exp(-dist(vi,vj)2啄2),摇 j 沂 k(i)0,otherwise(2)啄2=1nk移ni=1移j沂k(i)dist(vi,vj)2(3)Aij=1k+1移v沂k(i),iwvj,摇 i 屹 j(4)其中,k(i)表示当 dist(vi,vj)臆酌 时,节点 vi的 k 个邻域节点的集合,dist(vi,vj)为节点 vi到 vj的欧氏距离,酌 为设置的阈值,当两个节点的距离超出阈值 酌 的大小,则代表两节点的权重系数为 0,无依赖性。通过式(2)和式(3)可知,两节点的相对邻近度与如下因素有关:(1)两节点的欧氏距离;(2)整个路网图中的检测节点集合;(3)两节点的局部邻域节点。1.2摇 模型设计针对交通流的特点,提出一种基于注意力机制和时空图卷积循环神经网络的交通流预测模型(ASTG-CRNN),将交通路网中的部分路段作为检测路段来进行交通流预测。图 1 给出了 ASTG-CRNN 模型的总体框架。根据人们的日常生活规律,交通流数据会存在一定的周期性和规律性变化。因此,模型采用三个具有相同结构的独立组件,分别对历史交通流数据中的最近周期依赖关系、日周期依赖关系和周周期依赖关系进行建模,以便获取到交通流数据中更深度的周期性信息。rdYYY(0)wX(0)dX(1)rhX?(1)rdX?(0)hX?()rhH()(res)rhX()rhXhYwY?()rdH()(res)rdX()rdX(1)rwX?()rwH()(res)rwX()rwX图 1摇 ASTG-CRNN 模型的总体框架摇 摇 假设采样频率每天为 q 次,当前为 t0时刻且预测序列段的长度为 Tp。如图 2 所示,沿时间轴截取三个长度为 Th、Td和 Tw的时间序列片段,分别作为最近周期、日周期、周周期组件的输入,其中 Th、Td和 Tw均为Tp的整数倍,三个时间序列片段的具体情况如式(5)式(7)所示:(1)最近周期时间序列片段,如图 2 中淤所示,此序列段与要预测的时间序列段在时间维度上直接相邻。由于在现实生活中,交通拥堵的形成和分散是渐进的,因此与之直接相邻的历史时间序列会对其有较大影响。wTdThTpT图 2摇 三个时间序列片段的构建示例摇 摇Xh=(Xt0-Th+1,Xt0-Th+2,Xt0)(5)(2)日周期时间序列片段,如图 2 中于所示,此序列段由相邻几天中与要预测的序列段时间相同的序列段组成。由于人们在工作日的活动情况几乎是没有太大变化的,比如公司和家庭两点一线,存在固定的生活模式,因此交通流数据可能会呈现出重复的日周期模式。日周期组件的目的就是用来在时间维度上建模交通流数据中的日周期依赖性。341摇 第 9 期摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 贵向泉等:基于 ASTG-CRNN 模型的多步长交通流预测Xd=(Xt0-(TdTp)*q+1,Xt0-(TdTp)*q+Tp,Xt0-(TdTp-1)*q+1,Xt0-(TdTp-1)*q+Tp,Xt0-q+1,Xt0-q+Tp)(6)(3)周周期时间序列片段,如图 2 中盂所示,此序列段是由相邻几周中与要预测的序列段时间相同的序列段组成。通常来说,当前周的工作日的交通流状况与历史周的工作日流量状况存在相似性,但是与历史周的周末的交通流状况存在不同。周周期组件是用来在时间维度上捕获交通流数据中的周周期性时间变化趋势。Xw=(Xt0-7*(TwTp)*q+1,Xt0-7*(TwTp)*q+Tp,Xt0-7*(TwTp-1)*q+1,Xt0-7*(TwTp-1)*q+Tp,Xt0-7*q+1,Xt0-7*q+Tp)(7)这三个组件具有相同的网络结构,每个组件由若干个时空模块组成。每个时空模块包含时空注意力模块、图卷积模块、卷积神经网络与双向门控循环神经单元的组合模块以及全连接层模块。最后,根据权重矩阵对三个分量进行融合,得到最终的预测结果。1.3摇 时空注意力模块(1)时间注意力模块。在时间维度上,不同时间切片上的交通状况之间存在相互关系,且相互关系在不同情况下也存在差异。使用一种注意力机制来自适应地赋予数据不同的重要性。以最近周期组件中的在第 r 层上时间注意力模块为例,如式(8)式(9)所示:E=Ve滓(X(r-1)hU1)U2(U3X(r-1)h)T+be)(8)E=exp(Eij-max(Ej)移Ni=1exp(Eij)(9)其中,Ve、be、U1、U2、U3都为权重矩阵,Eij的值在语义上表示 i 时刻和 j 时刻之间的依赖强度,Ej表示第 j列的所有元素,E为 E 进行归一化处理后的矩阵。然后将 E直接应用于 X(r-1)h,如式(10)所示,并将其作为空间注意力模块的输入。X(r-1)h=(X1,X2,XTr-1)=(X1,X2,XTr-1)E(10)(2)空间注意力模块。在空间维度上,不同区域内的交通状况存在一种相互的影响,且这种影响实际上还存在相当高度的动态性。用注意力机制来自适应地捕获在空间维度上节点之间的动态相关性。同样地,以最近周期组件第 r 层上的空间注意力模块为例,如式(11)式(12)所示:S=Vs滓(X(r-1)hW1)W2(W3X(r-1)h)T+bs)(11)Sij=exp(Sij-max(Sj)移Ni=1exp(Sij)(12)其中,Vs、bs、W1、W2、W3都为权重矩阵,滓 被用作激活函数。Sij的值在语义上表示的是节点 vi和节点 vj之间的相关强度,Sj表示第 j 列中的所有元素。同样地,还需要对 S 进行归一化操作得到矩阵 S。1.4摇 图卷积模块在谱图分析中,一个图用它对应的拉普拉斯矩阵表示。图结构的性质可以从对应的拉普拉斯矩阵中分析获得。图的拉普拉斯矩阵可以定义为 L=D-A,A 为带权重的邻接矩阵,D 为包含节点度的对角矩阵,拉普拉斯矩阵的特征分解如式(13)式(14)所示:L=U撰UT(13)撰=diag(姿0,姿1,姿N-1)(14)然而,当图的尺度变大时,对拉普拉斯矩阵进行特征值分解所付出的代价也随之变高。该文采用切比雪夫多项式来解决该问题,如式(15)式(16)所示:g兹*Gx=g兹(L)x=移K-1k=0兹k(Tk(L)已S)x(15)L=2姿maxL-IN(16)其中,参数 兹 为多项式系数的向量,姿max为拉普拉斯矩阵的最大特征值,已为哈达玛积。切比雪夫多项式的递归定义如式(17)所示:Tk(x)=2xTk-1(x)-Tk-2(x)(17)且 T0(x)=1,T1(x)=x,通过卷积核 g兹提取图中每个节点周围 0 到 K-1 阶邻域的信息。1.5摇 CNN-BiGRU 组合模块该文使用卷积神经网络(CNN)与双向门控循环神经单元(BiGRU)的组合模块对已经提取空间依赖关系的交通流数据再进行时间依赖关系的提取。通过合并相邻时间片上的信息来进一步堆叠时间维度上的卷积层,同样地,以最近周期时间组件在第 r 层的卷积操作为例,如式(18)所示:H(r)h=ReLU(椎*ReLU(g兹(L)X(r-1)h)(18)使用 ReLU 作为组合模块中的激活函数,相较于SELU(缩放指数线性单元)具有更好的收敛性并且可以有效地避免梯度消失的问题。在进行完卷积操作后,还需要将结果输入到 BiGRU 中,用来增强对交通流数据中的时间依赖关系的提取。如图 3 所示,为了防止过拟合和提高运行效率,在卷积层之后添加了池化层。CNNBiGRU(1)rhX?()rhH()f rhH()b rhH()rhH图 3摇 CNN-BiGRU 组合模型结构441摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 计算机技术与发展摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 第 33 卷BiGRU 是由向前和向后两个单向的 GRU 层堆叠而成,其中前向 GRU 是对输入从开始到结束的前向交通流序列进行预测,后向 GRU 是对输入从结束到开始的反向交通流序列进行预测。由于综合考虑了前后时刻交通流对当前时刻交通流的影响,模型的精度得到了提升。以最近周期时间组件在第 r 层上 BiGRU 的输出为例,如式(19)所示:H(r)h=WyHf(r)h,Hb(r)h+by(19)其中,Hf(r)h,Hb(r)h为前向和反向 GRU 的输出向量,Wy、by为权值和偏移量。然而深度学习网络随着网络层数的加深会产生退化问题,为解决这个问题,还需要将输入数据进行残差卷积操作,如图 1 所示,为ReLU 操作。最近周期时间组件第 r 层的输出如式(20)所示:X(r)h=ReLU(H(r)h+X(r)h(res)(20)最后还要将处理后的数据输入到全连接层模块,以确保当前时间组件下每个时空模块的输出具有相同的尺寸和形状。将多个时空模块进行叠加,可以进一步提取到更大范围的动态时空相关性。1.6摇 多组件融合当融合不同组件的输出时,三个组件的影响权重是不同的,并且影响权重可以从历史的交通流数据中学习到。融合后的最终预测结果如式(21)所示:Y=Wh已Yh+Wd已Yd+Ww已Yw(21)其中,Wh、Wd、Ww为权重矩阵,分别反映了三个时间维度组件对预测目标的影响程度。2摇 实验设计2.1摇 数据集描述为了验证模型的有效性,选取美国公路的两个公开交通流数据集 PeMS04 和 PeMS08 进行实验验证。这些 数 据 集 是 由 Caltrans Performance MeasurementSystem(PeMS)每 30 秒实时收集一次,并且交通流数据每隔 5 分钟汇总一次。PeMS04 包含 16 992 条交通流数据以及采集这些信息的 307 个检测节点的地理位置信息;PeMS08 包含 17 856 条交通流数据以及采集这些信息的 170 个检测节点的地理位置信息。2.2摇 对比基线模型为了对比分析预测模型的预测效果,选择在交通流预测中比较有代表性的预测模型,包括 LSTM15、GRU16、ConvLSTM17、ASTGCN13以及 STSGCN14作为对比基线模型。这些基线模型是参考对应文献进行构建。2.3摇 评价指标设计选取平均绝对误差(MAE)、均方根误差(RMSE)和平均绝对百分比误差(MAPE)作为实验验证评价指标。指标算法如式(22)式(24)所示:(1)平均绝对误差。MAE(yi,yi)=1n移ni=1yi-yi(22)(2)均方根误差。RMAE(yi,yi)=1n移ni=1(yi-yi)2(23)(3)平均绝对百分比误差。MAPE(yi,yi)=移ni=1(yi-yiyi)100%n(24)其中,yi为观察值,yi为预测值,n 为训练数据的个数。2.4摇 模型参数以 6 颐 2 颐 2 的比例将数据集划分为训练集、验证集和测试集。考虑到计算效率和预测性能的提升,设置切比雪夫多项式的项数为 3。在训练阶段将 MAE作为损失函数并进行重复实验对超参数进行调整,最后设置学习率为 0.000 1,批量大小为 64,训练迭代次数在 PeMS04 和 PeMS08 数据集上分别为 200 和 80。摇 图 4摇 在 PeMS04 数据集上 ASTG-CRNN 模型置不同的 Th、Td以及 Tw值时在未来 1h上预测结果的 MAE 对比对于三个时间组件的长度,设置不同的 Th、Td以及 Tw值来比较模型的预测效果,如图 4 所示。从图 4中可以看出,当 Th=36、Td=24 和 Tw=24 时,模型的预测效果较佳。同时设置预测窗口 Tp=12,即目标是541摇 第 9 期摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 贵向泉等:基于 ASTG-CRNN 模型的多步长交通流预测预测在未来一小时内的交通流。2.5摇 实验结果与分析(1)各模型多步长预测结果。用 ASTG-CRNN 模型和前述的 5 种基线模型分别对未来 1h 内的交通流进行预测,每 5 分钟为一步长,预测步长 12 步。在数据集 PeMS04 上 ASTG-CRNN 模型在前 3 步的预测效果优势不明显,如图 5(a)(b)所示,MAE、RMSE 要略差于 STSGCN 模型,而在后 9 步 ASTG-CRNN 模型的预测效果要优于其它基线模型,尤其是在 MAPE 指标上优势明显,如图 5(c)所示。而在数据集 PeMS08 上 ASTG-CRNN 模型在前 12 步的预测效果均要优于其它基线模型,如图 6(a)(b)以及(c)所示,尤其是在后 6 步上当 STSGCN、ASTGCN 的三项指标大幅降低的时候,ASTG-CRNN模型的三项指标降幅并不明显且指标数值低于其它基线模型,显示出该文提出的模型在多步长交通流预测方面有着一定的优越性。但在两项数据集上 ASTG-CRNN 等模型在三项评价指标上的抖动幅度要明显大于 LSTM、GRU 以及 CNN-LSTM 模型,这是因为LSTM、GRU 以及 CNN-LSTM 模型只对交通流序列的时间或空间特征进行提取,而 ASTG-CRNN 等模型要同时考虑到时空相关性,对交通流序列的时间和空间特征进行捕捉,因而在稳定性方面 ASTG-CRNN 模型的效果要差些。ASTG-CRNN 模型相较于其它基线模型的具体预测误差数值对比如表 1 所示。图 5摇 各模型在 PeMS04 数据集上不同时间步长的预测结果对比图 6摇 各模型在 PeMS08 数据集上不同时间步长的预测结果对比表 1摇 在未来 15 min、30 min 和 1 h 上 ASTG-CRNN 模型和基线模型的预测误差数值数据集模型摇 摇 摇 摇 摇 15 min摇 摇 摇 摇 摇摇 摇 摇 摇 摇 30 min摇 摇 摇 摇 摇摇 摇 摇 摇 摇 摇 1 h摇 摇 摇 摇 摇MAERMSEMAPE/%MAERMSEMAPE/%MAERMSEMAPE/%PeMS04LSTM23.5440.1816.6423.6240.3116.6923.8040.4816.83GRU23.1038.5215.9323.1738.6515.9823.4138.9016.22ConvLSTM22.8736.2415.1222.9436.3715.5523.1236.5815.96ASTGCN21.0233.3914.0521.8334.6814.5422.8436.7415.30STSGCN19.8031.5813.4121.3033.8314.2724.4738.4616.27ASTG-CRNN19.9232.2713.1020.4033.2213.3621.3534.8914.24PeMS08LSTM19.4930.1312.8219.5030.6712.8319.6731.2812.94GRU17.9626.8212.4718.3927.6412.4919.0428.8812.67ConvLSTM16.8225.8811.6216.8526.0011.7717.0426.8612.18ASTGCN16.0724.3111.4316.5325.2011.6717.3326.7212.21STSGCN16.5925.3710.9817.5427.2711.5620.1130.6413.04ASTG-CRNN14.8323.7310.6915.1124.4910.8015.8326.2511.39641摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 计算机技术与发展摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 第 33 卷摇 摇(2)各模型交通流预测值与真实值对比。为了能直观地说明 ASTG-CRNN 模型的预测效果,分别在 PeMS04 和 PeMS08 数据集上随机选取一个路段,分别命名为路段 1 和路段 2,并对这两个路段提前一小时的交通流预测情况进行可视化,对比各模型的预测值与真实值之间的误差,时间间隔为 5 分钟,如图 7、图 8 所示。从图中可以看出,ASTG-CRNN 模型对真实值的拟合效果最好,而其它模型存在明显的滞后性,尤其在时间段 9:00-10:00 上对比效果明显。这说明 ASTG-CRNN 模型能够有效预测出未来的交通流情况和趋势。/(veh.(5?min)?)-1图 7摇 各模型在 PeMS04 数据集上路段 1 的预测值与真实值对比/(veh.(5?min)?)-1图 8摇 各模型在 PeMS08 数据集上路段 2 的预测值与真实值对比(3)消融实验。为了凸显该文使用的注意力机制、CNN-BiGRU组合模型以及构建的有向带权邻接矩阵的优势,设置了对比实验分别验证其对模型的影响。STG-CRNN为 ASTG-CRNN 模型去除注意力机制后的模型,ASTG-RNN、ASTG-CNN 分别为 ASTG-CRNN 模型去除 CNN、BiGRU 后的模型,NMASTG-CRNN 模型为在 ASTG-CRNN 模型的基础上将邻接矩阵相连节点的权重设置为 0 或 1 后的模型。各模型在两种数据集上不同时间步长的 RMSE 预测结果如表2 所示。由表 1、表 2 可知,注意力机制对 ASTG-CRNN 模型的预测效果都有一定作用。ASTG-CRNN 的效果要好于STG-CRNN,这是因为时空注意力机制可以对交通流数据中的时空相关性进行建模;ASTG-RNN、ASTG-CNN 的预测效果比 ASTG-CRNN 差,这是因为 CNN-BiGRU 组合模型可以对交通流数据中的时间特征进行充分捕捉,从而捕获到隐藏的信息;而 ASTG-CRNN相较于 NMASTG-CRNN 有着更低的预测误差,则说明通过构建权重邻接矩阵挖掘路网表征结构关系,比以简单的逻辑关系结构设计权重能够更好地表征路网结构。表 2摇 对比模型在不同时间步长上的 RSME 预测数值数据集模型时间区间/RMSE15 min30 min1 hPeMS04STG-CRNN32.8233.8635.48ASTG-RNN32.9533.8735.72ASTG-CNN33.0334.0035.93NMASTG-CRNN33.2434.1735.17ASTG-CRNN32.2733.2234.89PeMS08STG-CRNN23.9224.8126.63ASTG-RNN25.1527.0327.54ASTG-CNN25.5326.4127.98NMASTG-CRNN24.6925.5926.99ASTG-CRNN23.7324.4926.253摇 结束语提出的 ASTG-CRNN 交通流预测模型能够对路网表征结构进行准确刻画,解决交通流数据中动态时空相关性难以进行建模以及时间特征捕获不充分的问题。该模型首先定义检测节点相对邻近度对路网表征结构进行刻画,再引入时空注意力机制对交通流数据中动态时空相关性进行建模,随后再利用图卷积对交通流数据中的空间特征进行捕捉,最后利用 CNN-BiGRU 组合模块对交通流数据中时间特征进行充分捕捉。实验结果表明,ASTG-CRNN 模型可以有效降低多步长交通流预测误差。参考文献:1摇 白伟华,张传斌,张塽旖,等.基于异常值识别卡尔曼滤波器的短期交通流预测J.计算机应用研究,2021,38(3):817-821.2摇赵摇鹏,李摇璐.基于 ARIMA 模型的城市轨道交通进站量预测研究J.重庆交通大学学报:自然科学版,2020,39(1):40-44.3摇 童摇 林,官摇 铮.改进鲸鱼优化支持向量机的交通流量模糊粒化预测J.计算机应用,2021,41(10):2919-2927.4摇GU Y L,LU W Q,XU X Y,et al.An improved Bayesiancombination model for short-term traffic prediction withdeep learningJ.IEEE Transactions on Intelligent Transpor鄄741摇 第 9 期摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 贵向泉等:基于 ASTG-CRNN 模型的多步长交通流预测tation Systems,2020,21(3):1332-1342.5摇 MALLEK A,KLOSA D,BUSKENS C.Enhanced k-nearestneighbor model for multi-steps traffic flow forecast in urbanroadsC/Proceedings of the 2022 IEEE international smartcities conference(ISC2).Pafos:IEEE,2022:1-5.6摇 WANG Z,SU X,DING Z.Long-term traffic predictionbased on LSTM encoder-decoder architecture J.IEEETransactions on Intelligent Transportation Systems,2021,22(10):6561-6571.7摇 CHEN X,XIE X,TENG D.Short-term traffic flow predic鄄tion based on ConvLSTM modelC/Proceedings of the2020 IEEE 5th information technology and mechatronics en鄄gineering conference(ITOEC).Chongqing:IEEE,2020:846-850.8摇 袁摇 华,陈泽濠.基于时间卷积神经网络的短时交通流预测算法J.华南理工大学学报:自然科学版,2020,48(11):107-113.9摇 桂智明,李壮壮,郭黎敏.基于 ACGRU 模型的短时交通流预测J.计算机工程与应用,2020,56(21):260-265.10 凤少伟,凤摇 超,申摇 浩.基于 K-means 与 GRU 的短时交通流预测研究J.计算机技术与发展,2020,30(7):125-129.11 LV M,HONG Z,CHEN L,et al.Temporal multi-graph con鄄volutional network for traffic flow prediction J.IEEETransactions on Intelligent Transportation Systems,2020,22(6):3337-3348.12 谷振宇,陈摇 聪,郑家佳,等.基于时空图卷积循环神经网络的交通流预测J.控制与决策,2022,37(3):645-653.13 GUO S N,LIN Y F,FENG N,et al.Attention based spatial-temporal graph convolutional networks for traffic flow fore鄄castingC/Proceedings of the AAAI conference on artifi鄄cial intelligence.Hawai:AAAI Association,2019:922-929.14 SONG C,LIN Y F,GUO S N,et al.Spatial-temporal syn鄄chronous graph convolutional networks:a new framework forspatial-temporal network data