基于图对比学习的多模态交通流量协同预测方法 |
| |
作者姓名: | 肖杨 秦建阳 李肯立 王鸽 李瑞 廖清 |
| |
作者单位: | 1. 哈尔滨工业大学(深圳)计算机科学与技术学院;2. 湖南大学信息科学与工程学院;3. 西安交通大学计算机学院;4. 西安电子科技大学计算机科学与技术学院;5. 鹏城实验室 |
| |
基金项目: | 国家重点研发计划(2020YFB2104003)~~; |
| |
摘 要: | 准确的城市区域交通流量预测对市区车辆调度、公交系统优化等具有重要指导意义。目前,大多数现有的交通流量预测方法只考虑规则网格区域上单一种类的交通流量预测,忽略了交通网络中空间的不规则性和异质性以及不同出行模式交通流的交互性。针对上述问题,提出了一种基于图对比学习的多模态交通流量协同预测方法(CoF-MGCL),以揭示各类出行方式之间的交互对不规则异构区域的交通需求的影响。具体而言,根据现实中城市的不规则区域采集多模态流量数据,包括各类出行模式流量(如自行车和出租车流量)和总流量;并对不规则区域构建多关系异构图,包括地理邻近和功能相似关系。通过异构图编码模块,可以结合异构图中不同的关系来学习各区域各类交通流量的高质量表征信息。学习到的单一交通流量表征经过注意力机制加权融合后与总交通流量表征进行图对比学习,以捕获不同出行模式之间的交互关系。最后,使用互信息约束实现多模态流量的协同预测,确保多模态信息学习最大化。为了实现不规则区域的多模态交通流量预测,自行构建了新的纽约市曼哈顿区和芝加哥市两地多模态交通流量数据集,并在此基础上进行实验。实验结果表明,所提方法可以结合现有的单模态交通流量预测模型...
|
关 键 词: | 交通流量预测 多模态 异构图表示 图对比学习 互信息 |
|