基于图注意力机制的城市路网短时交通速度预测

作者：杨婧琰; 郑玉卿; 景荣荣; 周烽; 张坤鹏^*

来源：现代信息科技, 2022, 6(22): 86-92.

DOI：10.19850/j.cnki.2096-4706.2022.22.021

• 短时交通速度预测
• 时空相关性
• 图注意力网络
• 门控循环单元

针对城市路网短时交通速度预测问题，在考虑路网交通状态时空相关性的情况下，提出一种基于图注意力机制的预测方法。该方法利用图注意力网络(GAT)和门控循环单元(GRU)构建了GAT-GRU模型，在路网层面对交通状态的时空相关性进行有效地建模，进而预测路网短时交通速度。以城市道路网的交通速度数据为数据源展开数值实验，结果表明GAT-GRU模型的表现均优于对比模型。

• 单位
  清华大学; 河南工业大学