摘要:精准预测路网交通流是保障智能交通系统高效运行的基础. 针对现有方法存在难以有效建模交通流数据中复杂非线性时空动态依赖关系的问题, 提出一种基于增强时空Transformer (ESTformer)的交通流预测方法. 所提出方法设计多尺度时间Transformer和增强空间Transformer分别建模交通流序列数据间的时间依赖关系和路网不同节点间的空间依赖关系: 多尺度时间Transformer构建短期门控卷积网络来捕获交通流数据中的短期时间依赖关系, 并引入时间多头自注意力机制以捕获长期动态时间依赖关系; 增强空间Transformer通过对偶变换来增强键向量的特征表达能力, 利用时变掩码矩阵动态更新键向量, 提高模型的同时捕获节点特征和边特征的能力. 在4个真实交通流数据集上的测试结果表明, 与基线方法相比, 所提出基于ESTformer的交通流预测方法具有更优越的预测性能. 相比于13种基线方法在不同数据集上表现最佳者, 所提出方法在12个时间步上的平均绝对误差(MAE)和均方根误差(RMSE)分别改进了1.14% $\sim $ 3.88%, 0.36% $\sim $ 1.78%.