论文标题
基于动作的对比度学习轨迹预测
Action-based Contrastive Learning for Trajectory Prediction
论文作者
论文摘要
轨迹预测是成功的人类机器人相互作用的必不可少的任务,例如在自动驾驶中。在这项工作中,我们解决了使用移动摄像机在第一人称视图设置中预测未来行人轨迹的问题。为此,我们提出了一种基于动作的新型对比学习损失,该损失利用行人行动信息来改善学习的轨迹嵌入。这一新损失背后的基本思想是,在特征空间中,执行相同行动的行人的轨迹比具有明显不同行动的行人的轨迹更接近彼此。换句话说,我们认为有关行人行动的行为信息会影响他们的未来轨迹。此外,我们为轨迹引入了一种新型的抽样策略,能够有效地增加负面和阳性对比样品。使用训练有素的条件变异自动编码器(CVAE)生成其他合成轨迹样品,该样品是为轨迹预测开发的几种模型的核心。结果表明,我们提出的对比框架采用了有关行人行为的上下文信息,即有效的行动,并学习了更好的轨迹表示。因此,将所提出的对比框架整合在轨迹预测模型中可以改善其结果,并在三个轨迹预测基准上胜过最先进的方法[31,32,26]。
Trajectory prediction is an essential task for successful human robot interaction, such as in autonomous driving. In this work, we address the problem of predicting future pedestrian trajectories in a first person view setting with a moving camera. To that end, we propose a novel action-based contrastive learning loss, that utilizes pedestrian action information to improve the learned trajectory embeddings. The fundamental idea behind this new loss is that trajectories of pedestrians performing the same action should be closer to each other in the feature space than the trajectories of pedestrians with significantly different actions. In other words, we argue that behavioral information about pedestrian action influences their future trajectory. Furthermore, we introduce a novel sampling strategy for trajectories that is able to effectively increase negative and positive contrastive samples. Additional synthetic trajectory samples are generated using a trained Conditional Variational Autoencoder (CVAE), which is at the core of several models developed for trajectory prediction. Results show that our proposed contrastive framework employs contextual information about pedestrian behavior, i.e. action, effectively, and it learns a better trajectory representation. Thus, integrating the proposed contrastive framework within a trajectory prediction model improves its results and outperforms state-of-the-art methods on three trajectory prediction benchmarks [31, 32, 26].