-
题名基于时间图注意力的交通流量预测模型
被引量:2
- 1
-
-
作者
姚晓敏
张心蓝
张振国
-
机构
延边大学工学院智能信息处理研究室
-
出处
《计算机应用研究》
CSCD
北大核心
2022年第3期770-773,779,共5页
-
基金
国家自然科学基金资助项目(62162062)。
-
文摘
交通状况预测是智能交通系统的一个重要组成部分,而车流量是交通状况最直接的体现,因而对交通流量进行预测具有重要的应用价值。一方面,城市中的道路本身带有空间拓扑性质,另一方面车流量随时间动态变化。因此交通流量预测问题的关键在于对数据中存在的时间和空间依赖进行建模。针对这一特性,使用神经网络模型和注意力机制来探索交通流量数据中的时空依赖关系,提出基于时间图注意力的交通流量预测模型。空间依赖方面,使用图卷积网络与注意力结合的学习算法对不同影响程度节点分配不同的权重,加入节点自适应学习,有效提取空间特征;时间依赖方面,使用时序卷积网络对时间特征进行提取,通过扩张卷积扩大感受域从而捕获较长时间序列数据的特征。由图注意力网络和时间卷积网络构成一个时空网络层,最终连接到输出层输出预测结果。该模型使用图卷积神经网络和注意力机制结合的方式提取空间特征,充分考虑了道路间的空间关系,利用时序卷积网络捕获时间特征。在两个真实的数据集上进行实验后发现,在未来15 min、30 min、60 min的时间段内该模型都有良好表现,结果优于现有基准模型。
-
关键词
交通预测
时间依赖
空间依赖
时序卷积网络
注意力机制
-
Keywords
traffic prediction
temporal dependence
spatial dependence
temporal convolutional network
attention mechanism
-
分类号
TP391
[自动化与计算机技术—计算机应用技术]
-