基于改进GCN-sbuLSTM模型的高速公路交通量预测方法

李嘉, 文婧, 周正, 苏骁, 杜朝阳, 杨婉澜

交通运输研究 ›› 2024, Vol. 10 ›› Issue (3) : 56-65.

交通运输研究 ›› 2024, Vol. 10 ›› Issue (3) : 56-65. DOI: 10.16503/j.cnki.2095-9931.2024.03.007
技术前沿

基于改进GCN-sbuLSTM模型的高速公路交通量预测方法

作者信息 +

Expressway Traffic Volume Prediction Method Based on Improved GCN-sbuLSTM Model

  • LI Jia 1 ,  
  • WEN Jing 1 ,  
  • ZHOU Zheng 1, * ,  
  • SU Xiao 2 ,  
  • DU Chaoyang 2 ,  
  • YANG Wanlan 1
Author information +
文章历史 +

摘要

为解决现有高速公路交通量预测方法在捕捉动态时空依赖关系方面的不足,提出了一种融合信息几何方法与注意力机制的新型高速路网交通量预测模型。首先,利用信息几何方法量化ETC门架之间的动态数据分布差异。然后,利用注意力机制来捕获交通的动态空间依赖关系。最后,结合一种堆叠的双向递归层结构,提出了一种长时间跨度的并行子模型算法,即基于信息几何方法(Information Geometry)和注意力机制(Attention Mechanism)优化的图卷积神经网络(GCN)结合堆叠双向单向长短期记忆神经网络(sbuLSTM)的组合模型—IGAGCN-sbuLSTM。采用该模型对100多条路段、3 000多处门架近7亿条高速公路ETC门架系统数据进行分析,结果显示:与LSTM、GCN、GCN-LSTM、ASTGCN等现有4种模型相比,在10 min时间尺度下,IGAGCN-sbuLSTM组合模型的平均绝对误差(MAE)分别降低了2.39, 3.72, 1.02, 1.46,均方根误差(RMSE)分别降低了3.25, 4.32, 2.05, 5.65,平均绝对百分比误差(MAPE)分别降低了5.49%, 12.54%, 1.56%, 0.5%。研究表明,IGAGCN-sbuLSTM模型在预测精度和不同时间间隔的预测性能上均优于现有的单一捕获特性模型及其他常用的组合模型,可广泛应用于高速公路收费、车速等数据的预测分析。

Abstract

In order to address the shortcomings of existing expressway traffic volume prediction methods in capturing dynamic spatio-temporal dependencies, the paper proposed a novel traffic volume prediction model for expressway networks that incorporates information geometry methods and attention mechanisms. Initially, an information geometry method was used to quantify the difference in dynamic data distribution between ETC gantries. Subsequently, an attention-based mechanism was devised to encapsulate the dynamic spatial dependencies inherent in traffic patterns. In conclusion, by amalgamating a configuration of stacked bidirectional recurrent layers, a long-time span parallel sub-model algorithm named IGAGCN-sbuLSTM was proposed. This acronym denoted a synergistic model that combined GCN(Graph Convolutional Neural Network), which was optimized using Information Geometry Approach and Attention Mechanism, with sbuLSTM(Stacked Bidirectional Unidirectional Long Short-Term Memory Neural Network). Leveraging over 100 road segments, surpassing 3 000 gantries from nearly 700 billion entries in the expressway ETC gantry system database, experimental outcomes demonstrated that the IGAGCN-sbuLSTM algorithm, when benchmarked against existing models such as LSTM, GCN, GCN-LSTM, ASTGCN, registered a reduction in MAE (Mean Absolute Error) by 2.39, 3.72, 1.02, and 1.46 respectively on a 10-minute time scale. The RMSE (Root Mean Square Error) diminished by 3.25, 4.32, 2.05, and 5.65, while the MAPE (Mean Absolute Percentage Error) decreased by 5.49%, 12.54%, 1.56%, and 0.5% respectively. These results affirm that the IGAGCN-sbuLSTM model surpasses its predecessors with single capture characteristic and other prevalent combined models in terms of predictive accuracy and capability to handle expanding time intervals, rendering it highly applicable for predictions analysis in areas such as expressway tolling and vehicular speeds.

关键词

高速公路 / 交通量预测 / ETC门架系统 / 信息几何方法 / 注意力机制 / 堆叠双向单向长短期记忆神经网络 / 图卷积神经网络

Key words

expressway / traffic volume prediction / ETC(Electronic Toll Collection) gantry system / information geometry method / attention mechanism / sbu-LSTM(Stacked Bidirectional Unidirectional Long Short-Term Memory Neural Network) / GCN(Graph Convolutional Network)

引用本文

导出引用
李嘉, 文婧, 周正, . 基于改进GCN-sbuLSTM模型的高速公路交通量预测方法[J]. 交通运输研究. 2024, 10(3): 56-65 https://doi.org/10.16503/j.cnki.2095-9931.2024.03.007
LI Jia, WEN Jing, ZHOU Zheng, et al. Expressway Traffic Volume Prediction Method Based on Improved GCN-sbuLSTM Model[J]. Transport Research. 2024, 10(3): 56-65 https://doi.org/10.16503/j.cnki.2095-9931.2024.03.007
中图分类号: TP183 (人工神经网络与计算)    U491    U495   

参考文献

[1]
ANSARI M A, ARUNDHATHI T. Street traffic forecasting ongoing advances and new challenges[J]. International Journal of Computer Sciences and Engineering, 2019, 7(3): 650-656.
[2]
BELLETTI F, HAZIZA D, GOMES G, et al. Expert level control of ramp metering based on multi-task deep reinforcement learning[J]. IEEE Trans on Intelligent Transportation Systems, 2017, 19(4): 1198-1207.
[3]
刘群, 杨濯丞, 蔡蕾. 基于ETC门架数据的高速公路短时交通流预测[J]. 公路交通科技, 2022, 39(4):123-130.
[4]
SIAMI-NAMINI S, TAVAKOLI N, NAMIN A S. The performance of lstm and bilstm in forecasting time series[C]//2019 IEEE International Conference on Big Data(Big Data). Los Angeles: IEEE, 2019: 3285-3292.
[5]
VIDYA G S, HARI V S, SHIVASAGARAN S. Intelligent traffic prediction using LSTM network[C]// International Conference on Automation, Signal Processing, Instrumentation and Control. Singapore: Springer Nature Singapore, 2020: 1661-1669.
[6]
BRUNA J, ZAREMBA W, SZLAM A, et al. Spectral networks and locally connected networks on graphs[J]. International Conference on Learning Representations. Scottsdale, USA: ICLR, 2013. DOI: 10.48550/arXiv.1312.6203.
[7]
YU B, YIN H, ZHU Z. Spatio-Temporal graph convolutional networks: A deep learning framework for traffic forecasting[J]. Proceedings of the 27th International Joint Conference on Artificial Intelligence. Stockholm, Sweden: IJCAI, 2018: 3634-3640.
[8]
SONG C, LIN Y, GUO S, et al. Spatial-Temporal synchronous graph convolutional networks: A new framework for spatial-temporal network data forecasting[C]// Proceedings of the AAAI Conference on Artificial Intelligence. New York: AAAI, 2020, 34(1): 914-921.
[9]
PHAN H, OLIVER Y, CHÉN K P, et al. Unifying isolated and overlapping audio event detection with multi-label multi-task convolutional recurrent neural networks[C]// 2019 IEEE International Conference on Acoustics, Speech and Signal Processing(ICASSP 2019). Brighton, UK: IEEE, 2019: 51-55.
[10]
AMARI S I, NAGAOKA H. Methods of information geometry[M]. Oxford: Oxford University Press, 2000.
[11]
邵海琴, 王兆峰. 中国交通碳排放效率的空间关联网络结构及其影响因素[J]. 中国人口·资源与环境, 2021, 31(4): 32-41.
[12]
DISSANAYAKE B, HEMACHANDRA O, LAKS-HITHA N, et al. A comparison of arimax, var and lstm on multivariate short-term traffic volume forecasting[C]// Proceedings of 28th Conference of Open Innovations Association. Moscow: FRUCT, 2021: 564-570.
[13]
YANG H, LU X, CHERRY C, et al. Spatial variations in active mode trip volume at intersections: a local analysis utilizing geographically weighted regression[J]. Journal of Transport Geography, 2017, 64: 184-194.
[14]
CAI L, JANOWICZ K, MAI G, et al. Traffic transformer: capturing the continuity and periodicity of time series for traffic forecasting[J]. Transactions in GIS, 2020, 24(3): 736-755.
[15]
YANG H, LU X, CHERRY C, et al. Spatial variations in active mode trip volume at intersections: A local analysis utilizing geographically weighted regression[J]. Journal of Transport Geography, 2017, 64:184-194.
[16]
ZHANG J, ZHENG Y, QI D. Deep spatio-temporal residual networks for citywide crowd flows prediction[C]// Proceedings of the AAAI Conference on Artificial Intelligence. San Francisco: AAAI, 2017, 31(1). DOI: 10.48550/arXiv.1610.00081.
[17]
ZHAO J, LIU Z, SUN Q, et al. Attention-Based dynamic spatial-temporal graph convolutional networks for traffic speed forecasting[J]. Expert Systems with Applications, 2022, 204: 117511.
[18]
王健宗, 孔令炜, 黄章成, 等. 图神经网络综述[J]. 计算机工程, 2021, 47(4):1-12.
[19]
王竟成, 张勇, 胡永利, 等. 基于图卷积网络的交通预测综述[J]. 北京工业大学学报, 2021, 47(8):954-970.
[20]
ABDULJABBAR R L, DIA H, TSAI P W. Unidirectional and bidirectional LSTM models for short-term traffic prediction[J]. Journal of Advanced Transportation, 2021, 2021(1): 5589075.1-5589075.16.
[21]
WENINGER F, JÜRGEN GEIGER, WLLMER M, et al. The Munich 2011 CHiME challenge contribution: NMF-BLSTM speech enhancement and recognition for reverberated multisource environments[C]// Proceedings of Machine Listening in Multisource Environments(CHiME 2011), satellite workshop of Interspeech 2011. Florence, Italy: ISCA, 2011: 24-29.
[22]
PARASCANDOLO G, HUTTUNEN H, VIRTA-NEN T. Recurrent neural networks for polyphonic sound event detection in real life recordings[C]// 2016 IEEE International Conference on Acoustics, Speech and Signal Processing(ICASSP). Shanghai: IEEE, 2016: 6440-6444.

基金

科技部科技创新2030—“新一代人工智能”重大项目(2022ZD0115600)
四川省科技计划项目(2023YFG0312)

Accesses

Citation

Detail

段落导航
相关文章

/