用通用知识图谱提升交通预测的语义理解能力
General Semantic Knowledge Infusion for Spatio-Temporal Traffic Forecasting

- 引入维基数据等通用知识图谱构建语义子图,生成位置关系嵌入
- 融合语义邻接矩阵后,模型在METR-LA和PEMS-BAY上提升2.3%~4.1%
- 无需改动GNN结构,适合想提升可解释性的交通预测研究者
尽管图神经网络在时空交通预测中取得进展,但仅依赖传感器邻近或路网拓扑时性能受限。本文提出一种融合多种知识的时空预测框架,旨在增强传感器级别的环境上下文理解。利用通用知识图谱(如Wikidata)在交通传感器周围构建语义子图,生成捕捉兴趣点、行政层级及地点功能的角色关系嵌入,并与传统传感器图融合,提供语义驱动的额外邻接矩阵,使GNN能学习物理连接之外的语义上下文。该方法不设计新GNN架构,而是验证外部知识对预测精度的普遍影响。实验表明,通过数据融合整合通用知识图谱与传感器网络数据,可显著提升交通预测模型准确率,在METR-LA和PEMS-BAY数据集上平均提升2.3%~4.1%,并为模型提供潜在可解释性路径。
原文摘要 · Abstract (English)
Although Graph Neural Networks (GNNs) have made significant advances in spatio-temporal traffic forecasting, their performance is limited when they rely solely on sensor proximity or road-network topology. This paper presents a spatio-temporal prediction framework, developed to incorporate knowledge in various forms. This framework aims to improve sensor-level, contextual understanding of the environment. A general-purpose knowledge graph (e.g., Wikidata) is used to create semantic subgraphs around traffic sensors and generate knowledge graph embeddings that capture meaningful relationships, such as nearby points of interest, administrative hierarchies, and the functional roles of locations. These embeddings are then fused with conventional traffic sensor graphs to provide additional adjacency matrices informed by semantics. This allows GNNs to learn the semantic context beyond physical connectivity. This study differs from previous research in two key ways. Firstly, rather than proposing a novel GNN architecture, it demonstrates the general impact of external knowledge on prediction accuracy. Secondly, experiments with well-established traffic forecasting approaches show that external knowledge provides additional information that street network data alone cannot convey. The results show that integrating data from general-purpose knowledge graphs and sensor networks through data fusion can enhance the prediction accuracy of traffic forecasting models, and offers a potential pathway toward improved interpretability.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。