论文标题

符合场景的运动预测的椭圆损失

Ellipse Loss for Scene-Compliant Motion Prediction

论文作者

Cui, Henggang, Shajari, Hoda, Yalamanchi, Sai, Djuric, Nemanja

论文摘要

运动预测是自动驾驶技术的关键部分,负责推断自动驾驶汽车周围环境中交通行为者的未来行为。为了确保安全有效的操作,预测模型需要输出遵守地图约束的准确轨迹。在本文中,我们解决了这项任务,并提出了一种新颖的椭圆损失,使模型可以更好地理解场景合规性并预测更现实的轨迹。椭圆损失通过使用可区分的轨迹栅格器模块将输出轨迹投射到自上而下的地图框架中,直接以监督方式惩罚越野预测。此外,它考虑了参与者的维度和方向,为模型提供了更多直接的培训信号。我们将椭圆损失应用于最近提出的最新联合检测预测模型,以展示其收益。对大规模自动驾驶数据的评估强烈表明该方法允许更准确,更现实的轨迹预测。

Motion prediction is a critical part of self-driving technology, responsible for inferring future behavior of traffic actors in autonomous vehicle's surroundings. In order to ensure safe and efficient operations, prediction models need to output accurate trajectories that obey the map constraints. In this paper, we address this task and propose a novel ellipse loss that allows the models to better reason about scene compliance and predict more realistic trajectories. Ellipse loss penalizes off-road predictions directly in a supervised manner, by projecting the output trajectories into the top-down map frame using a differentiable trajectory rasterizer module. Moreover, it takes into account actor dimensions and orientation, providing more direct training signals to the model. We applied ellipse loss to a recently proposed state-of-the-art joint detection-prediction model to showcase its benefits. Evaluation on large-scale autonomous driving data strongly indicates that the method allows for more accurate and more realistic trajectory predictions.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源