交通流量预测:T-GCN A Temporal Graph Convolutional Network for Traffic Prediction

发布时间:2024年01月17日

摘要

为了同时捕捉时空相关性,将图卷积网络(GCN)和门控递归单元(GRU)相结合,提出了一种新的基于神经网络的流量预测方法–时态图卷积网络(T-GCN)模型。具体地,GCN用于学习复杂的拓扑结构以捕获空间相关性,而门控递归单元用于学习交通数据的动态变化以捕获时间相关性。

Ⅱ. 方法

A. 问题定义
定义一:

G = (V, E) 邻接矩阵A(1表示两个路有连接,反之没有),把路当作结点,N是结点的数量

定义二:

X N × P X^{N×P} XN×P特征矩阵,P是结点特征的数量,也就是历史时间序列的长度

X t ∈ R N × i X_t∈R^{N×i} Xt?RN×i 用来表示每条道路在时间 i 时的速度

结点属性可以是速度,流量和密度

image-20231210112451507

B. 回顾总结

我们首先使用历史n个时间序列数据作为输入,并使用图形卷积网络来捕获城市道路网的拓扑结构,从而获得空间特征。

其次,将得到的具有空间特征的时间序列输入到门控递归单元模型中,通过单元间的信息传输获得动态变化,以获取时间特征。

最后,我们通过完全连接层得到结果。

C. 方法
空间相关性:

GCN:

image-20231210114112309

两层GCN:

image-20231210114440075

时间相关性:

GRU:

image-20231210115636521

image-20231210115710123

时间图卷积网络:

image-20231210115840662

GC表示图卷积过程,equation 3

image-20231210120303705

损失函数:

image-20231210120508971

The second term Lreg is the L2 regularization term that helps to avoid an overfitting problem

λ is a hyperparameter

文章来源:https://blog.csdn.net/weixin_46285416/article/details/135637298
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。