发明 一种基于多头注意力机制的交通流预测方法 特价15
智能交通管理 路网管控 人工智能 深度学习 机器学习 训练模型 卷积神经网络 1人
G08G1/01 G08G1/052 G06Q10/04 G06Q50/26 G06N3/0442 G06N3/042 G06N3/0464 G06N3/048
摘要：本发明涉及一种基于多头注意力机制的交通流预测方法，包括获取目标道路所在目标区域的路网地理图及其历史交通流数据；构建地理邻域子图和语义相似性子图；利用多头注意力机制分别计算地理邻域子图和语义相似性子图中道路节点与其相邻节点之间的权重，更新地理邻域子图和语义相似性子图的邻接矩阵；利用参数融合法将更新后的地理邻域子图和语义相似性子图的邻接矩阵进行融合得到融合矩阵，根据融合矩阵利用GCN提取目标道路节点的时空特征序列；将目标道路节点的时空特征序列和目标道路节点的交通流数据输入双向GRU模型，根据双向GRU模型的正向输出和反向输出通过全连接层得到目标道路节点的交通流预测结果。
