一种用于交通预测的自适应时空图神经网络

甘萍, 林基明, 农丽萍, 王俊义

桂林电子科技大学学报 ›› 2023, Vol. 43 ›› Issue (01) : 7-13. DOI: 10.16725/j.cnki.cn45-1351/tn.2023.01.010

一种用于交通预测的自适应时空图神经网络

  • 甘萍, 林基明, 农丽萍, 王俊义
作者信息 +
History +

摘要

针对以往复杂的神经网络构架在空间维度上所采用的预定义图结构未包含完整交通数据空间信息,且在时间维度上不能很好地捕获交通数据长期依赖关系的问题,提出一种新的时空图神经网络。通过自适应图卷积网络(AGCN)自动捕获节点的特定状态以及自动推断不同节点之间的相互依赖关系,提取更完整的交通数据空间特征,再通过时空长短期记忆网络(ST-LSTM)中的时间记忆模块来提取交通数据的时间特征,捕获短中长期的时间依赖关系。在PeMSD4和PeMSD8数据集上进行了验证,实验结果表明,所提网络相比基线模型能够更好地提升交通预测性能。

关键词

交通预测 / 自适应图卷积网络 / 时空相关性 / 时空图神经网络 / 长短期记忆网络

中图分类号

TP183 / U491.14

引用本文

导出引用
甘萍, 林基明, 农丽萍, 王俊义. 一种用于交通预测的自适应时空图神经网络. 桂林电子科技大学学报. 2023, 43(01): 7-13 https://doi.org/10.16725/j.cnki.cn45-1351/tn.2023.01.010

基金

国家自然科学基金(61966007); 广西自然科学基金(2020GXNSFAA159105); 广西无线宽带通信与信号处理重点实验室基金(GXKL06190204,GXKL06200116,GXKL06190117); 认知无线电与信息处理教育部重点实验室基金(CRKL180201,CRKL180106)

评论

Accesses

Citation

Detail

段落导航
相关文章

/