arXiv:2410.15589cs.LG2024-10被引 2

仅用一个城市数据,实现小样本交通预测

SSMT: Few-Shot Traffic Forecasting with Single Source Meta-Transfer

  • 用记忆增强注意力保存源城市空间知识,迁移至数据少的目标城市
  • 在五个真实数据集上优于现有方法,小样本下仍保持高精度
  • 适合缺乏多城市数据的城市交通系统快速部署

智能交通系统中的交通预测对城市运行至关重要。然而,许多城市缺乏足够的传感器或智能基础设施,导致数据稀缺。现有研究虽采用元学习利用多城市数据提升泛化能力,但跨城数据收集成本高。为此,我们提出仅依赖单一源城市数据的元迁移学习方法SSMT,实现小样本交通预测。通过记忆增强注意力机制,存储并选择性调用源城市异构空间信息;将正弦位置编码扩展为元学习任务,利用源城市多样时间模式构建训练信号;引入元位置编码,学习所有任务中最优的时间模式表征。在五个真实世界基准数据集上的实验表明,该方法在时间序列交通预测任务中优于多种现有方法。

原文摘要 · Abstract (English)

Traffic forecasting in Intelligent Transportation Systems (ITS) is vital for intelligent traffic prediction. Yet, ITS often relies on data from traffic sensors or vehicle devices, where certain cities might not have all those smart devices or enabling infrastructures. Also, recent studies have employed meta-learning to generalize spatial-temporal traffic networks, utilizing data from multiple cities for effective traffic forecasting for data-scarce target cities. However, collecting data from multiple cities can be costly and time-consuming. To tackle this challenge, we introduce Single Source Meta-Transfer Learning (SSMT) which relies only on a single source city for traffic prediction. Our method harnesses this transferred knowledge to enable few-shot traffic forecasting, particularly when the target city possesses limited data. Specifically, we use memory-augmented attention to store the heterogeneous spatial knowledge from the source city and selectively recall them for the data-scarce target city. We extend the idea of sinusoidal positional encoding to establish meta-learning tasks by leveraging diverse temporal traffic patterns from the source city. Moreover, to capture a more generalized representation of the positions we introduced a meta-positional encoding that learns the most optimal representation of the temporal pattern across all the tasks. We experiment on five real-world benchmark datasets to demonstrate that our method outperforms several existing methods in time series traffic prediction.

交通预测元学习小样本迁移学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。