T-GCN: A Temporal Graph Convolutional Network for Traffic Prediction Lin

用于流量预测的时序图卷积

问题背景:交通流量预测。

提出模型:为了同时捕获空间和时间相关性,提出了时间图卷积网络(T-GCN)模型,即将图卷积网络(GCN)和GRU结合。GCN用于学习复杂的拓扑结构以捕获空间依赖性,而GRU用于学习交通数据的动态变化以捕获时间依赖性。T-GCN的代码:GitHub - lehaifeng/T-GCN: Temporal Graph Convolutional Network for Urban Traffic Flow Prediction Method


现有的流量预测方法:自回归综合移动平均(ARIMA)模型,SVM 和部分神经网络,考虑了交通的动态变化而忽略了空间依赖性。为了更好地刻画空间特征,引入CNN进行空间建模。然而,CNN通常用于欧氏数据,如图像,规则网格等。CNN无法描述具有复杂拓扑结构的城市道路网络的空间依赖性。

本文贡献:

(1)T-GCN集成GCN和GRU。GCN用于捕获道路网络的拓扑结构,对空间依赖性进行建模。 GRU用于捕获道路上交通数据的动态变化,对时间依赖性进行建模。 

(2)T-GCN模型不仅可以短期预测,还可以用于长期的交通预测任务 。


方法

整体框架

定义1:G=(V,E). V-N个road节点,E-边,邻接矩阵表示road节点的连接,元素为0/1。

定义2:特征矩阵,P表示node属性特征的数量(历史时间序列的长度)。

通过学得一个映射函数f,根据图G和特征矩阵X,计算接下来T个时间步的预测值:

,n为历史时序长度。


空间依赖建模:

给定邻接矩阵A和特征矩阵X,GCN会在傅立叶域中构造一个滤波器。 过滤器作用在图的节点上,通过其一阶邻域捕获节点之间的空间特征,然后通过堆叠多个卷积层构建GCN,其表示:

为第l层输出,为第l层参数,.

本论文用两层GCN获取空间依赖信息:

表示前向传播,是输入层→输出层的权重矩阵,H为隐藏单元个数,P为特征矩阵长度。是隐藏层→输出层的权重矩阵,,T为预测长度。

获取空间特征示意图

时间依赖建模:

选用GRU的原因:LSTM训练时间较长,GRU结构相对简单,参数较少,训练能力较快。


左侧是时空预测的过程,右侧是T-GCN单元的具体结构

损失函数:

.为L2范数,防止过拟合。


实验设置

数据集:1.深圳出租车。156条道路(节点),每15min汇总流量速度。邻接矩阵A[156,156], 特征矩阵 X[156,15]。2.Los-loop。A[207,207],X [207,5]。

数据集划分:80%-训练,20%-测试。

预测:15,30,45,60min后的流量速度。

评价指标:RMSE,MAE,accuracy,R^2, var

j-时间样本,i-road,M-时间样本数,N-road总数。

RMSE和MAE来测量预测误差:值越小,预测效果越好。 精度用于检测预测精度:值越大,预测效果越好。 R2和var计算相关系数,该系数表示预测实际数据的能力:值越大,预测效果越好。

超参数:lr=0.001,batchsize=32,epoch=5000. hidden units={8,16,32,64,100,128}. 优化器Adam.

深圳出租车。不同隐藏单元大小的预测性能。a,b训练集,c,d测试集。总的来看,hidden units=100最好。
Los-loop. hidden units=64.

Baseline: ARIMA, SVR, GCN,GRU.

实验结果

1.5个度量标准结果比较

2.时空预测能力--与GCN, GRU比较

3.长期预测能力--不同预测长度的比较

4.扰动分析和鲁棒性

添加两种噪声-高斯分布 ,σ ∈ (0.2,0.4,0.8,1,2) 和泊松分布 λ ∈ (1, 2, 4, 8, 16),看出T-GCN的度量并没有很大变化,所以T-GCN具有很高的鲁棒性。

5.可解释性

(1)T-GCN模型在局部最小值/最大值处预测不佳。推测:GCN在傅立叶域中定义平滑滤波器,并通过不断移动滤波器来捕获空间特征,这个导致峰变得更平滑。

(2)实际交通信息与预测结果之间存在一定的误差,这是由“零出租车值”引起的。出租车零值是将真实值不为零的交通特征矩阵设置为零的现象。

(3)T-GCN模型可以捕获时空特征并学习变化趋势,可以检测高峰时间的开始和结束,并以与实际交通速度相似的模式得出预测结果。

你可能感兴趣的:(T-GCN: A Temporal Graph Convolutional Network for Traffic Prediction Lin)