基于图Transformer网络的城市路网短时交通流预测模型

周烽 ,  王世璞 ,  张坤鹏

科学技术与工程 ›› 2024, Vol. 24 ›› Issue (10) :4307-4316

PDF (2785KB)
科学技术与工程 ›› 2024, Vol. 24 ›› Issue (10) :4307-4316 DOI: 10.12404/j.issn.1671-1815.2303913
交通运输

基于图Transformer网络的城市路网短时交通流预测模型

作者信息 +

A Short-term Traffic Flow Prediction Model for Urban Road Network Based on Graph Transformer Network

Author information +
文章历史 +

摘要

针对城市路网短时交通流预测问题,在考虑路网交通状态时空相关性基础上,提出了一种基于图Transformer(graph transformer, Graformer)的预测方法。该方法将多条路段的交通状态预测问题转化为图节点状态预测问题,针对区分相同结构的空间路网结构图,将带有边的图同构网络(graph isomorphism network with edges, GINE)和Transformer网络相结合,对交通状态在路网层面的时空相关性进行建模,从而实现城市路网短时交通流预测。具体来说,Graformer模型首先利用长短期记忆网络(long short-term memory, LSTM)对交通数据的时序信息进行预处理,接着采用基于GINE与Transformer的全局注意力机制提取交通数据的空间特征,最后实现路网各路段交通流的同步预测。

Abstract

A prediction method based on graph transformer (Graformer) was proposed for short-term traffic flow prediction in urban road networks, considering the spatiotemporal correlation of road network traffic status. This method transformed the traffic state prediction problem of multiple road sections into a graph node state prediction problem. For distinguishing spatial road network structures with the same structure, a graph isomorphism network with edges (GINE) and a Transformer network to model the spatiotemporal correlation of traffic states at the road network level were combined, thereby achieving short-term traffic flow prediction for urban road networks. Specifically, the Graformer model first used the long short term memory (LSTM) network to preprocess the temporal information of the traffic data, then used the global attention mechanism based on GINE and Transformer to extract the spatial characteristics of the traffic data, and finally realized the synchronous prediction of the traffic flow of each road section of the road network. Through experimental verification using the PeMS dataset, the results show that the proposed Graformer model outperformed the comparative model in various performance indicators, demonstrating its effectiveness as a reliable and efficient short-term traffic flow prediction method for road networks.

引用本文

引用格式 ▾
周烽, 王世璞, 张坤鹏. 基于图Transformer网络的城市路网短时交通流预测模型[J]. 科学技术与工程, 2024, 24(10): 4307-4316 DOI:10.12404/j.issn.1671-1815.2303913
ZHOU Feng, WANG Shi-pu, ZHANG Kun-peng. A Short-term Traffic Flow Prediction Model for Urban Road Network Based on Graph Transformer Network[J]. Science Technology and Engineering, 2024, 24(10): 4307-4316 DOI:10.12404/j.issn.1671-1815.2303913

登录浏览全文

4963

注册一个新账户 忘记密码

参考文献

基金资助

国家自然科学基金(62002101)

郑州市科技局自然科学项目(22ZZRDZX05)

PDF (2785KB)

32

访问

0

被引

详细

导航
相关文章

/

〈 〉