返回
Contrastive optimized graph convolution network for traffic forecasting
DOI:10.1016/j.neucom.2024.128249.png)
摘要
En 中文
Traffic forecasting is an increasingly important research topic in the field of Intelligent Transportation Systems (ITS). In this field, prediction models based on Graph Convolution Networks (GCN) have become very popular. Most GCN-based models focus on constructing various optimized or dynamic road network graphs to represent the spatio-temporal correlation hidden in traffic data. However, these methods currently only consider the construction of a single improved road network graph and ignore the relationship of these existing optimized road network graphs. Therefore, in this paper, we propose a Contrastive Optimized Graph Convolution Network (COGCN) to connect two kinds of optimized road network graphs and maintain their global-local feature consistency through contrastive learning. The proposed COGCN model is evaluated in detail using four real traffic datasets: two traffic speed datasets and two traffic flow datasets. Experimental results show that COGCN improves forecasting accuracy by at least 2% on the two speed datasets and 9% on the two flow datasets compared to the existing state-of-the-art GCN-based methods.
Keyword:
Graph convolution network
Contrastive learning
期刊
IF:
6.5
论文数:
2.5W
被引数:
6.5W
机构
引用论文
Dynamic Graph Convolution Network for Traffic Forecasting Based on Latent Network of Laplace Matrix Estimation基于Laplace矩阵估计的潜在网络的动态图卷积网络交通预测
Localized Extended Kalman Filter for Scalable Real-Time Traffic State Estimation可扩展实时交通状态估计的局部扩展卡尔曼滤波器

