arXiv:2607.09741cs.ROcs.AI2026-07TPAMI

用小世界网络建模交通交互,提升自动驾驶轨迹预测的准确性与泛化能力。

SWIFT: A Small-World Interaction Framework for Flow-Aware Trajectory Prediction in Autonomous Driving

论文配图:SWIFT: A Small-World Interaction Framework for Flow-Aware Trajectory Prediction in Autonomous Driving
图 1 · 摘自论文原文
  • 引入小世界交互网络和交通流编码器,融合局部与全局依赖关系。
  • 在nuScenes、MoCAD、NGSIM数据集上均超越基线模型,尤其在复杂场景表现优异。
  • 适合关注交通预测泛化性与鲁棒性的研究者,尤其对少样本场景有优势。

自动驾驶中的精准轨迹预测依赖于对交通参与者动态且上下文相关的交互建模。然而,现有方法多为纯数据驱动,缺乏结构先验,导致在分布偏移下泛化能力受限。本文通过交通网络的结构与动态重新审视交互建模,提出统一框架SWIFT(Small-World Interaction Framework for Trajectory Prediction)。SWIFT结合小世界网络与交通流理论,引入小世界交互网络以捕捉局部与全局依赖,并设计流量状态编码器根据场景级交通状态自适应调整交互结构。进一步通过多关系图模块显式编码直接及高阶代理关系。在nuScenes、MoCAD和NGSIM三个真实世界数据集上的大量实验表明,SWIFT在不同交通状态下持续优于强基线模型。除精度提升外,其在未见地点、噪声观测及小样本训练下仍具良好泛化性与鲁棒性,验证了其结构感知设计的有效性。

原文摘要 · Abstract (English)

Accurate trajectory prediction in autonomous driving hinges on modeling dynamic and context-dependent interactions among traffic agents. However, most existing approaches are purely data-driven and lack structural priors, which limits their generalization under distribution shifts. In this work, interaction modeling is revisited through the structure and dynamics of traffic networks, and SWIFT (Small-World Interaction Framework for Trajectory prediction) is proposed as a unified framework that integrates small-world networks with traffic flow theory. SWIFT introduces structural inductive biases via a Small-World Interaction Network that captures both local and global dependencies, and a Flow Regime Encoder that adapts the interaction structure to scene-level traffic states. Interaction reasoning is further enhanced through a multi-relational graph module that explicitly encodes direct and higher-order agent relationships. Extensive experiments on three real-world datasets, nuScenes, MoCAD, and NGSIM, show that SWIFT consistently outperforms strong baselines in prediction accuracy across diverse traffic regimes. Beyond accuracy, SWIFT exhibits improved generalization to unseen locations and regimes, robustness under noisy observations, and strong performance with limited training data, supporting the effectiveness of its structure-aware design.

轨迹预测小世界网络自动驾驶交互建模

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。