提出局部多尺度图模型,高效建模交通中异质与多尺度交互。
HeLoFusion: An Efficient and Scalable Encoder for Modeling Heterogeneous and Multi-Scale Interactions in Trajectory Prediction
- 以每个智能体为中心构建局部多尺度图,捕捉成对与群体交互。
- 在Waymo数据集上达到Soft mAP和minADE新基准,优于现有方法。
- 适合关注自动驾驶轨迹预测中复杂社会行为建模的研究者。
自动驾驶中的多智能体轨迹预测需全面理解复杂的社交动态。现有方法难以充分捕捉多尺度交互与异质智能体行为的共存问题。为此,本文提出HeLoFusion,一种高效可扩展的编码器,用于建模异质性与多尺度交互。该方法不依赖全局上下文,而是以每个智能体为中心构建局部多尺度图,有效建模直接成对依赖及复杂群体交互(如车队行驶或人群聚集)。此外,通过聚合-分解消息传递机制与类型特异性特征网络,解决智能体异质性挑战,学习细粒度的类型相关交互模式。这种基于局部性的设计,实现了多层次社交上下文的合理表征,生成强大且富有表现力的智能体嵌入。在具有挑战性的Waymo Open Motion Dataset上,HeLoFusion取得当前最优性能,刷新Soft mAP与minADE等关键指标。结果表明,显式建模多尺度与异质交互的局部性架构,是推进运动预测的有效策略。
原文摘要 · Abstract (English)
Multi-agent trajectory prediction in autonomous driving requires a comprehensive understanding of complex social dynamics. Existing methods, however, often struggle to capture the full richness of these dynamics, particularly the co-existence of multi-scale interactions and the diverse behaviors of heterogeneous agents. To address these challenges, this paper introduces HeLoFusion, an efficient and scalable encoder for modeling heterogeneous and multi-scale agent interactions. Instead of relying on global context, HeLoFusion constructs local, multi-scale graphs centered on each agent, allowing it to effectively model both direct pairwise dependencies and complex group-wise interactions (\textit{e.g.}, platooning vehicles or pedestrian crowds). Furthermore, HeLoFusion tackles the critical challenge of agent heterogeneity through an aggregation-decomposition message-passing scheme and type-specific feature networks, enabling it to learn nuanced, type-dependent interaction patterns. This locality-focused approach enables a principled representation of multi-level social context, yielding powerful and expressive agent embeddings. On the challenging Waymo Open Motion Dataset, HeLoFusion achieves state-of-the-art performance, setting new benchmarks for key metrics including Soft mAP and minADE. Our work demonstrates that a locality-grounded architecture, which explicitly models multi-scale and heterogeneous interactions, is a highly effective strategy for advancing motion forecasting.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。