针对交通轨迹长尾分布难题,提出自适应动量与解耦对比学习框架
AMD: Adaptive Momentum and Decoupled Contrastive Learning Framework for Robust Long-Tail Trajectory Prediction
- 融合无监督与有监督对比学习,增强对稀有轨迹的识别能力
- 在nuScenes和ETH/UCY数据集上长尾预测性能领先,整体准确率提升显著
- 适合关注自动驾驶轨迹预测与长尾分布问题的研究者
准确预测交通参与者未来轨迹是自动驾驶的关键。然而,自然数据集中轨迹分布存在固有不均衡,尾部数据往往代表更复杂且危险的场景。现有方法仅依赖基础模型的预测误差,未考虑长尾轨迹模式的多样性与不确定性。本文提出自适应动量与解耦对比学习框架(AMD),结合改进的动量对比学习(MoCo-DT)与解耦对比学习(DCL)模块,提升模型对稀有复杂轨迹的识别能力。设计四种轨迹随机增强方法,并引入在线迭代聚类策略,实现伪标签动态更新,更好适应长尾数据的分布偏移。提出三种定义长尾轨迹的标准,在nuScenes和ETH/UCY数据集上进行广泛对比实验。结果表明,AMD不仅在长尾轨迹预测中表现最优,整体预测精度也显著提升。
原文摘要 · Abstract (English)
Accurately predicting the future trajectories of traffic agents is essential in autonomous driving. However, due to the inherent imbalance in trajectory distributions, tail data in natural datasets often represents more complex and hazardous scenarios. Existing studies typically rely solely on a base model's prediction error, without considering the diversity and uncertainty of long-tail trajectory patterns. We propose an adaptive momentum and decoupled contrastive learning framework (AMD), which integrates unsupervised and supervised contrastive learning strategies. By leveraging an improved momentum contrast learning (MoCo-DT) and decoupled contrastive learning (DCL) module, our framework enhances the model's ability to recognize rare and complex trajectories. Additionally, we design four types of trajectory random augmentation methods and introduce an online iterative clustering strategy, allowing the model to dynamically update pseudo-labels and better adapt to the distributional shifts in long-tail data. We propose three different criteria to define long-tail trajectories and conduct extensive comparative experiments on the nuScenes and ETH$/$UCY datasets. The results show that AMD not only achieves optimal performance in long-tail trajectory prediction but also demonstrates outstanding overall prediction accuracy.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。