针对稀有交通场景的轨迹预测难题,提出基于偏差知识的Transformer模型
CDKFormer: Contextual Deviation Knowledge-Based Transformer for Long-Tail Trajectory Prediction
- 构建上下文偏差知识模块,融合场景与车辆状态异常特征
- 在Argoverse 2数据集上显著提升长尾轨迹预测准确率
- 适合自动驾驶系统在复杂城市环境中提升安全性
预测周边车辆未来运动对保障自动驾驶汽车在城市交通环境中的安全运行与高效导航至关重要。现有轨迹预测方法多关注整体性能提升,但在长尾场景下表现不佳,导致罕见情况下的预测质量差,显著增加安全事故风险。以Argoverse 2运动预测数据集为例,我们从个体运动和群体交互两个角度分析轨迹样本的长尾特性,并提取偏差特征以区分异常与常规场景。在此基础上,提出CDKFormer——一种基于上下文偏差知识的Transformer模型。该模型集成注意力机制的场景上下文融合模块,编码时空交互与道路拓扑信息;另设偏差特征融合模块,捕捉目标车辆状态的动态偏离。进一步设计双查询解码器,结合多流解码块,分步解码异质场景偏差特征并生成多模态轨迹预测。大量实验表明,相比现有方法,CDKFormer在长尾轨迹预测上达到最优性能,显著提升预测精度与鲁棒性,推动自动驾驶系统在复杂真实环境中的可靠性。
原文摘要 · Abstract (English)
Predicting the future movements of surrounding vehicles is essential for ensuring the safe operation and efficient navigation of autonomous vehicles (AVs) in urban traffic environments. Existing vehicle trajectory prediction methods primarily focus on improving overall performance, yet they struggle to address long-tail scenarios effectively. This limitation often leads to poor predictions in rare cases, significantly increasing the risk of safety incidents. Taking Argoverse 2 motion forecasting dataset as an example, we first investigate the long-tail characteristics in trajectory samples from two perspectives, individual motion and group interaction, and deriving deviation features to distinguish abnormal from regular scenarios. On this basis, we propose CDKFormer, a Contextual Deviation Knowledge-based Transformer model for long-tail trajectory prediction. CDKFormer integrates an attention-based scene context fusion module to encode spatiotemporal interaction and road topology. An additional deviation feature fusion module is proposed to capture the dynamic deviations in the target vehicle status. We further introduce a dual query-based decoder, supported by a multi-stream decoder block, to sequentially decode heterogeneous scene deviation features and generate multimodal trajectory predictions. Extensive experiments demonstrate that CDKFormer achieves state-of-the-art performance, significantly enhancing prediction accuracy and robustness for long-tailed trajectories compared to existing methods, thus advancing the reliability of AVs in complex real-world environments.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。