首页|基于相互学习的短时交通流预测研究

基于相互学习的短时交通流预测研究

扫码查看
交通流预测是智能交通系统(ITS)的核心,其中时空特性是最主要的特征;由于不同道路之间存在复杂的空间相关性和时间依赖性,因此交通流预测成为一项具有挑战性的任务;目前,基于图卷积神经网络的预测方法在网络局部以及整体的特征感知和提取方面,仍存在优化空间;为了解决以上问题,提出了一种基于图神经网络的优化模型(DMCRNN),该模型以DCRNN为基准模型,利用相互学习策略对其进行优化;在训练过程中,两个DCRNN网络之间相互学习、相互指导,以此来增强每个网络的特征学习能力;在METR-LA和PEMS-BAY两个真实数据集上验证优化策略的有效性;结果表明,经过优化后的模型预测误差显著降低,在两个数据集上一小时的MAE与DCRNN相比分别降低了0。15和0。12,即相互学习优化策略具有较好的性能。
Research on Short-term Traffic Flow Prediction Based on Mutual Learning
Traffic flow prediction is the core of intelligent transportation systems(ITS),with spatiotemporal characteristics being the most important feature.Due to the complex spatial correlations and time dependencies between different roads,the traffic flow prediction has become a challenging task.Currently,prediction methods based on graph convolutional neural networks are further op-timized on the feature perception and extraction of local and global networks.To address above issues,a diffusion mutual convolution-al recurrent neural network(DMCRNN)optimized model based on graph neural networks is proposed.The model is based on the DCRNN as a benchmark model,and the mutual learning strategy is utilized to optimize it.During training,two DCRNN networks learn from and guide each other to enhance their respective feature learning capabilities.The effectiveness of the optimization strategy is verified on two real datasets of the METR-LA and PEMS-BAY.The results show that the optimized model significantly reduces the prediction errors,with a decrease in the MAE of 0.15 and 0.12 for one hour on the two datasets than that in the DCRNN,respective-ly,indicating that the mutual learning optimization strategy has a good performance.

traffic flow predictionspatiotemporal characteristicsgraph convolutional neural networksknowledge distillationmutual learning

刘忠伟、李萍、周盛、闫豆豆、李颖、安毅生

展开 >

云基智慧工程有限公司,广东深圳 518000

长安大学信息工程学院,西安 710064

交通流预测 时空特性 图神经网络 知识蒸馏 相互学习

国家自然科学基金青年基金国家自然科学基金面上项目国家重点研发计划

52002031521723252021YFB1600104

2024

计算机测量与控制
中国计算机自动测量与控制技术协会

计算机测量与控制

CSTPCD
影响因子:0.546
ISSN:1671-4598
年,卷(期):2024.32(4)
  • 30