计算机与数字工程2025，Vol.53Issue(5)：1242-1250,9.DOI:10.3969/j.issn.1672-9722.2025.05.006
基于门控机制时空图卷积模型的交通预测
Traffic Prediction Based on Spatio-temporal Graph Convolution Model of Gating Mechanism
张锦 1王卉 2邝利丹 3刘园园 2李强 2孙程2
作者信息
- 1. 湖南师范大学 长沙 410081||长沙理工大学 长沙 410114
- 2. 湖南师范大学 长沙 410081
- 3. 长沙理工大学 长沙 410114
- 折叠
摘要
Abstract
The traditional traffic prediction method ignores the spatio-temporal dependence of traffic data and shows poor per-formance.The majority of the existing traffic prediction models are based on GNN and RNN.Although these models improve the pre-diction performance to a large extent,they still have some limitations,such as the gradient explosion problem caused by RNN when dealing with long time sequences.In this paper,a gating mechanism is proposed based on spatio-temporal graph convolution net-work(GSTGCN)for traffic prediction.The core of GSTGCN is the spatio-temporal convolution module(ST-Conv module),which mainly consists of three parts,which are standard gated temporal convolution layer,spatial graph convolution layer and gated diffu-sion causal temporal convolution layer.In addition,this paper also introduces the mask matrix to construct the adjacency matrix of the traffic graph and skip connection and residual connection are added to gated diffusion causal temporal convolutional layer.Final-ly,experiments are carried out on PeMSD7(M)and METR-LA datasets,and the experimental results show that the prediction per-formance of GSTGCN model is better than that of advanced baselines in medium-term and long-term prediction.关键词
邻接矩阵/门控机制/门控扩散因果时间卷积/跳步连接/残差连接Key words
mask matrix/gating mechanism/gated diffusion causal temporal convolution/skip connection/residual connec-tion分类
数理科学引用本文复制引用
张锦,王卉,邝利丹,刘园园,李强,孙程..基于门控机制时空图卷积模型的交通预测[J].计算机与数字工程,2025,53(5):1242-1250,9.
