arXiv:2507.22615cs.CV2025-07ICCV被引 10

用可控扩散模型主动生成罕见轨迹,提升自动驾驶预测能力

Generative Active Learning for Long-tail Trajectory Prediction via Controllable Diffusion Model

  • 通过生成式主动学习识别模型失败的罕见场景并增强训练
  • 在WOMD和Argoverse2数据集上显著提升尾部样本预测准确率
  • 适合关注自动驾驶长尾问题的研究者与工程师

尽管数据驱动的轨迹预测提升了自动驾驶系统的可靠性,但在罕见的长尾场景中仍表现不佳。现有方法多通过修改模型结构(如超网络)应对,而本文提出不改变模型结构,仅优化训练过程的新方法GALTraj,首次将生成式主动学习引入轨迹预测。该方法主动识别模型失效的罕见样本,并利用可控扩散模型生成多样、真实且保留尾部特征的轨迹进行数据增强。为此,设计了面向尾部场景的生成策略,通过定制化扩散引导,生成既捕捉罕见行为又符合交通规则的轨迹。不同于以往仅关注场景多样性的仿真方法,GALTraj首次证明模拟器驱动的数据增强能有效促进长尾学习。在多个轨迹数据集(WOMD、Argoverse2)和主流骨干网络(QCNet、MTR)上的实验表明,该方法不仅显著提升尾部样本性能,也改善头部样本精度。

原文摘要 · Abstract (English)

While data-driven trajectory prediction has enhanced the reliability of autonomous driving systems, it still struggles with rarely observed long-tail scenarios. Prior works addressed this by modifying model architectures, such as using hypernetworks. In contrast, we propose refining the training process to unlock each model's potential without altering its structure. We introduce Generative Active Learning for Trajectory prediction (GALTraj), the first method to successfully deploy generative active learning into trajectory prediction. It actively identifies rare tail samples where the model fails and augments these samples with a controllable diffusion model during training. In our framework, generating scenarios that are diverse, realistic, and preserve tail-case characteristics is paramount. Accordingly, we design a tail-aware generation method that applies tailored diffusion guidance to generate trajectories that both capture rare behaviors and respect traffic rules. Unlike prior simulation methods focused solely on scenario diversity, GALTraj is the first to show how simulator-driven augmentation benefits long-tail learning in trajectory prediction. Experiments on multiple trajectory datasets (WOMD, Argoverse2) with popular backbones (QCNet, MTR) confirm that our method significantly boosts performance on tail samples and also enhances accuracy on head samples.

轨迹预测扩散模型长尾学习自动驾驶

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