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

摘要

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

关键词

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

基金资助

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