公开(公告)号：CN118762528A

公开(公告)日：2024-10-11

申请号：CN202411255469.X

申请日：2024-09-09

Applicant: 南京信息工程大学

Inventor： 余文斌 ,  奚馗文 ,  张成军

IPC: G08G1/01 ,  G08G1/0969 ,  G06N3/0442 ,  G06N3/0455 ,  G06N3/0464 ,  G06N3/048 ,  G06N3/0499 ,  G06N3/08

Abstract: 本发明公开了一种基于CATST网络的短时交通拥塞预测方法，包括以下步骤：（1）获取开源网站地图的实时路况图并进行预处理；（2）构建基于卷积注意力Transformer时空网络即CATST网络，输入经过预处理后的数据进行训练；其中，CATST网络包括：2个时空注意力模块、2个卷积注意力Transformer时空模块、2个批标准化层和3D卷积层；（3）使用均方误差和平均绝对百分比误差作为评价指标，评价CATST网络对于交通拥塞预测的准确度；本发明不仅适用于流量交通数据集的空间结构，而且还拥有根据输入数据的状态和过去状态来预测未来该点的状态的能力。