arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2506.08051cs.LG
收藏

ST-GraphNet:一种用于理解和预测自动驾驶车辆碰撞严重程度的时空图神经网络

ST-GraphNet: A Spatio-Temporal Graph Neural Network for Understanding and Predicting Automated Vehicle Crash Severity

Texas State University(德克萨斯州立大学)

Mahmuda Sultana Mimi, Md Monzurul Islam, Anannya Ghosh Tusti, Shriyank Somvanshi, Subasish Das

AI总结:

ST-GraphNet通过时空图神经网络整合多模态数据,以高准确率预测自动驾驶车辆碰撞严重程度。

摘要:

理解自动驾驶车辆(AV)碰撞严重程度的空间和时间动态对于推进城市交通安全和基础设施规划至关重要。在本工作中,我们介绍了ST-GraphNet,一种时空图神经网络框架,旨在通过使用细粒度和区域聚合的空间图来建模和预测AV碰撞严重程度。使用来自德克萨斯州(2024年)的2352份真实世界AV相关碰撞报告平衡数据集,包括地理坐标、碰撞时间戳、SAE自动化级别和叙述描述,我们构建了两种互补的图表示:(1)细粒度图,其中每个碰撞事件作为节点,边通过时空接近性定义;(2)粗粒度图,其中碰撞被聚合到基于Hexagonal Hierarchical Spatial Indexing(H3)的空间单元中,通过六边形邻接连接。图中的每个节点都增强了多模态数据,包括语义、空间和时间属性,包括使用预训练的Sentence-BERT模型从碰撞叙述中提取的文本嵌入。我们评估了多种图神经网络(GNN)架构,如图卷积网络(GCN)、图注意力网络(GAT)和动态时空GCN(DSTGCN),以分类碰撞严重程度和预测高风险区域。我们提出的ST-GraphNet,利用DSTGCN作为粗粒度H3图的骨干网络,达到了97.74%的测试准确率,明显优于最佳细粒度模型(64.7%的测试准确率)。这些发现突显了空间聚合、动态信息传递和多模态特征整合在捕捉AV碰撞严重程度背后的复杂时空模式中的有效性。

英文摘要:

Understanding the spatial and temporal dynamics of automated vehicle (AV) crash severity is critical for advancing urban mobility safety and infrastructure planning. In this work, we introduce ST-GraphNet, a spatio-temporal graph neural network framework designed to model and predict AV crash severity by using both fine-grained and region-aggregated spatial graphs. Using a balanced dataset of 2,352 real-world AV-related crash reports from Texas (2024), including geospatial coordinates, crash timestamps, SAE automation levels, and narrative descriptions, we construct two complementary graph representations: (1) a fine-grained graph with individual crash events as nodes, where edges are defined via spatio-temporal proximity; and (2) a coarse-grained graph where crashes are aggregated into Hexagonal Hierarchical Spatial Indexing (H3)-based spatial cells, connected through hexagonal adjacency. Each node in the graph is enriched with multimodal data, including semantic, spatial, and temporal attributes, including textual embeddings from crash narratives using a pretrained Sentence-BERT model. We evaluate various graph neural network (GNN) architectures, such as Graph Convolutional Networks (GCN), Graph Attention Networks (GAT), and Dynamic Spatio-Temporal GCN (DSTGCN), to classify crash severity and predict high-risk regions. Our proposed ST-GraphNet, which utilizes a DSTGCN backbone on the coarse-grained H3 graph, achieves a test accuracy of 97.74\%, substantially outperforming the best fine-grained model (64.7\% test accuracy). These findings highlight the effectiveness of spatial aggregation, dynamic message passing, and multi-modal feature integration in capturing the complex spatio-temporal patterns underlying AV crash severity.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑