交通运输工程

基于注意力机制的CNN-GRU船舶交通流预测模型

展开
  • (大连海事大学 航海学院,辽宁 大连 116026)
吴莹莹(1997 — ),女,硕士生;赵丽宁(1968 — ),女,博士,教授,E-mail:zhaolining@dlmu.edu.cn

网络出版日期: 2023-01-20

基金资助

中央高校基本科研业务费专项资金资助项目(3132022635)

CNN-GRU ship traffic flow prediction model based on attention mechanism

Expand
  • (Navigation College, Dalian Maritime University, Dalian 116026,China)

Online published: 2023-01-20

摘要

为更准确地预测内河船舶交通流,提出基于注意力机制的CNN-GRU船舶交通流预测模型。该模型主要借助一维卷积单元提取数据的高维特征,GRU单元学习数据中的时序特征,并通过引入注意力机制加强重要特征的学习,实现对超长序列的学习。此外,通过分析内河上下游航道交通流间的关联性,提取长江中下游6个航段的船舶AIS数据,构造多航段船舶交通流序列数据集,并将其输入本文模型中进行训练及测试。结果表明:相比序列预测模型中的SAE、LSTM、GRU、CNN+GRU和GRU+Attention,本文模型在针对不同交通流参数的预测中均具有更高的预测精度,交通流量、交通流密度和交通流速度的预测精度分别达到95.42%、97.33%、94.99%,可更好地满足工程应用需求。

本文引用格式

吴莹莹, 赵丽宁, 袁志鑫, 张灿 . 基于注意力机制的CNN-GRU船舶交通流预测模型[J]. 大连海事大学学报, 2023 , 49(1) : 75 -84 . DOI: 10.16411/j.cnki.issn1006-7736.2023.01.008

Abstract

In order to predict inland waterway ship traffic flow more accurately, a CNN-GRU ship traffic flow prediction model was proposed based on attention mechanism. The model mainly extracted the high-dimensional features of the data by means of  one-dimensional convolution unit. The GRU unit learned the temporal characteristics in the data and enhanced the learning of important features by introducing attention mechanism to realize the learning of ultra-long sequences. In addition, by analyzing the correlation between the upper and lower waterway traffic flows, the ship AIS data of 6 sections in  the middle and lower reaches of the Yangtze River were extracted to construct the multi-segment ship traffic flow sequence data set and then input into the proposed model for training and testing. The  results show that compared with SAE, LSTM, GRU, CNN+GRU and GRU+Attention in the series prediction model,the proposed model has higher prediction accuracy in the prediction of different traffic flow parameters, and the prediction accuracy of traffic flow, traffic flow density and traffic flow speed are 95.42%, 97.33% and 94.99% respectively, which can better meet the needs of engineering applications.

参考文献

[1]SHAHRIARI S, GHASRI M, SISSON S A, et al. Ensemble of ARIMA: combining parametric and bootstrapping technique for traffic flow prediction[J]. Transportmetrica A: Transport Science, 2020, 16(3): 1552-1573.
[2]KUMAR P B, HARIHARAN K. Timeseries traffic flow prediction with hyper-parameter optimized ARIMA models for intelligent transportation system[J]. Journal of Scientific and Industrial Research (JSIR), 2022, 81(4): 408-415.
[3]张矢宇,杨宇昊,陈尘,等.基于回归-卡尔曼滤波组合模型的航道整治区域船舶交通流时空预测[J].大连海事大学学报,2021,47(1):37-44.
ZHANG S Y, YANG Y H, CHEN C, et al. Spatio-temporal prediction of ship traffic flow in waterway regulation area based on regression-Kalman filter combination model[J]. Journal of Dalian Maritime University,2021,47(1):37-44. (in Chinese)
[4]FAN D, ZHANG X L. Short-term traffic flow prediction method based on balanced binary tree and K-nearest neighbor nonparametric regression[C]// Proceedings of the 2017 2nd International Conference on Modelling, Simulation and Applied Mathematics (MSAM2017). [S.l.]: Atlantis Press, 2017: 118-121. doi:10.2991/msam-17.2017.27.
[5]殷礼胜,唐圣期,李胜,等.基于整合移动平均自回归和遗传粒子群优化小波神经网络组合模型的交通流预测[J].电子与信息学报,2019,41(9):2273-2279.
YIN L S, TANG S Q, LI S, et al. Traffic flow prediction based on hybrid model of auto-regressive integrated moving average and genetic particle swarm optimization wavelet neural network[J]. Journal of Electronics & Information Technology, 2019,41(9):2273-2279. (in Chinese)
[6]刘静,关伟.交通流预测方法综述[J].公路交通科技, 2004(3):82-85.
LIU J, GUAN W. A summary of traffic flow forecasting methods[J]. Journal of Highway and Transportation Research and Development, 2004(3):82-85. (in Chinese)
[7]卢建中,程浩.改进GA优化BP神经网络的短时交通流预测[J].合肥工业大学学报(自然科学版),2015,38(1):127-131.
LU J Z, CHENG H. Short-term traffic flow forecast based on modified GA optimized BP neural network[J]. Journal of Hefei University of Technology(Natural Science),2015,38(1):127-131. (in Chinese)
[8]肖进丽, 李晓磊. 基于集合经验模态分解和差分进化算法优化BP神经网络的船舶交通流预测[J]. 大连海事大学学报, 2018, 44(2):9-14.
XIAO J L, LI X L. Vessel traffic flow prediction method based on ensemble empirical mode decomposition and back propagation neural network optimized with differential evolution algorithm[J]. Journal of Dalian Maritime University, 2018, 44(2):9-14. (in Chinese)
[9]ZHANG J X, QU S R. Optimization of back propagation neural network under the adaptive genetic algorithm[J/OL]. Complexity, 2021. https://doi.org/10.1155/2021/1718234.
[10]李洁,林永峰.基于多时间尺度RNN的时序数据预测[J].计算机应用与软件, 2018, 35(7): 33-37.
LI J, LIN Y F. Prediction of time series data based on multi-time scale RNN[J]. Computer Applications and Software, 2018, 35(7): 33-37. (in Chinese)
[11]索永峰, 陈文科, 杨神化, 等. 基于深度神经网络的船舶交通流预测[J]. 集美大学学报(自然科学版), 2020,25(6):430-436.
SUO Y F, CHEN W K, YANG S H, et al. Prediction of ship traffic flow based on deep neural network[J]. Journal of Jimei University (Natural Science), 2020,25(6):430-436. (in Chinese)
[12]YANG B L, SUN S L, LI J Y, et al. Traffic flow prediction using LSTM with feature enhancement[J]. Neurocomputing, 2019, 332: 320-327.
[13]温惠英,张东冉,陆思园.GA-LSTM模型在高速公路交通流预测中的应用[J].哈尔滨工业大学学报, 2019, 51(9): 81-87.
WEN H Y, ZHANG D R, LU S Y. Application of GA-LSTM model in highway traffic flow prediction[J]. Journal of Harbin Institute of Technology, 2019, 51(9): 81-87. (in Chinese)

[14]张阳,胡月,陈德旺,等.基于GCN-BiLSTM的短时交通流预测模型[J/OL].武汉理工大学学报(交通科学与工程版),2023:

1-9(2023-01-19). http://kns.cnki.net/kcms/detail/42.1824.U.20220923.1352.044.html.

ZHANG Y, HU Y, CHEN D W, et al. Short-term traffic flow prediction model based on GCN-BiLSTM[J/OL]. Journal of Wuhan University of Technology(Transportation Science & Engineering),2023:1-9(2023-01-16).http://kns.cnki.net/kcms/detail/42.1824.U.20220923.1352.044.html. (in Chinese)
[15]MUTHUKUMARAN V, NATARAJAN R, KALADEVI A C, et al. Traffic flow prediction in inland waterways of Assam region using uncertain spatiotemporal correlative features[J]. Acta Geophysica,2022,70(6):2979-2990.
[16]刘敬贤,高广旭,刘奕,等.基于卷积神经网络及长短时记忆网络的短时船舶交通流量预测[J].中国航海,2022,45(2):56-61.
LIU J X, GAO G X, LIU Y, et al. Short-term water traffic flow prediction with convolutional neural network and long short-term memory network[J]. Navigation of China, 2022,45(2):56-61. (in Chinese)
[17]高昆仑, 杨帅, 刘思言,等. 基于一维卷积神经网络的电力系统暂态稳定评估[J]. 电力系统自动化, 2019,43(12):18-26.
GAO K L, YANG S, LIU S Y, et al. Transient stability assessment for power system based on one-dimensional convolutional neural network[J]. Automation of Electric Power Systems, 2019,43(12):18-26. (in Chinese)
[18]LI C K, HOU Y H, WANG P C, et al. Joint distance maps based action recognition with convolutional neural networks[J]. IEEE Signal Processing Letters, 2017, 24(5):624-628.
[19]司念文, 王衡军, 李伟,等. 基于注意力长短时记忆网络的中文词性标注模型[J]. 计算机科学, 2018, 45(4): 66-70.
SI N W, WANG H J, LI W, et al. Chinese part-of-speech tagging model using attention-based LSTM[J]. Computer Science, 2018, 45(4): 66-70. (in Chinese)
[20]张博中.基于船舶大数据的交通状态识别及可视化技术研究[D]. 大连:大连海事大学,2020.
ZHANG B Z. Research on traffic status recognition based on ship big data and visualization technology[D]. Dalian:Dalian Maritime University, 2020. (in Chinese)
[21]刘钊,刘敬贤,周锋,等.船舶交通流行为特征及其在海上交通组织中的应用[J].大连海事大学学报,2014,40(2):22-26.
LIU Z, LIU J X, ZHOU F, et al. Characteristics of ship traffic prevalence and its application in maritime traffic organization[J].Journal of Dalian Maritime University, 2014,40(2):22-26. (in Chinese)
文章导航

/