用状态空间模型高效建模社交互动,提升人群轨迹预测精度与速度
Social-Mamba: Socially-Aware Trajectory Forecasting with State-Space Models

- 将社交交互重构为结构化序列过程,引入双向信息流模块
- 在5个基准上达最优性能,参数量少且计算效率高
- 适合需要实时推理的智能导航系统研究者使用
人类轨迹预测对拥挤环境中的安全导航至关重要,需在精度与计算效率间取得平衡。高效建模社交交互是密集人群场景下表现优异的关键。然而,现有方法多依赖注意力机制,虽能捕捉复杂依赖关系,但计算成本呈二次增长,难以应对邻近者数量增多。近期提出的选通状态空间模型提供线性时间替代方案,但其固有的顺序设计与社交互动的非结构化、动态特性不匹配。为此,我们提出Social-Mamba,一种将社交交互重新构造成结构化序列过程的预测架构。核心为循环Mamba块,实现连续双向信息流动。Social-Mamba以自我中心网格组织代理,并引入社交三元组分解,将交互分解为时序、自我中心和目标中心扫描。通过可学习社交门控与全局扫描动态融合,生成精准高效的轨迹预测。在五个轨迹预测基准上的广泛实验表明,Social-Mamba在保持顶尖准确率的同时,展现出卓越的参数效率与计算可扩展性。此外,将其嵌入流匹配框架进一步提升精度与效率,确立其作为未来轨迹预测研究灵活可靠基础的地位。代码已公开:https://github.com/vita-epfl/Social-Mamba
原文摘要 · Abstract (English)
Human trajectory forecasting is crucial for safe navigation in crowded environments, requiring models that balance accuracy with computational efficiency. Efficiently modeling social interactions is key to performance in dense crowds. Yet, most recent methods rely on attention mechanisms, which are effective at capturing complex dependencies, but incur quadratic computational costs that scale poorly with the growing number of neighbors. Recently, Selective State-Space Models have provided a linear-time alternative; however, their inherently sequential design is misaligned with the unstructured and dynamic nature of social interactions. To address this challenge, we propose Social-Mamba, a forecasting architecture that reformulates social interactions as structured sequential processes. At its core is the Cycle Mamba block, a novel module that enables continuous bidirectional information flow. Social-Mamba organizes agents on an egocentric grid and introduces social triplet factorization, which decomposes interactions into temporal, egocentric, and goal-centric scans. These are dynamically integrated through a learnable social gate and global scan to generate accurate and efficient trajectory predictions. Extensive experiments on five trajectory forecasting benchmarks show that Social-Mamba achieves state-of-the-art accuracy while offering superior parameter efficiency and computational scalability. Furthermore, embedding Social-Mamba into a flow-matching framework further enhances both accuracy and efficiency, establishing it as a flexible and robust foundation for future trajectory forecasting research. The code is publicly available: https://github.com/vita-epfl/Social-Mamba
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。