Ellipse Loss for Scene-Compliant Motion Prediction
arXiv:2011.03139
Abstract
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.
Henggang Cui and Hoda Shajari contributed equally to this work. Accepted for publication at IEEE International Conference on Robotics and Automation (ICRA) 2021
References in corpus (5)
- PyTorch: An Imperative Style, High-Performance Deep Learning Library
- Fast and Furious: Real Time End-to-End 3D Detection, Tracking and Motion Forecasting with a Single Convolutional Net
- IntentNet: Learning to Predict Intention from Raw Sensor Data
- MultiPath: Multiple Probabilistic Anchor Trajectory Hypotheses for Behavior Prediction
- Spatially-Aware Graph Neural Networks for Relational Behavior Forecasting from Sensor Data