TrajFlow用单次推断实现多模态轨迹预测,提升自动驾驶安全性。
TrajFlow: Multi-modal Motion Prediction via Flow Matching
- 基于流匹配的单次推断方法,生成多个合理未来轨迹。
- 在Waymo数据集上达到顶尖性能,多指标优于现有方法。
- 适合需要高效多模态预测的自动驾驶系统应用。
高效且准确的运动预测对保障自动驾驶安全和辅助决策至关重要,尤其在动态真实场景中需进行多模态预测。我们提出TrajFlow,一种基于流匹配的新型运动预测框架,解决了现有生成式轨迹预测方法在可扩展性和效率方面的挑战。与传统生成方法依赖独立同分布采样、需多次推理以捕捉多样化结果不同,TrajFlow可在一次前向传播中预测多个合理的未来轨迹,显著降低计算开销,同时保持预测结果的一致性。此外,我们设计了一种基于Plackett-Luce分布的排序损失,以改进轨迹不确定性估计;并提出自条件训练技术,在第二次前向传播中复用模型自身预测构建噪声输入,从而增强泛化能力并加速推理。在大规模Waymo Open Motion Dataset(WOMD)上的大量实验表明,TrajFlow在多个关键指标上均达到当前最优水平,验证了其在安全关键型自动驾驶应用中的有效性。代码及其他详情见项目网站https://traj-flow.github.io/。
原文摘要 · Abstract (English)
Efficient and accurate motion prediction is crucial for ensuring safety and informed decision-making in autonomous driving, particularly under dynamic real-world conditions that necessitate multi-modal forecasts. We introduce TrajFlow, a novel flow matching-based motion prediction framework that addresses the scalability and efficiency challenges of existing generative trajectory prediction methods. Unlike conventional generative approaches that employ i.i.d. sampling and require multiple inference passes to capture diverse outcomes, TrajFlow predicts multiple plausible future trajectories in a single pass, significantly reducing computational overhead while maintaining coherence across predictions. Moreover, we propose a ranking loss based on the Plackett-Luce distribution to improve uncertainty estimation of predicted trajectories. Additionally, we design a self-conditioning training technique that reuses the model's own predictions to construct noisy inputs during a second forward pass, thereby improving generalization and accelerating inference. Extensive experiments on the large-scale Waymo Open Motion Dataset (WOMD) demonstrate that TrajFlow achieves state-of-the-art performance across various key metrics, underscoring its effectiveness for safety-critical autonomous driving applications. The code and other details are available on the project website https://traj-flow.github.io/.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。