提出GNNUI模型,解决城市交通流量稀疏数据下的插值难题
Spatio-Temporal Graph Neural Network for Urban Spaces: Interpolating Citywide Traffic Volume
- 用掩码学习和节点特征融合,提升城市道路网插值精度
- 在90%到1%传感器覆盖率下,MAE仅从7.1升至10.5(Strava数据)
- 适用于低覆盖城市交通监测,尤其适合缺乏完整数据的场景
图神经网络在高速公路和主干道交通量预测中表现优异,但在城市环境中面临诸多挑战:城市路网结构多样、交通量高度过分散且含大量零值、空间依赖关系建模尚不明确,且传感器覆盖稀疏。本文提出城市交通量插值图神经网络(GNNUI),通过掩码算法学习插值机制,融合节点特征捕捉道路功能角色,并采用针对零膨胀分布设计的损失函数。同时,构建两个新的大规模开源城市交通数据集,涵盖柏林与纽约市的自行车骑行(Strava)和出租车(Taxi)数据。GNNUI在多个指标(MAE、RMSE、真零率、KL散度)上优于现有图基插值方法,在传感器覆盖率从90%降至1%时仍保持鲁棒性。例如,Strava数据集上MAE由7.1增至10.5,出租车数据集上由23.0增至40.4。结果表明,GNNUI在极端数据稀缺条件下仍具强性能。此外,研究了图连通性选择对模型准确率的影响。
原文摘要 · Abstract (English)
Graph Neural Networks have shown strong performance in traffic volume forecasting, particularly on highways and major arterial networks. Applying them to urban settings, however, presents unique challenges: urban networks exhibit greater structural diversity, traffic volumes are highly overdispersed with many zeros, the best way to account for spatial dependencies remains unclear, and sensor coverage is often very sparse. We introduce the Graph Neural Network for Urban Interpolation (GNNUI), a novel urban traffic volume estimation approach. GNNUI employs a masking algorithm to learn interpolation, integrates node features to capture functional roles, and uses a loss function tailored to zero-inflated traffic distributions. In addition to the model, we introduce two new open, large-scale urban traffic volume benchmarks, covering different transportation modes: Strava cycling data from Berlin and New York City taxi data. GNNUI outperforms recent, some graph-based, interpolation methods across metrics (MAE, RMSE, true-zero rate, Kullback-Leibler divergence) and remains robust from 90% to 1% sensor coverage. For example, on the Strava dataset, the MAE increases only from 7.1 to 10.5, and on the Taxi dataset, from 23.0 to 40.4. These results demonstrate that GNNUI maintains strong performance despite extreme data scarcity, a common condition in real-world urban settings. We also examine how graph connectivity choices influence model accuracy.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。