LLM2D
统一的时空边缘增强图网络用于行人轨迹预测
Unified Spatial-Temporal Edge-Enhanced Graph Networks for Pedestrian Trajectory Prediction
作者: Ruochen Li, Tanqiu Qiao, Stamos Katsigiannis, Zhanxing Zhu, Hubert P. H. Shum
发布日期: 2/5/2025
arXiv ID: 2502.02504

摘要

arXiv:2502.02504v1 宣告类型:交叉 摘要:行人轨迹预测旨在根据历史路径预测未来的运动。时空(ST)方法通常分别建模行人间的空间交互和个体的时间依赖性。它们忽略了不同行人在各种时间步长之间交互的直接影响(即高阶跨时间交互)。这限制了它们捕捉ST间依赖性的能力,并阻碍了预测性能。为了解决这些限制,我们提出了UniEdge,并设计了三大创新。首先,我们引入了一种统一的时空图数据结构,将其高阶跨时间交互简化为一阶关系,从而允许在单一步骤中学习时空间依赖性,避免了多步聚合引起的信息丢失。其次,传统的GNN专注于聚合行人的节点特征,忽略了隐含在边特征中编码的交互模式的传播。我们提出了边缘到边缘节点到节点图卷积(E2E-N2N-GCN),这是一种新型的双图网络,可以同时建模行人间显式的N2N社会交互以及这些交互模式中隐含的E2E影响传播。最后,为了解决自回归架构的有限感受野和捕捉长期依赖性的挑战,我们引入了一种基于变压器编码器的预测器,以实现对时间相关性的全局建模。在ETH、UCY和SDD等多个数据集上,UniEdge 的表现优于现有最佳方法。