基于GCN-BiLSTM的短时交通流预测模型 (1).pdf

第 卷第期 年 月武汉理工大学学报(交通科学与工程版)J o u r n a l o fWu h a nU n i v e r s i t yo fT e c h n o l o g y(T r a n s p o r t a t i o nS c i e n c e&E n g i n e e r i n g)V o l N o O c t 基于G C N B i L S TM的短时交通流预测模型张阳)胡月)陈德旺,)陈云飞)(福建工程学院交通运输学院)福州 )(福州大学计算机与大数据学院)福州 )(福建工程学院电子电气与物理学院)福州 )摘要:文中提出图卷积网络(G C N)与双向长短时记忆神经网络(B i L S TM)组合短时交通流预测模型利用图卷积网络提取路网拓扑结构解决拓扑关系问题,提取路网间的空间特征,利用双向长短时记忆神经网络用于学习交通数据的动态变化以获取时间相关性,融合G C N B i L S TM模型同时考虑路网时空关系实现交通流预测结果表明:文中提出的方法能更好地适应在不同交通流特性条件下的交通流,工作日和周末的预测偏差相较于经典算法降低 和 关键词:深度学习;交通流预测;B i L S TM;G C N;城市路网中图法分类号:U d o i:/j i s s n 收稿日期:第一作者:张阳(),男,博士,副教授,主要研究领域为智能交通信息处理、交通大数据处理、交通流预测基金项目:国家自然科学基金()、福建省自然科学基金(J )引言道路交通流的变化是一个实时性、非线性及非平稳性的随机过程采样间隔时间越短,交通流变化的随机性和不确定性越强,越难以准确分析其变化规律目前,城市道路的短时交通流预测方法包括传统方法和机器学习方法传统方法主要利用时间序列 预 测 短 时 交 通 流,如 卡 尔 曼 滤 波 模 型(K F)、自回归移动平均(A R I MA)模型、多元线性回归模型等 K u m a r等将短期交通流预测与季节性A R I MA模型相结合,并通过有限的数据来使用A R I MA模型预测交通流 K u m a r等提出了一种只需有限的数据即可进行交通流量预测的方法,即基于卡尔曼滤波模型来解决依赖大量数据进行开发的不足常用的机器学习模型通过自适应学习不断调整自身参数来捕捉复杂的非线性关系,如支持向量回归(S V R)、K近邻(KNN)等虽然这些模型可以解决传统方法性能差的问题,但追求高精度容易导致模型训练时间成本大、数据样本要求过高李林超等提出一种基于核函数切换的预测方法,根据不同时刻的核函数的S V R模型在处理交通流时的差异性,在不同时刻选择最合适的核函数进行短时交通流量预测 C a i等在考虑道路网络时空关联性的前提下,提出使用高斯加权方法,来增强KNN模型的预测精度深度学习是近年来机器学习领域中的主要研究方向,包括循环神经网络(r e c u r r e n tn e u r a l n e t w o r k s,R NN)、长短时记忆网络(l o n gs h o r t t e r mm e m o r y,L S TM)等深度学习模型这些模型虽然能有效地学习交通流序列的长期相关性以及捕捉复杂的时间特征,但对于交通路网的空间性这一特性鲜有考虑,因此难以解决路网间的交通流预测问题 B r u n a等提出使用图卷积网络(g r a p hc o n v o l u t i o n a ln e t w o r k,G C N)来解决道路网络空间相关性问题,通过利用图形结构信息提取道路网络拓扑结构,再处理图结构中的不规则数据来实现交通 预测 X i e等提出 了一种新的 基于R NN G C N和B R B的交通流预测方法 Wu等用G C N提取交通数据中的拓扑结构特征,使用L S TM结构提取时间特征,结合R e s N e t优化整体模型,减少网络退化中梯度消失或爆炸的发生,最终实现交通流预测张阳等 利用长短时记忆神经元提取时间关联性,并利用卷积传输提取空间关联性,从而对交通流进行预测由于现有的交通预测模型很少从路网的角度考虑,只是单纯的从单一道路的角度考虑,没有将路网拓扑结构融入预测模型中,预测精度有待提高文中在以往预测模型的基础上,通过提取道路路网拓扑结构,考虑路网路段交通状态和交通流时空相关性,以此来实现短时交通流预测,并以实例验证了模型的预测精度基于G C N B i L S TM的短时交通流预测模型 图卷积神经网络交通流预测的准确度得益于邻近交通信息和时间信息的使用 用图卷积网络可以通过捕捉交通数据空间相关性来预测路网交通流,进而提高交通流预测的精度假设有一批N个节点的图数据,构成ND矩阵X,然后节点间形成NN矩阵A,车流特征矩阵X和邻接矩阵A为图卷积模型的输入,中间层和中间层之间的传播模式为H(l)(DADH(l)W(l)()式中:A为邻接矩阵和单位矩阵之和,即AAI;D为A的度矩阵;H(l),H(l)分别为l层和(l)层的特征矩阵初始层H()的特征矩阵为客流特征矩阵X,图卷积的结构见图图图卷积神经网络结构图W(l)为l层的权重矩阵;为激活函数式()式()分别示出了Ai j和D的计算Ai j,ij,ij()Di jNjAi j,ij,ij()式中:Ai j为节点i和节点j之间的连接关系 长短时记忆神经网络在L S TM体系结构中,遗忘门、输入门和输出门这三种特殊的结构赋予了其处理短期和长期时间序列内相关性的能力 遗忘门丢弃原单元状态中的信息,输入门存储来自外部状态的信息用来更新单元状态,输出门得到所有结果,计算并生成L S TM结构的输出遗忘门取xt和ht作为输入信息,利用s i g m o i d激活函数丢弃信息,为ft(Wfht,xtbf)()s i g m o i d激活函数决定更新哪个状态,再将该状态与由t a n h函数创建的向量Ct相加,输入门的计算公式为it(Wtht,xtbi)()Ctt a n h(Wcht,xtbc)()CtftCt itCt()结合当前记忆Ct和长期记忆Ct,形成新的细胞状态Ct输出门产生ht的最终输出整个过程分为两个阶段,为ot(Woht,xtbo)()htott a n h(Ct)()式中:t a n h函数为双曲正切激活函数;为S i g m o i d函数;Wf,Wt,Wo,Wc为遗忘门、输入门、输出门和记忆单元的权重系数矩阵;bf,bt,bo,bc分别为遗忘门、输入门、输出门和记忆单元的偏置条件 G C N B i L S TM模型本文使用双向L S TM模型来提取交通流数据的时间周期特征双向L S TM的组成是由前向L S TM和反向L S TM叠加,从当前节点的正向和反向两个角度分析交通数据的时间周期性,它不仅可以捕获未来数据的信息特征,而且可以捕获过去数据的信息特征所以在发现重复出现的交通模式,以及在处理数据时比单向L S TM效果更好使预测结果更加准确双向L S TM网络结构见图图双向L S TM网络结构图定义G(V,E)为交通道路网络的拓扑结构每条道路都被视为一个节点,V为一组道路节点,E为边集,邻接矩阵A为道路之间的连接,特征矩阵X为网络中节点的属性因此,交通流预测的时空问题可以看作是路网拓扑结构G和特征X通过一系列变换得到的映射函数f(本文中第期张阳,等:基于G C N B i L S TM的短时交通流预测模型f表示G C N B i L S TM模型)即:Xtf(G;(Xtn,Xt)()G C N B i L S TM模型由图卷积神经网络和双向长短时记忆网络这两部分组成本次研究将长度为n的时间序列数据输入模型,利用G C N结构解析道路网络监测流量的拓扑结构,提取交通流量的空间特征;将具备空间特征的时间序列数据输入B i L S TM中来学习时间特征,通过线性回归层得到预测数据,见图图G C N B i L S TM模型体系 基于G C N B i L S T M的短时交通流预测实现步骤本文利用地感线圈、浮动车等采集的交通流量、速度等数据,实现基于G C N B i L S TM的短时交通流预测该方法主要实现步骤如下步骤采集预测目标路段的历史交通流流量和速度等数据,路网区域内其他路段与目标路段之间的距离以及路网区域内其他路段的历史交通流流量和速度数据并对采集的数据进行分析和预处理步骤设定G C N模型的结构和参数,将数据集转换成拓扑图结构的形式,作为G C N模型的数据输入设定B i L S TM模型的参数,确立输入层、隐含层和输出层等网络结构步骤构建基于G C N B i L S TM的短时交通流预测模型,设置卷积层、池化层、全连接层等显隐层结构,各层神经元利用L S TM模型构建,初步设定显隐层节点数、隐层层数等参数指标步骤选取预测区域内路段的交通流量数据作为训练集和测试集利用训练集训练短时交通流预测模型,再用测试集测试训练好的模型步骤将预测结果和真实值进行分析,检查预测精度是否达到预期目标,如达到则输出预测结果,若未达标则返回步骤,对模型结构参数指标进行重置,直至精度达标 实验结果及分析 数据描述与评价指标本次实验研究范围为福州市晋安区,道路网络由福新路、福新中路、五一北路、五一中路、国货西路、国货东路、连江北路、福马路和古田路等道路构成,具有图结构的区域相关性图见图本实验以福马路某断面为预测对象,研究范围内的其他道路为区域内的空间相关道路图研究对象路网结构示意图实验随机采集福马路 年 月 日共 d的交通流时间序列数据,以m i n为时间步长,每天可划分为 个时间段,共有 个数据样本考虑到周一至周五工作日的交通特性与周末的交通特性有一定的差异,分别对工作日和周末的不同时间段进行交通预测,全面验证算法性能故选取前 d中的工作日和周末这两个时间段为两个训练样本集 性能指标为了定量分析G C N B i L S TM模型性能,引入三种常用的性能指标来比较流量预测模型均方根误差(RMS E):R MS Enni(yiyi)()平均绝对误差(MA E):M A Enni|yiyi|()平均绝对百分比误差(MA P E):M A P ENNi|yiyiyi|()式中:i为样本的数量;yi为实际交通流量序列观测值;yi为交通流量序列预测值 G C N B i L S TM预测性能与算法性能比较将本文提出的G C N B i L S TM与其他算法的短时预测效果进行比较,利用性能指标对预测结构进行评估预测结果见图和表将G C N B i L S TM模型的预测性能与以下个基准模型进行比较:D C R NN扩散卷积递归神经网络,提出交通流建模为向上的扩散卷积,将交通流的空间和时间结合起来,预测未来时刻的武汉理工大学学报(交通科学与工程版)年第 卷图工作日及周末预测结果对比表G C N B i L S TM模型和其他基准模型在工作日预测结果对比评价指标G C N B i L S TMD C R NNT G C NRMS E MA E MA P E/表G C N B i L S TM模型和其他基准模型在周末预测结果对比评价指标G C N B i L S TMD C R NNT G C NRMS E MA E MA P E/交通数据 T G C N时间图卷积网络,该模型通过使用G C N和G RU相结合的方法来同时捕捉空间和时间的依赖关系,进行交通速度预测本文提出的短时交通流预测算法的主要参数设置如下:使用A d a mO p t i m i z e r自动优化,设置初始学习率为 ,设置批量大小为,设置训练次数为 图卷积核设为当G C N图卷积层数设置为,B i L S TM层数设置为时,模型的预测效果最好选择两个图的卷积层,加上四个B i L S TM和一个全连接层的组合模型进行预测交通流数据具有一定的周期性和规律性,周末与工作日的交通流所表现的数据特性差异性较大分别对工作日和周末进行交通流预测,将h设置为一个时间周期,使用过去h的历史数据来预测后 m i n的交通流数据,并与测试集中的真实值进行比较由图和表可知:本文提出的G C N B i L S TM模型在工作日和周末预测效果均优于其他两种经典的预测算法相较于D C R NN,工作日与 周 末RMS E指 标 分 别 少 了 和 ,MA E指 标 少 了 和 ,MA P E指标减少了 和 由实验结果表明,G C N B i L S TM在捕获时空相关性方面效果更好将G C N B i L S TM和T G C N这个预测模型进行比较,G C N B i L S TM和T G C N这两个模型都利用了图卷积的方法从交通数据中提取空间相关性特征相较于T G C N,对于工作日与周末的预测结果,其中RMS E指标分别减少了 和 MA E指 标 分 别 减 少 了 和 MA P E指 标 分 别 减 少 了 和 在数据集上的 m i n交通流预测任务中,G C N B i L S TM的预测准确率高于T G C N该实验结果表明了G C N B i L S TM模型在挖掘交通信息的时空相关性特征方面具有优势结 束 语本文提出了一种基于G C N B i L S TM的城市道路网络预测交通流模型该模型通过建立观测点邻接矩阵,利用G C N和B i L S TM分别捕捉道路交通流数据的时间特征和空间特征结果表明:G C N B i L S TM组合模型具有较高的预测精度,预测的交通流数据相比与另外两个模型与真实交通流数据保持了较高的一致性,是一种实用的短时交通流预测模型后续研究将考虑天气、事故等外部因素对交通流量的影响,进一步提高模型的预测精度参 考 文 献WAN GFY P a r a l l e l c o n t r o l a n dm a n a g e m e n t f o r i n t e l l i g e n tt r a n s p o r t a t i o ns y s t e m s:c o n c e p t s,a r c h i t e c t u r e s,a n da p p l i c a t i o n sJ I E E ET r a n s a c t i o n so n I n t e l l i g e n tT r a n s p o r t a t i o nS y s t e m s,():KUMA RSV,VANA J AK S H IL S h o r t t e r mt r a f f i cf l o wp r e d i c t i o nu s i n gs e a s o n a lA R I MA m o d e lw i t hl i m i t e d i n p u td a t aJE u r o p e a nT r a n s p o r tR e s e a r c hR e v i e w,():KUMA RSV T r a f f i cf l o wp r e d i c t i o nu s i n gK a l m a nf i l t e r i n gt e c h n i q u eJP r o c e d i aE n g i n e e r i n g,():第期张阳,等:基于G C N B i L S TM的短时交通流预测模型WAN GXX,X UL H S h o r t t e r mt r a f f i cf l o wp r e d i c t i o nb a s e do nd e e pl e a r n i n gJT r a n s p o r t a t i o nS y s t e mE n g i n e e r i n ga n dI n f o r m a t i o n,():李林超,张健,杨帆,等基于核函数切换和支持向量回归的交通量短时预测模型J东南大学学报(自然科学版),():C A IP,WANG Y,L U G,e ta l As p a t i o t e m p o r a lc o r r e l a t i v ek n e a r e s tn e i g h b o r m o d e lf o rs h o r t t e r mt r a f f i cm u l t i s t e pf o r e c a s t i n gJT r a n s p o r t a t i o nR e s e a r c hP a r tC:E m e r g i n gT e c h n o l o g i e s,():B RUNAJ,Z A R EMB A W,S Z L AM A,e t a l S p e c t r a ln e t w o r k s a n d l o c a l l y c o n n e c t e d n e t w o r k s o ng r a p h sJJ o u r n a lo f A d v a n c e d T r a n s p o r t a t i o n,():X I EYX,Z HOU G Z,HE W H,e ta l A n o v e lt r a f f i cf l o wf o r e c a s t i n gm e t h o db a s e do nR NN G C Na n dB R BJ J o u r n a lo fA d v a n c e dT r a n s p o r t a t i o n,():WUZ,HUAN G M,Z HA O AT r a f f i cp r e d i c t i o nb a s e do nG C N L S TM m o d e lC J o u r n a lo fP h y s i c s:C o n f e r e n c eS e r i e s,L o n d o n,张阳,胡月,辛东嵘一种考虑时空关联的深度学习短时交通流预测方法J智能科学与技术学 报,():陈丹蕾,陈红,任安虎考虑时空影响下的图卷积网络短时交通流预测J计算机工程与应用,():WUZ,HUAN G M,Z HA O AT r a f f i cp r e d i c t i o nb a s e do nG C N L S TM m o d e lC J o u r n a lo fP h y s i c s:C o n f e r e n c eS e r i e s,I O PP u b l i s h i n g,():S h o r t t e r mT r a f f i cF l o wP r e d i c t i o nM o d e lB a s e do nG C N B i L S TMZ H A N GY a n g)H UY u e)C H E ND e w a n g,)C H E NY u n f e i)(S c h o o l o fT r a n s p o r t a t i o n,F u j i a nU n i v e r s i t yo fT e c h n o l o g y,F u z h o u ,C h i n a)(C o l l e g e o fC o m p u t e ra n dD a t aS c i e n c e,F u z h o uU n i v e r s i t y,F u z h o u ,C h i n a)(S c h o o l o fE l e c t r o n i c,E l e c t r i c a lE n g i n e e r i n ga n dP h y s i c s,F u j i a nU n i v e r s i t yo fT e c h n o l o g y,F u z h o u ,C h i n a)A b s t r a c t:As h o r t t e r mt r a f f i cf l o wf o r e c a s t i n gm o d e lc o m b i n i n gg r a p hc o n v o l u t i o nn e t w o r k(G C N)a n db i d i r e c t i o n a l l o n g t e r m m e m o r yn e u r a ln e t w o r k(B i L S TM)w a sp r o p o s e d T h et o p o l o g i c a l s t r u c t u r eo f r o a dn e t w o r kw a se x t r a c t e db yg r a p hc o n v o l u t i o nn e t w o r kt os o l v e t h e t o p o l o g i c a l r e l a t i o n s h i pp r o b l e ma n de x t r a c tt h es p a t i a lc h a r a c t e r i s t i c sb e t w e e nr o a dn e t w o r k s T h eb i d i r e c t i o n a l l o n g t e r ma n ds h o r t t e r mm e m o r yn e u r a l n e t w o r kw a su s e d t o l e a r n t h ed y n a m i c c h a n g e so f t r a f f i cd a t a t oo b t a i nt h e t i m ec o r r e l a t i o n,a n d t h eG C N B i L S TM m o d e lw a s f u s e d t or e a l i z e t h e t r a f f i c f l o wp r e d i c t i o nc o n s i d e r i n gt h e t i m e s p a c er e l a t i o n s h i po f t h er o a dn e t w o r k T h er e s u l t ss h o wt h a t t h em e t h o dp r o p o s e di nt h i sp a p e rc a nb e t t e ra d a p tt ot h et r a f f i cf l o wu n d e rd i f f e r e n tt r a f f i cf l o wc h a r a c t e r i s t i c s,a n dt h ep r e d i c t i o nd e v i a t i o no fw o r k i n gd a y sa n dw e e k e n d s i s r e d u c e db y a n d c o m p a r e dw i t ht h ec l a s s i c a l a l g o r i t h mK e yw o r d s:d e e pl e a r n i n g;t r a f f i cp r e d i c t i o n;B i L S TM;G C N;u r b a nr o a dn e t w o r k武汉理工大学学报(交通科学与工程版)年第 卷