一种基于时空图卷积网络的公共交通流量预测方法技术

本发明专利技术公开了一种基于时空图卷积网络的公共交通流量预测方法，包括以下步骤：步骤(1)、获取公共交通流量数据并进行预处理；步骤(2)、公共交通网络建模；步骤(3)、构建时间序列特征；步骤(4)、构建时空图卷积网络；步骤(5)、训练时空图卷积网络；步骤(6)、时空图卷积网络的测试过程；针对复杂的公共交通流量预测问题，该方法能够高效准确地预测未来一段时间内乘坐公共交通工具出行的乘客流量，便于规划线路，错开高峰时段，提高乘坐体验，提升百姓生活质量。质量。质量。

【技术实现步骤摘要】
一种基于时空图卷积网络的公共交通流量预测方法


[0001]本专利技术涉及交通流量预测领域，具体涉及时空图卷积网络用于回归问题及其实际应用于公共交通流量预测领域。

技术介绍

[0002]近年来，随着城市化的进一步发展，智能交通系统对城市管理起到了重要作用，而公共交通流量预测是智能交通系统中的关键一环。城市公共交通运行速度和状态会直接影响乘客的乘坐体验和交通运输容量。若能够对公共交通运输状态进行准确预测，则有助于规划交通路线和动态管理，方便百姓出行，提高生活质量。
[0003]实际公共交通流量通常具有非常高的非线性和复杂性，单纯使用已有的时间序列算法只能够简单预测车站某时段内乘客流量，无法考虑车站间的空间关系。若将车站网络以图的形式建模，然后采用基于图的深度学习方法预测交通流量则只考虑了空间关系，忽略了动态时空相关性。
[0004]通过整合以时间序列形式展现的交通流量数据以及以图的形式建模的车站网络数据，即结合循环神经网络和图卷积神经网络——时空图卷积网络，统筹交通流量的动态时空相关性，可以较好的预测某时段内经过某站点的交通流量。但目前的时空图卷积方法都是先将站点网络建模为普通无向图，边的权重由节点间的物理距离决定，这不符合实际车站分布情况，因为现实生活中有大量的站点相距不远却难以直接到达。因此，我们需要改变建模思路，寻求更合适的建模方式。

技术实现思路

[0005]针对上述情况，本专利技术提供一种基于时空图卷积网络的公共交通流量预测方法。提出的方法包括数据预处理，更合适的站点网络建模方式以及整合了循环神经网络的和图卷积神经网络的时空图卷积网络。将站点作为图中的节点，边的权重不再取决于节点间的距离而是一段时间内的车站的交通流量。考虑到不同的站点具有不同的实际意义，为每个站点都随机初始化了节点特征，并在训练过程中加以学习。对每一个时间段内的交通流量都以上述形式建模为图得到时序公共交通图后将其输入时空图卷积神经网络，便可预测未来一段时间内的公共交通流量。
[0006]一种基于时空图卷积网络的公共交通流量预测方法，步骤如下：
[0007]步骤(1)、获取公共交通流量数据并进行预处理。
[0008]步骤(2)、公共交通网络建模。
[0009]步骤(3)、构建时间序列特征；
[0010]步骤(4)、构建时空图卷积网络；
[0011]将时间序列特征A和节点特征X作为时空图卷积网络的输入。时空图卷积由图卷积网络和门控循环单元构成。
[0012]步骤(5)、训练时空图卷积网络；
[0013]步骤(6)、时空图卷积网络的测试过程；
[0014]训练完成后，将测试集输入在步骤(5)中保存下来的时空图卷积网络中即可得到测试结果，用于预测未来某一时间段交通流量。
[0015]步骤(1)具体方法如下：
[0016]收集三个月内某一地区地铁的乘客流量数据。以30分钟为时间窗口，统计每个时间窗口内的乘客数量以及他们的出发地和目的地。将所有数据按照6：2：2的比例划分为训练集、验证集和测试集。
[0017]步骤(2)具体方法如下：
[0018]将地铁站点作为顶点即节点，建立邻接矩阵其中N为节点数，即地铁站点数量。A
i,j
表示一个时间窗口内由站点i去往站点j的乘客数量，得到有向有权无环图。将该图命名为流量共现图，其邻接矩阵为流量共现矩阵。由于每个所处地理位置不同，其周围地域功能不同，故随机初始化可学习参数作为节点特征，其中D为特征维度。
[0019]步骤(3)具体方法如下：
[0020]考虑到乘客数量变化的周期性和实时性，选择邻接、日、周乘客流量数据作为特征。邻接特征属于实时特征，由被预测时间窗口邻接的T
n
个时间窗口的流量共现矩阵拼接后得到日特征属于周期特征，由前T
d
天与被预测时间窗口处于相同时段的流量共现矩阵拼接后得到周特征同样属于周期特征，由前T
w
周与被预测时间窗口处于相同时段的流量共现矩阵拼接后得到将三种特征拼接便得到最终用于训练的时间序列特征其中||表示拼接，T＝T
w
+T
d
+T
n
。被预测窗口的流量共现矩阵为实际值Y
t
。
[0021]步骤(4)具体方法如下：
[0022]将时间序列特征A和节点特征X作为时空图卷积网络的输入。
[0023]时空图卷积由图卷积网络和门控循环单元构成。
[0024][0025]z
t
＝σ(GCN(A
:,t
,X；W
z
)+h
t
‑1U
z
)
ꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀ
(2)
[0026]r
t
＝σ(GCN(A
:,t
XW
r
)+h
t
‑1U
r
)
ꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀ
(3)
[0027][0028][0029]其中为时间序列特征A中的第t个时间窗口的特征，是时间步t时的输入；是A
:,t
的度矩阵；h
t
是时间步t时的输出；z
t
和r
t
分别是重置门和更新门；X,W
z
,U
z
,W
r
,U
r
为可学习参数；σ是激活函数，
⊙
为哈达玛积。被预测时间段的流量共现矩阵由最后一个时间步的输出h
‑1经过全连接层得到：
[0030]Y
p
＝σ(h
‑1W+b)
ꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀ
(6)
[0031]其中W和b为可学习参数。
[0032]步骤(5)具体方法如下：
[0033]对时空图卷积网络进行训练。按照步骤(1)所述的比例选取训练集和验证集。采用Adam优化器，MSE损失函数。在验证过程中将预测得到的流量共现矩阵Y
p
和实际流量共现矩阵Y
p
展平为一维数据后使用MSE、RMSE和MAPE进行评价，将性能最优的网络参数保存下来。
[0034]本专利技术有益成果如下：
[0035]针对复杂的公共交通流量预测问题，该方法能够高效准确地预测未来一段时间内乘坐公共交通工具出行的乘客流量，便于规划线路，错开高峰时段，提高乘坐体验，提升百姓生活质量。
附图说明
[0036]图1为本专利技术中的流量共现图及流量共现矩阵示意图；
[0037]图2为本专利技术中的时空图卷积示意图。
具体实施方式
[0038]以下结合附图及实施例，对本专利技术进一步详细说明。
[0039]一种基于时空图卷积网络的公共交通流量预测方法，步骤如下：
[0040]步骤(1)、获取公共交本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于时空图卷积网络的公共交通流量预测方法，其特征在于，步骤如下：步骤(1)、获取公共交通流量数据并进行预处理；步骤(2)、公共交通网络建模；步骤(3)、构建时间序列特征；步骤(4)、构建时空图卷积网络；将时间序列特征A和节点特征X作为时空图卷积网络的输入；时空图卷积由图卷积网络和门控循环单元构成；步骤(5)、训练时空图卷积网络；步骤(6)、时空图卷积网络的测试过程；训练完成后，将测试集输入在步骤(5)中保存下来的时空图卷积网络中即可得到测试结果，用于预测未来某一时间段交通流量。2.根据权利要求1所述的一种基于时空图卷积网络的公共交通流量预测方法，其特征在于，步骤(1)具体方法如下：收集三个月内某一地区地铁的乘客流量数据；以30分钟为时间窗口，统计每个时间窗口内的乘客数量以及他们的出发地和目的地；将所有数据按照6：2：2的比例划分为训练集、验证集和测试集。3.根据权利要求2所述的一种基于时空图卷积网络的公共交通流量预测方法，其特征在于，步骤(2)具体方法如下：将地铁站点作为顶点即节点，建立邻接矩阵其中N为节点数，即地铁站点数量；A
i,j
表示一个时间窗口内由站点i去往站点j的乘客数量，得到有向有权无环图；将该图命名为流量共现图，其邻接矩阵为流量共现矩阵；由于每个所处地理位置不同，其周围地域功能不同，故随机初始化可学习参数作为节点特征，其中D为特征维度。4.根据权利要求3所述的一种基于时空图卷积网络的公共交通流量预测方法，其特征在于，步骤(3)具体方法如下：考虑到乘客数量变化的周期性和实时性，选择邻接、日、周乘客流量数据作为特征；邻接特征属于实时特征，由被预测时间窗口邻接的T
n
个时间窗口的流量共现矩阵拼接后得到日特征属于周期特征，由前T
d
天与被预测时间窗口处于相同时段的流量共现矩阵拼接后得到周特征同样属于周期特征，由前T
w
周与被预测时间窗口处于相同时段的流量共现矩阵拼接后得到将三种特征拼接便得到最终用于训练的时间序列特征其中||表示拼接，T＝T
w
+T
d
+T
n
；被预测窗口的流量共...

【专利技术属性】
技术研发人员：颜成钢，阮定，孙垚棋，张继勇，李宗鹏，
申请(专利权)人：杭州电子科技大学，
类型：发明
国别省市：