考虑外部因素的MCNN-ABiLSTM交通流量预测模型  

MCNN-ABiLSTM traffic flow prediction model considering external factors

作　　者：杨国威 陈静[1] 张昭冲 王伟 YANG Guowei;CHEN Jing;ZHANG Zhaochong;WANG Wei(School of Information Technology and Engineering,Tianjin University of Technology and Education,Tianjin 300222,China)

机构地区：[1]天津职业技术师范大学信息技术工程学院,天津300222

出　　处：《天津职业技术师范大学学报》2024年第1期66-73,共8页Journal of Tianjin University of Technology and Education

基　　金：天津市教委科研计划项目(2021KJ008);天津市津南区科技计划项目(20220105).

摘　　要：针对交通流量序列的时间依赖性、空间相关性及易受外部因素的干扰等问题,提出一种基于多尺度卷积神经网络和融合注意力机制的双向长短期记忆神经网络自适应融合预测模型(MCNN-ABiLSTM模型)。通过串联的多尺度结构增强卷积神经网络的特征提取能力,融合注意力机制的双向长短期记忆网络提升对时序特征的连续性、周期性的挖掘能力,将2个分支特征自适应融合以提升交通流量预测的准确性。同时,通过计算各路口时序流量的皮尔逊相关系数分析交通流量的空间相关性,并提出改进粒子群算法(IPSO)设置外部因素标签值。实验结果表明,MCNN-ABiLSTM模型比其他基线模型预测准确性更高,RMSE、MAE以及MAPE均有明显下降。An adaptive fusion prediction model of bidirectional long short-term memory neural network based on multi-scale convolutional neural network and fusion attention mechanism(MCNN-ABiLSTM)is proposed for the time depen-dency,spatial correlation,and susceptibility to external factors in traffic flow sequences.By concatenating a multi-scale structure to enhance the feature extraction capability of the convolutional neural network and integrating a bidirectional long short-term memory network with a fusion attention mechanism to improve the continuity and periodicity mining a-bility of temporal features,the adaptive fusion of two branch features is performed to enhance the accuracy of traffic flow prediction.Additionally,the spatial correlation of traffic flow is analyzed by calculating the Pearson correlation co-efficients of time series traffic volumes at each crossing.Furthermore,an improved Particle Swarm Optimization(IPSO)algorithm is proposed to set the label values of external factors.The experimental results indicate that the MCNN-A-BiLSTM model outperforms other baseline models in terms of prediction accuracy,with significant reductions in RMSE,MAE,and MAPE.

关 键 词：双向长短期记忆神经网络 交通流量预测 注意力机制 多尺度卷积 特征融合 改进粒子群算法 

分 类 号：TP391.9[自动化与计算机技术—计算机应用技术]