| 面向改进的时空Transformer的交通流量预测模型 |
| |
| 作者姓名： | 高榕 万以亮 邵雄凯 吴歆韵 |
| |
| 作者单位： | 1. 湖北工业大学计算机学院;2. 南京大学计算机软件新技术国家重点实验室 |
| |
| 基金项目： | 国家自然科学基金（61902116）；;南京大学计算机软件新技术国家重点实验室开放课题（KFKT2021B12）； |
| |
| 摘 要： | 针对基于时空Transformer模型的交通流量预测模型性能不高的问题，提出了一种基于编解码器的改进的时空Transformer模型（improved spatio-temporal Transformer model,ISTTM）。编码器对历史流量特征进行编码，解码器预测未来序列。编码器将空间稀疏自注意力和时间层次扩散卷积相结合，捕捉交通流量的动态空间相关性和局部空间特征，再利用时间自注意力建模非线性时间相关性；解码器与编码器类似地挖掘出输入序列的时空特征。基于编解码器提取的时空特征，采用双重交叉注意力模拟历史交通观测对未来预测的影响，建模每个历史时间步和每个未来时间步的直接关系以及对整个未来时间段的影响，并输出未来交通流量的最终表示。为了证实ISTTM的有效性，在METR-LA和NE-BJ两个真实世界的大规模数据集上进行实验，ISTTM结果优于6个先进的基线。
|
| 关 键 词： | 交通流量预测 时空特征 稀疏自注意力 扩散卷积 |
|
