提出可跨城市迁移的图神经网络初始化方法,解决交通预测模型泛化难题。
GUIDED Network-Agnostic Feature Initialization for Spatial Transferability in GNN-based Models

- 用虚拟边注入需求信息,实现不依赖具体路网的特征初始化
- 在数据稀缺下仍保持领先性能,训练速度提升约50%
- 适合需跨城市部署的交通与物流类空间预测任务
交通分配问题是交通规划中的核心但计算成本高昂环节。图神经网络虽成为快速的数据驱动替代方案,但其实际应用受限于空间泛化能力不足。现有模型依赖固定拓扑的归纳式特征初始化,难以迁移到新城市。本文提出一种网络无关的初始化层——几何无约束归纳需求嵌入(GUIDED),将出行需求作为标量属性注入辅助虚拟边,而非特定节点特征,使输入空间标准化,不受网络规模影响。在多个城市路网上的实验表明,集成GUIDED的异质图注意力网络(HetGAT)在单网络任务中保持顶尖预测精度,对分布外需求模式更具鲁棒性,且在严重数据稀缺时仍显著优于基线。该初始化方法无需人为输入统一化即可实现高度参数高效的跨网络迁移学习,为真正归纳模型奠定基础。同时,优化的散射操作使每轮训练时间减少约50%。尽管聚焦机动车交通,该空间拓扑抽象亦适用于货运物流、多模式网络优化等广义起讫点空间问题。
原文摘要 · Abstract (English)
The Traffic Assignment Problem is a fundamental but computationally expensive component of transportation planning. While Graph Neural Networks have emerged as fast, data-driven surrogates, their practical deployment is severely constrained by a spatial generalization gap. Standard models rely on transductive feature initializations that tie travel demand to fixed network topologies, preventing seamless transfer to new urban environments. To overcome this structural limitation, this research proposes a network-agnostic initialization layer, termed Geometrically Unconstrained Inductive Demand EmbeDding (GUIDED). By injecting travel demand as a scalar attribute on auxiliary virtual links rather than as specific node features, this modular framework standardizes the input space regardless of network scale. Extensive experimental evaluation across multiple urban topologies demonstrates that a Heterogeneous Graph Attention Network (HetGAT) model integrated with the proposed GUIDED layer maintains state-of-the-art predictive accuracy on single-network tasks, while demonstrating superior robustness to out-of-distribution demand patterns and maintaining a distinct performance advantage over the baseline even under severe data scarcity. Notably, the proposed feature initialization enables highly parameter-efficient domain adaptation for inter-network transfer learning without artificial input homogenization, establishing a robust foundation for truly inductive models. At the same time, the optimized scatter operations of the initialization layer yield an approximate 50% reduction in training time per epoch compared to the baseline approach. Furthermore, while demonstrated on vehicular traffic, this fundamental abstraction of spatial topology provides a versatile blueprint for generalized origin-destination spatial problems, such as freight logistics and multimodal network optimization.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。