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

• 作者: 刘忠伟 李萍 周盛 闫豆豆 李颖 安毅生
• 来源期刊: 计算机测量与控制
• 关键词: 交通流预测 时空特性 图神经网络 知识蒸馏 相互学习
• 年，卷（期）: 2024,（4）
• 页码范围: 166-173
• 页数: 8页
• 语种: 中文
• DOI: 10.16526/j.cnki.11-4762/tp.2024.04.024

摘要

交通流预测是智能交通系统(ITS)的核心，其中时空特性是最主要的特征；由于不同道路之间存在复杂的空间相关性和时间依赖性，因此交通流预测成为一项具有挑战性的任务；目前，基于图卷积神经网络的预测方法在网络局部以及整体的特征感知和提取方面，仍存在优化空间；为了解决以上问题，提出了一种基于图神经网络的优化模型(DMCRNN),该模型以DCRNN为基准模型，利用相互学习策略对其进行优化；在训练过程中，两个DCRNN网络之间相互学习、相互指导，以此来增强每个网络的特征学习能力；在METR-LA和PEMS-BAY两个真实数据集上验证优化策略的有效性；结果表明，经过优化后的模型预测误差显著降低，在两个数据集上一小时的MAE与DCRNN相比分别降低了0.15和0.12,即相互学习优化策略具有较好的性能。