计算机科学2021,Vol.48Issue(7) :112-117.DOI:10.11896/jsjkx.201000089

面向出租车空载时间预测的多任务时空图卷积网络

Multi-task Spatial-Temporal Graph Convolutional Network for Taxi Idle Time Prediction

宋龙泽 万怀宇 郭晟楠 林友芳
计算机科学2021,Vol.48Issue(7) :112-117.DOI:10.11896/jsjkx.201000089

面向出租车空载时间预测的多任务时空图卷积网络

Multi-task Spatial-Temporal Graph Convolutional Network for Taxi Idle Time Prediction

宋龙泽 1万怀宇 2郭晟楠 林友芳
扫码查看

作者信息

  • 1. 北京交通大学计算机科学与技术学院 北京 100044
  • 2. 交通数据分析与挖掘北京重点实验室 北京 100044
  • 折叠

摘要

出租车空载时间严重影响交通资源的利用效率和司机的收益.准确的出租车空载时间预测可以有效地指导司机进行合理的路径规划,辅助打车平台进行高效的资源调度.然而,在实际场景中,城市不同区域的空载时间受到区域车流量、客流量以及历史空载时长等多方面因素的影响.为解决该问题,提出一种基于多任务框架的时空图卷积网络(MSTGCN)模型.MSTGCN采用一种新颖的时空图卷积结构,全面建模上述影响空载时间的各种时、空相关性因素.使用多任务学习框架从不同视角学习数据的特征表示,并提出一种多任务注意力融合机制,通过对辅助任务信息的筛选来提升主任务的信息获取能力和预测性能.将所提模型在两个公开的滴滴数据集上进行了充分的实验,其取得了优于其他方法的预测效果.

关键词

时空数据预测/出租车空载时长/图卷积网络/多任务学习/注意力机制

引用本文复制引用

基金项目

出版年

2021
计算机科学
重庆西南信息有限公司(原科技部西南信息中心)

计算机科学

CSTPCDCSCD北大核心
影响因子:0.944
ISSN:1002-137X
被引量4
参考文献量1
段落导航相关论文