考虑外部因素的MCNN-ABiLSTM交通流量预测模型
MCNN-ABiLSTM traffic flow prediction model considering external factors
杨国威 1陈静 1张昭冲 1王伟1
作者信息
- 1. 天津职业技术师范大学信息技术工程学院,天津 300222
- 折叠
摘要
针对交通流量序列的时间依赖性、空间相关性及易受外部因素的干扰等问题,提出一种基于多尺度卷积神经网络和融合注意力机制的双向长短期记忆神经网络自适应融合预测模型(MCNN-ABiLSTM模型).通过串联的多尺度结构增强卷积神经网络的特征提取能力,融合注意力机制的双向长短期记忆网络提升对时序特征的连续性、周期性的挖掘能力,将2个分支特征自适应融合以提升交通流量预测的准确性.同时,通过计算各路口时序流量的皮尔逊相关系数分析交通流量的空间相关性,并提出改进粒子群算法(IPSO)设置外部因素标签值.实验结果表明,MCNN-ABiLSTM模型比其他基线模型预测准确性更高,RMSE、MAE以及MAPE均有明显下降.
Abstract
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.
关键词
双向长短期记忆神经网络/交通流量预测/注意力机制/多尺度卷积/特征融合/改进粒子群算法Key words
BiLSTM/traffic flow prediction/attention mechanism/multi-scale MCNN/feature fusion/improved particle swarm optimization引用本文复制引用
基金项目
天津市教委科研项目(2021KJ008)
天津市津南区科技计划(20220105)
出版年
2024
NETL
NSTL
万方数据