用目标引导自蒸馏,提升部分观测下的轨迹预测精度
Improving Partially Observed Trajectories Forecasting by Target-driven Self-Distillation
- 用预测的目标作为监督信号,实现单阶段自蒸馏
- 在多个数据集上显著提升部分观测场景的预测准确率
- 适合需要高鲁棒性轨迹预测的自动驾驶系统
准确预测交通参与者未来轨迹对保障自动驾驶安全至关重要。然而,部分观测轨迹会显著降低现有先进模型的性能。以往方法通常依赖知识蒸馏,将全观测轨迹特征迁移到部分观测场景,需分多阶段训练,成本高且可能造成性能下降。本文提出目标驱动自蒸馏方法(TSD),利用高精度预测目标引导模型在部分观测下的预测。通过自蒸馏机制,模型在单次端到端训练中同时学习全观测与部分观测轨迹的特征分布,增强对两类场景的预测能力。我们在多个数据集和主流模型上评估该方法,实验结果表明其在两种场景下均取得显著性能提升。代码与模型权重将公开以促进后续研究。
原文摘要 · Abstract (English)
Accurate prediction of future trajectories of traffic agents is essential for ensuring safe autonomous driving. However, partially observed trajectories can significantly degrade the performance of even state-of-the-art models. Previous approaches often rely on knowledge distillation to transfer features from fully observed trajectories to partially observed ones. This involves firstly training a fully observed model and then using a distillation process to create the final model. While effective, they require multi-stage training, making the training process very expensive. Moreover, knowledge distillation can lead to a performance degradation of the model. In this paper, we introduce a Target-drivenSelf-Distillation method (TSD) for motion forecasting. Our method leverages predicted accurate targets to guide the model in making predictions under partial observation conditions. By employing self-distillation, the model learns from the feature distributions of both fully observed and partially observed trajectories during a single end-to-end training process. This enhances the model's ability to predict motion accurately in both fully observed and partially observed scenarios. We evaluate our method on multiple datasets and state-of-the-art motion forecasting models. Extensive experimental results demonstrate that our approach achieves significant performance improvements in both settings. To facilitate further research, we will release our code and model checkpoints.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。