3 个月前

领域对抗时空网络:一种跨城市的短期交通预测可迁移框架

领域对抗时空网络:一种跨城市的短期交通预测可迁移框架

摘要

精准的实时交通预测对于智能交通系统(ITS)至关重要,是各类智慧出行应用的基石。尽管该研究领域目前主要由深度学习主导,但近期研究表明,通过设计新型模型结构来提升预测精度的潜力已逐渐趋于饱和。相反,我们提出一种新思路:通过在具有不同数据分布和网络拓扑结构的城市之间迁移“与预测相关的知识”,实现性能的进一步提升。为此,本文提出一种新型可迁移交通预测框架——领域对抗时空网络(Domain Adversarial Spatial-Temporal Network, DASTNet)。DASTNet首先在多个源城市网络上进行预训练,随后利用目标城市的真实交通数据进行微调。具体而言,我们结合图表示学习与对抗域自适应技术,学习具备领域不变性的节点嵌入表示,并将其进一步用于建模时间维度上的交通流量变化。据我们所知,DASTNet是首个将对抗多域自适应方法应用于全网交通预测问题的研究。在三个基准数据集上,DASTNet始终优于所有现有的最先进基线方法。该模型已成功部署于香港新安装的交通检测设备中,一旦检测器投入使用,即可在一天内迅速提供高精度的交通预测结果。总体而言,本研究为提升交通预测能力提供了一种新范式,并为缺乏历史交通数据的城市提供了切实可行的技术路径。

代码仓库

yihongt/dastnet
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
traffic-prediction-on-pemsd4-10-days-trainingDASTNet
MAE: 19.25
MAPE: 13.30
RMSE: 28.91
traffic-prediction-on-pemsd4-10-days-training-1DASTNet
MAE: 20.67
MAPE: 14.56
RMSE: 30.78
traffic-prediction-on-pemsd4-10-days-training-2DASTNet
MAE: 22.82
MAPE: 16.1
RMSE: 33.77
traffic-prediction-on-pemsd7-10-days-trainingDASTNet
MAE: 20.91
MAPE: 8.95
RMSE: 31.85
traffic-prediction-on-pemsd7-10-days-training-1DASTNet
MAE: 22.96
MAPE: 9.87
RMSE: 34.8
traffic-prediction-on-pemsd7-10-days-training-2DASTNet
MAE: 26.88
MAPE: 11.75
RMSE: 40.12
traffic-prediction-on-pemsd8-10-days-trainingDASTNet
MAE: 15.26
MAPE: 9.64
RMSE: 22.7
traffic-prediction-on-pemsd8-10-days-training-1DASTNet
MAE: 16.41
MAPE: 10.46
RMSE: 24.57
traffic-prediction-on-pemsd8-10-days-training-2DASTNet
MAE: 18.84
MAPE: 11.72
RMSE: 28.06

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
领域对抗时空网络:一种跨城市的短期交通预测可迁移框架 | 论文 | HyperAI超神经