一种基于多重图注意力机制的交通流预测方法、设备及介质

    公开(公告)号:CN113762338A

    公开(公告)日:2021-12-07

    申请号:CN202110868311.X

    申请日:2021-07-30

    Applicant: 湖南大学

    Abstract: 本发明公开了一种基于多重图注意力机制的交通流预测方法、设备及介质,方法为:获取训练样本;构建由若干交通流预测子模型构成的交通流预测生成组合模型GC‑GE和由编码器、注意力机制及全连接层构成的交通流预测优化模型GA‑OP;GC‑GE使用每个预测子模型进行初始预测;GA‑OP通过对历史交通流、初始预测交通流、交通流外部因素进行编码,利用注意力机制获取三者间的两两交互关系,并将这些张量拼接到全连接层上进行评分以对整个模型进行训练;最后根据评分确定GC‑GE中各子模型的权重,以此获得最终的预测交通流。本发明不仅考虑了交通流的时空相关性,还考虑交通流外部因素的影响,保证了预测结果的有效性。

Patent Agency Ranking