公开(公告)号：CN117037483A

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

申请号：CN202310995264.4

申请日：2023-08-09

Applicant: 重庆邮电大学

Inventor： 贾朝龙 ,  黄冰宇 ,  王蓉 ,  李暾 ,  庞育才 ,  段思睿 ,  肖云鹏

IPC: G08G1/01 ,  G08G1/052 ,  G06Q10/04 ,  G06Q50/26 ,  G06N3/0442 ,  G06N3/042 ,  G06N3/0464 ,  G06N3/048

Abstract: 本发明涉及一种基于多头注意力机制的交通流预测方法，包括获取目标道路所在目标区域的路网地理图及其历史交通流数据；构建地理邻域子图和语义相似性子图；利用多头注意力机制分别计算地理邻域子图和语义相似性子图中道路节点与其相邻节点之间的权重，更新地理邻域子图和语义相似性子图的邻接矩阵；利用参数融合法将更新后的地理邻域子图和语义相似性子图的邻接矩阵进行融合得到融合矩阵，根据融合矩阵利用GCN提取目标道路节点的时空特征序列；将目标道路节点的时空特征序列和目标道路节点的交通流数据输入双向GRU模型，根据双向GRU模型的正向输出和反向输出通过全连接层得到目标道路节点的交通流预测结果。