[发明专利]基于多元时序数据分析的交通序列预测方法

说明书

本发明公开了一种基于多元时序数据分析的交通序列预测方法，采用随机图扩散注意力机制提取交通序列的全局和局部空间特征，使用时间注意力提取时间特征，提升了预测精度，降低了模型的内存使用量，提升了模型在长期预测上的效果。本发明所述的基于多元时序数据分析的交通序列预测方法，使已有模型在保持长期预测的精度的前提下，解决短期预测精度不足、计算复杂度过高导致占用内存规模较大以及不够轻量化的问题。

技术领域

本发明涉及交通预测技术领域，尤其是涉及一种基于多元时序数据分析的交通序列预测方法。

背景技术

交通预测是智能交通系统(ITS)中的一项重要服务，它可以根据传感器观测到的历史交通状况(如交通流量，交通速度)预测未来的交通状况。这一功能促进了与交通管理、城市计算、公共安全等相关的广泛服务。交通预测模型通过学习过去的交通序列的特征规律，从而预测未来一段时间内的交通序列。不同于其他单变量时间序列预测问题，交通预测十分具有挑战性。作为交通领域的固有现象，当前路口的交通状况会受到附近和远处路口的交通状况的影响，这意味着它们之间存在一定的空间依赖性。同时，交通序列的走势会受到历史走势的影响，充分考虑交通序列的时间依赖性也是重中之重。交通路网的复杂的时空相关性严重限制了交通流预测的有效性。

近年来，越来越多的新方法被提出，它们通过对时空特征进行建模，很好的提升了短期交通预测的精度，在一定程度上也提升了长期交通预测的精度。虽然随着时间的流逝，众多方法在短期交通预测上展现出了优异的效果，但它们并没有很好的提升长期交通预测的效果。现有方法通常采用图卷积神经网络对局部空间信息进行建模，使用卷积神经网络(CNN)或者循环神经网络(RNN)对时间信息进行建模，这些方法都着重于对局部信息的提取而缺乏全局信息的利用。直到注意力机制(Attention，AT)被提出，深度学习上有了一个通用的可以捕获全局特征的方法。在最新的模型中，GMAN使用注意力机制构建模型，在长期交通预测中取得了最好的效果。

GMAN采用Encoder-Decoder架构，使用注意力机制分别对空间和时间两个维度提取特征，并使用转换注意力机制来缓解历史交通序列月预测交通序列之间的误差。GMAN有效提高了交通预测中长期预测的精度，但它存在一些问题：

1、空间注意力具有的高计算复杂度导致内存使用量大；

2、不具备对空间维度局部特征捕获的能力，导致短期预测的精度降低；

3、预测时间开销大，不够轻量化。

发明内容

本发明的目的是提供一种基于多元时序数据分析的交通序列预测方法，使已有模型在保持长期预测的精度的前提下，解决短期预测精度不足、计算复杂度过高导致占用内存规模较大以及不够轻量化的问题。

为实现上述目的，本发明提供了一种基于多元时序数据分析的交通序列预测方法，包括如下步骤：

步骤1：先对公共交通序列数据集进行数据预处理，获得训练数据集和验证数据集；

步骤2：借助于步骤1得到的训练数据集，每次随机选取16组训练数据，输入到随机图扩散注意力模型中，每组数据中的历史时间时间信息和未来时间时间信息和预定义图结构信息输入到图扩散注意力模型中的时空embedding生成器中生成历史时空embedding和未来时空embedding；

步骤3：将历史交通数据和历史时空embedding送入编码器，分别使用图扩散注意力模块和时间注意力模块提取空间特征和时间特征，并使用gate fusion模块融合两种特征生成历史时空特征；

步骤4：将历史时空特征、历史时空embedding和未来时空embedding送入转化注意力模块，生成未来时空特征；