arXiv:2501.12281cs.LG2025-01被引 22

无需传感器也能精准预测未知路段交通,还能适应传感器增减。

MoGERNN: An Inductive Traffic Predictor for Unobserved Locations

  • 用专家路由机制动态分配节点,高效捕捉不同区域的交通依赖。
  • 在真实数据集上对有无传感器区域均表现更优,尤其在无感测区提升显著。
  • 适合交通管理、智能导航等需灵活应对传感器变动的场景。

在部分观测的道路网络中,如何预测未部署传感器区域的交通状态?交通预测对智能交通系统至关重要,深度学习方法虽表现优异,但多数假设所有目标位置均部署传感器,这在现实中因成本限制难以实现。此外,现有方法对传感网络结构变化敏感,哪怕小规模传感器调整也需昂贵的重新训练。为此,我们提出MoGERNN,一种归纳式时空图模型,包含两个核心组件:(i) 稀疏门控的图专家混合(MoGE),动态将节点路由至专用图聚合器,高效建模异质空间依赖;(ii) 图编码器-解码器架构,利用嵌入表示同时捕捉空间与时间依赖,实现全面交通状态预测。在两个真实世界数据集上的实验表明,MoGERNN在观测与未观测位置上均持续优于基线方法。即使在无传感器区域,也能准确预测拥堵演变,为交通管理提供关键信息。此外,该模型可适应传感器网络变化,在性能上甚至媲美重训练版本。通过不同传感器数量的测试验证其稳定性,消融实验进一步证明关键模块的有效性。代码已开源:https://github.com/ZJU-TSELab/MoGERNN。

原文摘要 · Abstract (English)

Given a partially observed road network, how can we predict the traffic state of interested unobserved locations? Traffic prediction is crucial for advanced traffic management systems, with deep learning approaches showing exceptional performance. However, most existing approaches assume sensors are deployed at all locations of interest, which is impractical due to financial constraints. Furthermore, these methods are typically fragile to structural changes in sensing networks, which require costly retraining even for minor changes in sensor configuration. To address these challenges, we propose MoGERNN, an inductive spatio-temporal graph model with two key components: (i) a Mixture of Graph Experts (MoGE) with sparse gating mechanisms that dynamically route nodes to specialized graph aggregators, capturing heterogeneous spatial dependencies efficiently; (ii) a graph encoder-decoder architecture that leverages these embeddings to capture both spatial and temporal dependencies for comprehensive traffic state prediction. Experiments on two real-world datasets show MoGERNN consistently outperforms baseline methods for both observed and unobserved locations. MoGERNN can accurately predict congestion evolution even in areas without sensors, offering valuable information for traffic management. Moreover, MoGERNN is adaptable to the changes of sensor network, maintaining competitive performance even compared to its retrained counterpart. Tests performed with different numbers of available sensors confirm its consistent superiority, and ablation studies validate the effectiveness of its key modules. The code of this work is publicly available at: https://github.com/ZJU-TSELab/MoGERNN.

交通预测图神经网络归纳学习

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