公开(公告)号：CN117576910A

公开(公告)日：2024-02-20

申请号：CN202311591453.1

申请日：2023-11-27

Applicant: 中国矿业大学

Inventor： 徐潇 ,  张磊 ,  刘佰龙 ,  梁志贞 ,  江海峰 ,  祝传琦 ,  张增殖

IPC: G08G1/01 ,  G06F18/2131 ,  G06F18/25 ,  G06N3/0464 ,  G06N3/0455 ,  G06N3/044 ,  G06N3/0985 ,  G06F123/02

Abstract: 本发明公开了一种基于循环时空注意力机制的交通流量预测方法，属于交通流量预测领域。预测方法包括：首先，将来自传感器的原始数据经数据预处理模块得到交通流量数据集；其次，将划分后的数据集输入到基于循环时空注意力机制的时空特征提取模块，融合学习到的时空特征；最后，将得到的时空特征输入到预测模块并预测未来一段时间的交通流量数据。在数据预处理阶段的数据填充过程中，根据交通流量数据的特点设计了前向填充法，大幅降低了计算开销。本方法设计了带有循环特性的滑动自注意力位置编码，解决了交通流量预测在长依赖上表现不佳的问题，并通过并行计算架构骨架支持多个GPU上的分布式训练，提供高于单个GPU的计算并行度，提高运算效率。