李志鹃 , 张文标 , 孙兴 , 于淼 , 龚朝晖 , 王巍琳
控制工程. 2024, 31(8): 1426-1435.
如何准确且快速地预测长期的交通流量十分具有挑战。大多数研究采用的图卷积网络(graph convolution network, GCN)模型多用 k 阶切比雪夫多项式近似,同时需要特征分解或类似计算复杂的矩阵操作,且感受野有限,无法捕获动态的空间相关性。为了改进这种做法,提出了一种结合图注意增强网络(enhanced graph attention network, EGAT)、双向长短时记忆(bi-directional long short-term memory, Bi-LSTM)网络和残差网络(residual
network, ResNet)的多时空残差网络(multi-spatial-temporal residual network, MST-ResNet),用以捕获交通流量的时空动态相关性。该网络采用了
EGAT 提取道路路段之间的动态空间依赖关系。同时,将 EGAT 的输出作为输入,该网络引入 Bi-LSTM 提取时间和周期特性,同时增加残差模块以构建时空残差网络,最后三组件融合进行预测。基于 2 个真实数据集的实验结果表明,所提 MST-Res-Net 用于预测多步交通流量快速有效,优于现有先进的基线方法,且其训练时间仅需要先进基线方法的一半,更便于实时控制。