用骨架图神经网络实现通用图的高效最短路径搜索。
Skeleton-Guided Learning for Shortest Path Search
- 构建多层距离与跳数信息的骨架图,压缩图结构
- 模型预测节点间距离和跳数,支持精准搜索剪枝
- 适合大规模图、无需地理等特定特征的场景
最短路径搜索是基于图应用的核心操作,但现有方法存在显著局限。传统算法如Dijkstra和A*在图复杂度上升时效率下降,索引方法则需大量预处理与存储。近期学习方法多聚焦于空间图,依赖地理坐标等特定特征,泛化性差。本文提出一种无需领域特有特征的通用学习框架。核心是构建捕捉多层级距离与跳数信息的骨架图,通过骨架图神经网络(SGNN)学习节点嵌入并预测节点对间的距离与跳数。这些预测支持LSearch算法,利用模型驱动的剪枝缩小搜索空间同时保持精度。为应对更大规模图,引入分层训练策略,将图划分为子图并分别训练SGNN,形成可扩展的HLSearch方法。在五个真实世界图数据集上的实验表明,该框架在不同图类型上均表现优异,提供了灵活高效的基于学习的最短路径搜索方案。
原文摘要 · Abstract (English)
Shortest path search is a core operation in graph-based applications, yet existing methods face important limitations. Classical algorithms such as Dijkstra's and A* become inefficient as graphs grow more complex, while index-based techniques often require substantial preprocessing and storage. Recent learning-based approaches typically focus on spatial graphs and rely on context-specific features like geographic coordinates, limiting their general applicability. We propose a versatile learning-based framework for shortest path search on generic graphs, without requiring domain-specific features. At the core of our approach is the construction of a skeleton graph that captures multi-level distance and hop information in a compact form. A Skeleton Graph Neural Network (SGNN) operates on this structure to learn node embeddings and predict distances and hop lengths between node pairs. These predictions support LSearch, a guided search algorithm that uses model-driven pruning to reduce the search space while preserving accuracy. To handle larger graphs, we introduce a hierarchical training strategy that partitions the graph into subgraphs with individually trained SGNNs. This structure enables HLSearch, an extension of our method for efficient path search across graph partitions. Experiments on five diverse real-world graphs demonstrate that our framework achieves strong performance across graph types, offering a flexible and effective solution for learning-based shortest path search.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。