CN116168548A - 一种基于多特征融合的时空注意图卷积网络的交通流预测方法 - Google Patents

一种基于多特征融合的时空注意图卷积网络的交通流预测方法 Download PDF

Info

Publication number
CN116168548A
CN116168548A CN202310165586.6A CN202310165586A CN116168548A CN 116168548 A CN116168548 A CN 116168548A CN 202310165586 A CN202310165586 A CN 202310165586A CN 116168548 A CN116168548 A CN 116168548A
Authority
CN
China
Prior art keywords
time
data
matrix
representing
traffic flow
Prior art date
Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
Pending
Application number
CN202310165586.6A
Other languages
English (en)
Inventor
陈雅君
丁治明
郭黎敏
Current Assignee (The listed assignees may be inaccurate. Google has not performed a legal analysis and makes no representation or warranty as to the accuracy of the list.)
Beijing University of Technology
Original Assignee
Beijing University of Technology
Priority date (The priority date is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the date listed.)
Filing date
Publication date
Application filed by Beijing University of Technology filed Critical Beijing University of Technology
Priority to CN202310165586.6A priority Critical patent/CN116168548A/zh
Publication of CN116168548A publication Critical patent/CN116168548A/zh
Pending legal-status Critical Current

Links

Images

Classifications

    • GPHYSICS
    • G08SIGNALLING
    • G08GTRAFFIC CONTROL SYSTEMS
    • G08G1/00Traffic control systems for road vehicles
    • G08G1/065Traffic control systems for road vehicles by counting the vehicles in a section of the road or in a parking area, i.e. comparing incoming count with outgoing count
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06NCOMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
    • G06N3/00Computing arrangements based on biological models
    • G06N3/02Neural networks
    • G06N3/08Learning methods
    • G06N3/084Backpropagation, e.g. using gradient descent
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06QINFORMATION AND COMMUNICATION TECHNOLOGY [ICT] SPECIALLY ADAPTED FOR ADMINISTRATIVE, COMMERCIAL, FINANCIAL, MANAGERIAL OR SUPERVISORY PURPOSES; SYSTEMS OR METHODS SPECIALLY ADAPTED FOR ADMINISTRATIVE, COMMERCIAL, FINANCIAL, MANAGERIAL OR SUPERVISORY PURPOSES, NOT OTHERWISE PROVIDED FOR
    • G06Q10/00Administration; Management
    • G06Q10/04Forecasting or optimisation specially adapted for administrative or management purposes, e.g. linear programming or "cutting stock problem"
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06QINFORMATION AND COMMUNICATION TECHNOLOGY [ICT] SPECIALLY ADAPTED FOR ADMINISTRATIVE, COMMERCIAL, FINANCIAL, MANAGERIAL OR SUPERVISORY PURPOSES; SYSTEMS OR METHODS SPECIALLY ADAPTED FOR ADMINISTRATIVE, COMMERCIAL, FINANCIAL, MANAGERIAL OR SUPERVISORY PURPOSES, NOT OTHERWISE PROVIDED FOR
    • G06Q50/00Information and communication technology [ICT] specially adapted for implementation of business processes of specific business sectors, e.g. utilities or tourism
    • G06Q50/40Business processes related to the transportation industry
    • GPHYSICS
    • G08SIGNALLING
    • G08GTRAFFIC CONTROL SYSTEMS
    • G08G1/00Traffic control systems for road vehicles
    • G08G1/01Detecting movement of traffic to be counted or controlled
    • G08G1/0104Measuring and analyzing of parameters relative to traffic conditions
    • G08G1/0125Traffic data processing
    • G08G1/0129Traffic data processing for creating historical data or processing based on historical data
    • YGENERAL TAGGING OF NEW TECHNOLOGICAL DEVELOPMENTS; GENERAL TAGGING OF CROSS-SECTIONAL TECHNOLOGIES SPANNING OVER SEVERAL SECTIONS OF THE IPC; TECHNICAL SUBJECTS COVERED BY FORMER USPC CROSS-REFERENCE ART COLLECTIONS [XRACs] AND DIGESTS
    • Y02TECHNOLOGIES OR APPLICATIONS FOR MITIGATION OR ADAPTATION AGAINST CLIMATE CHANGE
    • Y02TCLIMATE CHANGE MITIGATION TECHNOLOGIES RELATED TO TRANSPORTATION
    • Y02T10/00Road transport of goods or passengers
    • Y02T10/10Internal combustion engine [ICE] based vehicles
    • Y02T10/40Engine management systems

Landscapes

  • Engineering & Computer Science (AREA)
  • Physics & Mathematics (AREA)
  • General Physics & Mathematics (AREA)
  • Business, Economics & Management (AREA)
  • Theoretical Computer Science (AREA)
  • Human Resources & Organizations (AREA)
  • Economics (AREA)
  • Strategic Management (AREA)
  • Health & Medical Sciences (AREA)
  • General Health & Medical Sciences (AREA)
  • Tourism & Hospitality (AREA)
  • General Business, Economics & Management (AREA)
  • Marketing (AREA)
  • Game Theory and Decision Science (AREA)
  • Data Mining & Analysis (AREA)
  • Operations Research (AREA)
  • Entrepreneurship & Innovation (AREA)
  • Development Economics (AREA)
  • Analytical Chemistry (AREA)
  • Life Sciences & Earth Sciences (AREA)
  • Artificial Intelligence (AREA)
  • Biomedical Technology (AREA)
  • Biophysics (AREA)
  • Computational Linguistics (AREA)
  • Quality & Reliability (AREA)
  • Evolutionary Computation (AREA)
  • Chemical & Material Sciences (AREA)
  • Molecular Biology (AREA)
  • Computing Systems (AREA)
  • General Engineering & Computer Science (AREA)
  • Mathematical Physics (AREA)
  • Software Systems (AREA)
  • Primary Health Care (AREA)
  • Traffic Control Systems (AREA)

Abstract

本发明公开了一种用于交通流预测的多特征融合的时空注意图卷积神经网络,从数据处理、空间相关性、时间相关性三个角度出发设计。首先,对获取到的历史交通流数据与外部特征因素融合,获取多特征融合的时空交通数据作为训练样本;其次,利用注意图卷积网络捕获城市道路网络的拓扑结构,获得多特征融合的交通流空间特征;接着,将获得的具有空间特征的时间序列输入到由门控循环单元和注意力单元相结合的时间特征提取模型中捕捉时间特征;最后,将输出的预测值与实际值相对比,更改相关参数直至模型最优化。

Description

一种基于多特征融合的时空注意图卷积网络的交通流预测 方法
技术领域
本发明属于交通流预测领域,具体涉及一种基于多特征融合的时空注意图卷积网络的交通流预测方法。
背景技术
在智能化城市建设当中,由于交通拥堵越来越严重,交通问题越来越被人们所重视。科技迅速的发展的同时,交通***也日益繁杂,智能交通***(IntelligentTransportation System,ITS)应运而生。及时准确的交通预测可以帮助人们更好的做好出行的决策,从而缓解城市的交通拥堵,减少了交通出行对大气的污染,最终既保证了交通的高效出行,又保证了汽车尾气的最低排放量。
现有的关于交通流预测的研究主要包括三类方法,分别是基于传统统计学模型、传统机器学***均模型(Historical Average,HA)、卡尔曼滤波模型(Kalman Filter,KF)和时间序列预测模型。在传统统计学模型中,最为经典的就是差分整合移动平均自回归模型(Autoregressive Integrated Moving Average model,ARIMA)及其变体。ARIMA通过分析自相关图和偏相关图得到被观测的***时间序列数据的预测值。然而,该模型是在平稳的时间序列的基础上建立起来的,对现今大规模动态的路网上并不适用。除此之外,该模型忽略了路网的时空特征,不适用于大型路网的交通流预测。
第二种方法是数据驱动的,无需数据满足平稳性的假设,并且此类方法能够对稍微复杂的数据进行建模并做出更精准的预测。常见的模型主要是支持向量机(SupportVector Machine,SVM)、K近邻(K-Nearest Neighbor,KNN)和人工神经网络(ArtificialNeural Network,ANN)。支持向量机的模型不仅可以提高交通流预测的精度,还大大减少了计算量。利用K近邻非参数回归模型预测短时交通量,不仅提高了预测算法的精度,还提高了预测算法的执行效率。然而,大多数传统的机器学习的模型都依赖于特征工程,无法将时间依赖性和空间依赖性同时建模,因此有一定的局限性。
第三种方法通过融合循环神经网络(Recurrent Neural Network,RNN)或者融合图卷积神经网络(Graph Convolution Network,GCN)不仅可以自动学习交通流的非线性特征并自动捕捉数据模式,还可以充分的利用时空特征。但是,现实生活中依然有大量的非欧几里得空间的数据需要处理,通过传统的机器学习方法和RNN等将其拓展为图卷积和RNN及其变体的时空预测框架,用此来解决非欧几里得空间的交通流预测问题。以上模型都只关注了交通数据这一部分因素,但交通流的变化除了受到时空的影响外还受一些外在因素的影响。
发明内容
本发明旨在提供一种基于多特征融合的时空注意图卷积网络的交通流预测方法,以解决上述技术存在的问题。
为实现上述目的,本发明提供了一种基于多特征融合的时空注意图卷积网络的交通流预测方法,包括以下步骤:
步骤1:构造图网络结构,通过计算交通节点之间的道路关系建立关系矩阵;
步骤2:获取各个交通节点的历史交通流量数据和特征数据,通过特征数据融合交通流量数据的方式获取多特征融合的时空交通数据;
步骤3:构建时空注意图卷积网络模型,将步骤2所述的交通数据作为该模型的输入,提取交通流量数据的空间特征和时间特征;
步骤4:定义模型损失函数,根据损失函数反向传播优化模型参数,最后实现道路交通流的预测。
进一步的,步骤1中图网络结构由一个邻接矩阵A表示,Aij表示第i个节点和第j个节点的连通性,若存在空间连接关系,则Aij为1,否则为0。
进一步的,步骤2中具体包括:
获取道路网络上的交通信息作为网络中节点的属性特征,用矩阵X表示,交通信息每固定时间取样一次,总共取得T个样本,对于一个节点得到的数据表示为X=(x1,x2,…,xT-1,xT),则N个节点的道路网络用矩阵表示为
Figure BDA0004095819640000021
式中xNT表示第N个节点T时刻的交通流数据信息,(x1T,x2T,…,xNT)表示第T时刻的所有节点的交通流数据信息;
获取影响该道路网络的辅助属性,用矩阵S表示,辅助属性每固定时间取样一次,总共取得T个样本,对于单个辅助属性得到的数据表示为S=(s1,s2,…,sT-1,sT),则M个辅助属性用矩阵表示为
Figure BDA0004095819640000022
式中sMT表示第M个辅助属性T时刻的数据信息。(s1T,s2T,…,sMT)表示第T时刻的所有辅助的外部属性特征数据信息;
获取多特征融合的时空数据,用矩阵F表示,将道路网络的交通特征矩阵和影响该道路网络的辅助属性在同一时间内进行融合,则T时刻内最终的多特征数据矩阵表示为
Figure BDA0004095819640000023
式中xNT表示第N个节点T时刻的交通流数据信息,sMT表示第M个辅助属性T时刻的数据信息,(x1T,x2T,…,xNT,s1T,s2T,…,sMT)表示第T时刻N个节点和M个外部辅助属性所融合成的多特征数据信息。
进一步的,步骤3中具体包括:
构建空间图卷积层,本模型采用的图卷积计算方法为
Figure BDA0004095819640000024
Figure BDA0004095819640000025
Figure BDA0004095819640000026
式中Fl+1表示图卷积网络的第1层的输出,A是邻接矩阵,I是单位矩阵,σ(·)表示非线性激活函数,
Figure BDA0004095819640000027
是带自环邻接矩阵,/>
Figure BDA0004095819640000028
是/>
Figure BDA0004095819640000029
的度矩阵,是对角矩阵,Wl为第1层的可训练权重参数矩阵,Fl表示第1层的输入数据;
构建时间特征提取层,本层采用门控循环单元和Transformer相结合的方式,门控循环单元用以捕捉每个空间节点在时间维度上的短时依赖,通过该单元忘记ht-1中的某些信息,并加入当前节点输入的某些信息。GRU的建模过程如下:
Figure BDA00040958196400000210
Figure BDA00040958196400000211
rt=σ(Wr·[ht-1,xt]+br)
zt=σ(Wz·[ht-1,xt]+bz)
式中(1-zt)*ht-1表示t-1时刻数据输入的选择性遗忘,
Figure BDA00040958196400000212
表示对当前时刻数据的选择性记忆;
式中ht表示t时刻的门控循环单元的数据输出,即传递到下一时刻的隐藏状态,zt表示更新门,
Figure BDA00040958196400000213
表示t时刻的候选隐藏状态,ht-1表示上一时刻的隐藏状态,tan(·)表示激活函数,rt表示重置门,xt表示为t时刻的输入信息,σ(·)表示非线性激活函数,W,Wr,wz,b,br和bz是可学习参数矩阵;
Transformer包含多头注意力层、前向反馈层、残差化和归一化层,将上层门控循环单元的值作为输入,经过位置编码后传入多头注意层后,多头注意力层的输入由序列中所有位置的查询矩阵
Figure BDA0004095819640000031
维度为dk的键矩阵/>
Figure BDA0004095819640000032
和维度为dv的值矩阵/>
Figure BDA0004095819640000033
组成,注意力层的公式为
Figure BDA0004095819640000034
Figure BDA0004095819640000035
Figure BDA0004095819640000036
Figure BDA0004095819640000037
式中Attention(·)代表是实现注意力机制的函数,softmax(·)代表是归一化指数函数,KT代表是键矩阵的转置,dk代表是隐藏层的维度,
Figure BDA00040958196400000315
表示经过位置编码后的输入,WQ,WK,WV为三个可学习的参数矩阵,/>
Figure BDA0004095819640000038
分别代表是查询矩阵、键矩阵和值矩阵。
进一步的,步骤4中的具体包括:
根据初始化的邻接矩阵A,以及模型参数矩阵W,进行交通数据预测过程,预测结果与真值之间存在误差,经变分推断方法,其变分损失的损失函数如下:
Figure BDA0004095819640000039
/>
式中
Figure BDA00040958196400000310
代表预测的交通信息,Xt+1代表输入的交通信息,Wθ表示本文公式中需要训练的参数;
使用平均绝对误差(Mean Absolute Error,MAE),平均百分比误差(MeanAbsolute Percentage Error,MAPE)和均方根误差(Root Mean Square Error,RMSE)作为评估指标,公式如下:
Figure BDA00040958196400000311
Figure BDA00040958196400000312
Figure BDA00040958196400000313
式中xi表示真实的交通信息数据,
Figure BDA00040958196400000314
表示模型预测的交通信息数据,n表示一共的训练样本数。
与现有技术相比较,本发明具有如下有益效果:
1.本发明提出了一种基于多特征数据融合的时空注意图卷积网络的交通流预测模型,通过将外部特征数据融合到交通数据中,弥补了原有单一数据的缺陷;
2.传统的图卷积网络模型在捕获空间依赖性效果不佳,本发明通过将图卷积单元与注意力机制层相结合来捕获各节点在时间维度上的依赖性,在注意图卷积单元中,可以动态地进行时间上和空间上的特征提取。
附图说明
图1为本发明基于多特征融合的时空注意图卷积网络交通流预测的流程图。
图2为本发明模型结构示例图。
图3为本发明多特征数据融合模块结构示例图。
图4为本发明注意图卷积模块结构示例图。
具体实施方式
下面将结合本说明书实施例中的附图,对本说明书实施例中的技术方案进行详清楚、完整的描述。显然,所描述的实施例仅是本说明书的一部分实施例,而不是全部的实施例。
如图1所示,本发明公开一种基于多特征融合的时空注意图卷积网络的交通流预测方法的处理流程图,包括以下步骤:
步骤1:构造图网络结构G=(V,E,A)。通过计算传感器之间成对道路网络距离,利用阈值高斯核建立邻接矩阵A,公式表达如下:
Figure BDA0004095819640000041
式中V代表道路节点集合,E代表节点与节点之间的边集合,即vi与vj之间的实际距离,K代表是控制邻接矩阵A的阈值,σ代表是距离的标准差。
步骤2:构造多特征融合矩阵F。如图3所示,对收取到的历史交通流数据和外部特征数据进行数据采集,获取训练样本。获取到的交通流历史时刻数据为(X1,X2,…,XT-1,XT),外部特征数据为(S1,S2,…,ST-1,ST),其中T为取样窗口大小,即历史交通流数据的数量,该数据每隔10分钟进行一次数据采集,取T′+T个连续时刻的交通流数据作为一个训练样本,记T′个连续时刻的交通流数据为历史交通流数据,即T个连续时刻的交通流数据为真实交通流数据。将外部特征数据作为独立的交通流数据特征,在相同连续时刻内进行一个等值融合,最后融合后的数据为:
Figure BDA0004095819640000042
步骤3:构造注意力图卷积网络模型,如图2所示。将融合后的多特征数据作为输入进入带有空间注意力机制的图卷积网络模型中,如图4所示。首先使用空间注意力机制捕获当前时间下的全局图结点特征MAt,相关注意力机制公式如下:
MAt=contact(h1,h2,…,hk)W°
Figure BDA0004095819640000043
Figure BDA0004095819640000044
Qt=FtWQ
Kt=FtWK
Vt=FtWV
式中,WQ,WK,WV为三个可学习的参数矩阵,softmax(·)代表是归一化指数函数,Attention(·)代表是实现注意力机制的函数,w°代表的是一个可学习的投影矩阵,
Figure BDA0004095819640000045
代表是t时间第i个注意力的输出,contact(·)代表是拼接操作;
接着,将上一时刻的各节点特征Ht-1={h1,h2,…,hN}与当前时刻通过注意力机制得到的各节点特征利用门控机制进行融合,得到注意力机制的图滤波AGt,公式表示如下:
AGt=zt*MAt+(1-zt)*H′t-1
zt=σ(MAtWz1+H′t-1Wz2)
H′t-1=Ht-1Wh+bh
式中,Wz1,Wz2,Wh和bh是可学习参数,σ(·)代表是激活函数;
接着,将注意力图滤波得到的输出进入到图卷积模块进行空间特征的提取,图卷积的公式如下:
Figure BDA0004095819640000046
式中Fl+1表示输出,
Figure BDA0004095819640000047
是带自环邻接矩阵,即邻接矩阵A和单位矩阵I的和,D是A的度矩阵,σ是非线性激活函数,Wl为第1层的可训练权重参数矩阵,AGl为输入,AGl+1为输出;
接着进行时间特征的提取,时间特征的提取是由门控循环单元和Transformer来构成;
具体的,门控循环单元过程将当前时刻的多特征数据Ft与多注意图卷积模块在上一时刻t-1的输出进行拼接,放入多注意图卷积模块,门控循环单元的公式如下:
Figure BDA0004095819640000051
Figure BDA0004095819640000052
rt=σ(Wr·[ht-1,xt]+br)
zt=σ(Wz·[ht-1,xt]+bz)
式中zt和rt分别表示为更新门和重置门,W,Wr,Wz,b,br和bz是可学习参数矩阵;
具体地,在经GRU处理过后输出的结果首先进行常规处理操作,即位置编码,具体过程如下:
X=xi+et
Figure BDA0004095819640000053
式中et表示正余弦编码,i对应的是10维不同维度的数,d代表的是position编码的维度,model代表的是d的维度数;
将经过位置编码后的数据送入多头注意力机制中从全局捕获其时间相关性,多头注意力机制公式如下:
Figure BDA0004095819640000054
Figure BDA0004095819640000055
Figure BDA0004095819640000056
Figure BDA0004095819640000057
将多头注意力输出的结果经过全连接层后得到最终的交通流预测输出,公式如下:
Figure BDA0004095819640000058
式中,W0和b0是可学习参数。
步骤4:设置损失函数,根据初始化的邻接矩阵A以及模型参数矩阵W,本示例采用的损失函数公式如下:
Figure BDA0004095819640000059
式中
Figure BDA00040958196400000510
代表预测的交通信息,Xt+1代表输入的交通信息,Wθ表示本文公式中需要训练的参数;
利用反向传播方法对模型参数不断优化,最后将测试集数据作为模型输入,得出的预测值与真实观察值作比较得出预测精度。
模型性能的评估指标如下:
平均绝对误差(Mean Absolute Error,MAE),结果为实际值与预测值之间绝对误差的平均值,公式如下:
Figure BDA00040958196400000511
平均百分比误差(Mean Absolute Percentage Error,MAPE),结果为实际值与预测值之间绝对百分比误差的平均值,公式如下:
Figure BDA00040958196400000512
均方根误差(Root Mean Square Error,RMSE),:结果为实际值与预测值之间均方误差的算术平方根,公式如下:
Figure BDA00040958196400000513
模型得出三个评价指标后,与当前主流模型对比,仍处于领先地位。
以上对本发明所公开的技术方案进行了详细介绍,应用了具体实施方式对本发明的原理进行了阐述,以上实施方式仅为本发明一种较佳实施例而已,当然不能以此来限定本发明之权利范围,因此依本发明权利要求所作的等同变化,仍属本发明所涵盖的范围。

Claims (5)

1.一种基于多特征融合的时空注意图卷积网络的交通流预测方法,其特征在于,该方法包括以下步骤:
步骤1:构造图网络结构,通过计算交通节点之间的道路关系建立关系矩阵;
步骤2:获取各个交通节点的历史交通流量数据和特征数据,通过特征数据融合交通流量数据的方式获取多特征融合的时空交通数据;
步骤3:构建时空注意图卷积网络模型,将步骤2所述的交通数据作为该模型的输入,提取交通流量数据的空间特征和时间特征;
步骤4:定义模型损失函数,根据损失函数反向传播优化模型参数,最后实现道路交通流的预测。
2.根据权利要求1所述的基于多特征融合的时空注意图卷积网络的交通流预测方法,其特征在于,步骤1中图网络结构由一个邻接矩阵A表示,Aij表示第i个节点和第j个节点的连通性,若存在空间连接关系,则Aij为1,否则为0。
3.根据权利要求1所述的基于多特征融合的时空注意图卷积网络的交通流预测方法,其特征在于,步骤2中具体包括:
获取道路网络上的交通信息作为网络中节点的属性特征,用矩阵X表示,交通信息每固定时间取样一次,总共取得T个样本,对于一个节点得到的数据表示为X=(x1,x2,…,xT-1,xT),则N个节点的道路网络用矩阵表示为
Figure FDA0004095819630000011
式中xNT表示第N个节点T时刻的交通流数据信息,(x1T,x2T,…,xNT)表示第T时刻的所有节点的交通流数据信息;
获取影响该道路网络的辅助属性,用矩阵S表示,辅助属性每固定时间取样一次,总共取得T个样本,对于单个辅助属性得到的数据表示为S=(s1,s2,…,sT-1,sT),则M个辅助属性用矩阵表示为
Figure FDA0004095819630000012
式中sMT表示第M个辅助属性T时刻的数据信息;(s1T,s2T,…,sMT)表示第T时刻的所有辅助的外部属性特征数据信息;
获取多特征融合的时空数据,用矩阵F表示,将道路网络的交通特征矩阵和影响该道路网络的辅助属性在同一时间内进行融合,则T时刻内最终的多特征数据矩阵表示为
Figure FDA0004095819630000013
式中xNT表示第N个节点T时刻的交通流数据信息,sMT表示第M个辅助属性T时刻的数据信息,(x1T,x2T,…,xNT,s1T,s2T,…,sMT)表示第T时刻N个节点和M个外部辅助属性所融合成的多特征数据信息。
4.根据权利要求1所述的基于多特征融合的时空注意图卷积网络的交通流预测方法,其特征在于,步骤3中具体包括:
构建空间图卷积层,本模型采用的图卷积计算方法为
Figure FDA0004095819630000014
Figure FDA0004095819630000021
Figure FDA0004095819630000022
式中Fl+1表示图卷积网络的第l层的输出,A是邻接矩阵,I是单位矩阵,σ(·)表示非线性激活函数,
Figure FDA0004095819630000023
是带自环邻接矩阵,/>
Figure FDA0004095819630000024
是/>
Figure FDA0004095819630000025
的度矩阵,是对角矩阵,Wl为第l层的可训练权重参数矩阵,Fl表示第l层的输入数据;
构建时间特征提取层,本层采用门控循环单元和Transformer相结合的方式,门控循环单元用以捕捉每个空间节点在时间维度上的短时依赖,通过该单元忘记ht-1中的某些信息,并加入当前节点输入的某些信息;GRU的建模过程如下:
Figure FDA0004095819630000026
Figure FDA0004095819630000027
rt=σ(Wr·[ht-1,xt]+br)
zt=σ(Wz·[ht-1,xt]+bz)
式中(1-zt)*ht-1表示t-1时刻数据输入的选择性遗忘,
Figure FDA0004095819630000028
表示对当前时刻数据的选择性记忆;
式中ht表示t时刻的门控循环单元的数据输出,即传递到下一时刻的隐藏状态,zt表示更新门,
Figure FDA0004095819630000029
表示t时刻的候选隐藏状态,ht-1表示上一时刻的隐藏状态,tan(·)表示激活函数,rt表示重置门,xt表示为t时刻的输入信息,σ(·)表示非线性激活函数,W,Wr,Wz,b,br和bz是可学习参数矩阵;
Transformer包含多头注意力层、前向反馈层、残差化和归一化层,将上层门控循环单元的值作为输入,经过位置编码后传入多头注意层后,多头注意力层的输入由序列中所有位置的查询矩阵
Figure FDA00040958196300000210
维度为dk的键矩阵/>
Figure FDA00040958196300000211
和维度为dv的值矩阵/>
Figure FDA00040958196300000212
组成,注意力层的公式为
Figure FDA00040958196300000213
Figure FDA00040958196300000214
Figure FDA00040958196300000215
Figure FDA00040958196300000216
式中Attention(·)代表是实现注意力机制的函数,softmax(·)代表是归一化指数函数,KT代表是键矩阵的转置,dk代表是隐藏层的维度,
Figure FDA00040958196300000217
表示经过位置编码后的输入,WQ,WK,WV为三个可学习的参数矩阵,/>
Figure FDA00040958196300000218
分别代表是查询矩阵、键矩阵和值矩阵。/>
5.根据权利要求1所述的基于多特征融合的时空注意图卷积网络的交通流预测方法,其特征在于,步骤4中的具体包括:
根据初始化的邻接矩阵A,以及模型参数矩阵W,进行交通数据预测过程,预测结果与真值之间存在误差,经变分推断方法,其变分损失的损失函数如下:
Figure FDA00040958196300000219
式中
Figure FDA00040958196300000220
代表预测的交通信息,Xt+1代表输入的交通信息,Wθ表示本文公式中需要训练的参数;
使用平均绝对误差MAE,平均百分比误差MAPE和均方根误差RMSE作为评估指标,公式如下:
Figure FDA00040958196300000221
Figure FDA00040958196300000222
Figure FDA0004095819630000031
式中xi表示真实的交通信息数据,
Figure FDA0004095819630000032
表示模型预测的交通信息数据,n表示一共的训练样本数。/>
CN202310165586.6A 2023-02-24 2023-02-24 一种基于多特征融合的时空注意图卷积网络的交通流预测方法 Pending CN116168548A (zh)

Priority Applications (1)

Application Number Priority Date Filing Date Title
CN202310165586.6A CN116168548A (zh) 2023-02-24 2023-02-24 一种基于多特征融合的时空注意图卷积网络的交通流预测方法

Applications Claiming Priority (1)

Application Number Priority Date Filing Date Title
CN202310165586.6A CN116168548A (zh) 2023-02-24 2023-02-24 一种基于多特征融合的时空注意图卷积网络的交通流预测方法

Publications (1)

Publication Number Publication Date
CN116168548A true CN116168548A (zh) 2023-05-26

Family

ID=86414500

Family Applications (1)

Application Number Title Priority Date Filing Date
CN202310165586.6A Pending CN116168548A (zh) 2023-02-24 2023-02-24 一种基于多特征融合的时空注意图卷积网络的交通流预测方法

Country Status (1)

Country Link
CN (1) CN116168548A (zh)

Cited By (3)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN116361662A (zh) * 2023-05-31 2023-06-30 中诚华隆计算机技术有限公司 机器学习模型的训练方法及量子网络设备性能预测方法
CN116885698A (zh) * 2023-06-16 2023-10-13 西安理工大学 基于多图融合的电动汽车充电站群负荷预测方法及装置
CN117150326A (zh) * 2023-10-31 2023-12-01 深圳市大数据研究院 新能源节点出力功率的预测方法、装置、设备及存储介质

Cited By (5)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN116361662A (zh) * 2023-05-31 2023-06-30 中诚华隆计算机技术有限公司 机器学习模型的训练方法及量子网络设备性能预测方法
CN116361662B (zh) * 2023-05-31 2023-08-15 中诚华隆计算机技术有限公司 机器学习模型的训练方法及量子网络设备性能预测方法
CN116885698A (zh) * 2023-06-16 2023-10-13 西安理工大学 基于多图融合的电动汽车充电站群负荷预测方法及装置
CN117150326A (zh) * 2023-10-31 2023-12-01 深圳市大数据研究院 新能源节点出力功率的预测方法、装置、设备及存储介质
CN117150326B (zh) * 2023-10-31 2024-02-23 深圳市大数据研究院 新能源节点出力功率的预测方法、装置、设备及存储介质

Similar Documents

Publication Publication Date Title
CN110889546B (zh) 一种基于注意力机制的交通流量模型训练方法
CN112801404B (zh) 一种基于自适应空间自注意力图卷积的交通预测方法
CN109492822B (zh) 空气污染物浓度时空域关联预测方法
CN116168548A (zh) 一种基于多特征融合的时空注意图卷积网络的交通流预测方法
CN111967343B (zh) 基于简单神经网络和极端梯度提升模型融合的检测方法
CN112241814A (zh) 一种基于强化时空图神经网络的交通预测方法
CN111899510A (zh) 基于发散卷积和gat的智能交通***流量短期预测方法及***
CN108009674A (zh) 基于cnn和lstm融合神经网络的空气pm2.5浓度预测方法
CN114330671A (zh) 一种基于Transformer时空图卷积网络的交通流预测方法
CN113487061A (zh) 一种基于图卷积-Informer模型的长时序交通流量预测方法
CN114802296A (zh) 一种基于动态交互图卷积的车辆轨迹预测方法
CN113673769B (zh) 基于多元时间序列插补的图神经网络交通流预测方法
CN111612243A (zh) 交通速度预测方法、***及存储介质
CN112085163A (zh) 一种基于注意力增强图卷积神经网络agc和门控循环单元gru的空气质量预测方法
CN112765896A (zh) 一种基于lstm的水处理时序数据异常检测方法
CN110675632B (zh) 针对多特征空间和数据稀疏的车辆短时轨迹预测控制方法
CN113112791A (zh) 一种基于滑动窗口长短时记忆网络的交通流量预测方法
CN112785077A (zh) 基于时空数据的出行需求预测方法及***
CN115376317B (zh) 一种基于动态图卷积和时序卷积网络的交通流预测方法
Ouyang et al. Spatial-temporal dynamic graph convolution neural network for air quality prediction
CN114973678B (zh) 一种基于图注意力神经网络和时空大数据的交通预测方法
CN115759461A (zh) 一种面向物联网的多元时间序列预测方法及***
Kulkarni et al. SLANT+: A nonlinear model for opinion dynamics in social networks
Shao et al. Failure detection for motion prediction of autonomous driving: An uncertainty perspective
CN117494034A (zh) 基于交通拥堵指数和多源数据融合的空气质量预测方法

Legal Events

Date Code Title Description
PB01 Publication
PB01 Publication
SE01 Entry into force of request for substantive examination
SE01 Entry into force of request for substantive examination