LLM2D
跨年空间偏移鲁棒交通预测
Robust Traffic Forecasting against Spatial Shift over Years
作者: Hongjun Wang, Jiyuan Chen, Tong Pan, Zheng Dong, Lingyu Zhang, Renhe Jiang, Xuan Song
发布日期: 10/2/2024
arXiv ID: oai:arXiv.org:2410.00373v1

摘要

近年来,时空图神经网络(ST-GNNs)和Transformer在交通预测方面取得了显著进展,有效地捕捉了时间和空间相关性。时空模型的泛化能力在最近的学术讨论中引起了相当大的关注。然而,目前还没有专门针对交通分布外(OOD)场景的实质性数据集被提出。现有的ST-OOD方法要么局限于在现有数据上进行测试,要么需要对数据集进行人工修改。因此,当前时空模型在OOD场景中的泛化能力在很大程度上尚未得到探索。在本文中,我们使用新提出的交通OOD基准对最先进的模型进行了研究,令人惊讶的是,我们发现这些模型的性能出现了显著下降。通过仔细分析,我们将这种下降归因于模型无法适应以前未观察到的空间关系。为了解决这一挑战,我们提出了一种新颖的专家混合(MoE)框架,该框架在训练过程中学习一组图生成器(即图子),并自适应地组合它们以根据新的环境条件生成新的图,从而在测试期间处理空间分布变化。我们进一步将这一概念扩展到Transformer架构,取得了实质性的改进。我们的方法既简洁又有效,可以无缝地集成到任何时空模型中,在解决空间动态方面优于当前最先进的方法。