PRECOG: PREdiction Conditioned On Goals in Visual Multi-Agent Settings

Nicholas Rhinehart, Rowan McAllister, Kris M. Kitani and Sergey Levine
Conference Paper, (ICCV) International Conference on Computer Vision, October, 2019

Download Publication

Copyright notice: This material is presented to ensure timely dissemination of scholarly and technical work. Copyright and all rights therein are retained by authors or by other copyright holders. All persons copying this information are expected to adhere to the terms and constraints invoked by each author's copyright. These works may not be reposted without the explicit permission of the copyright holder.


For autonomous vehicles (AVs) to behave appropriately on roads populated by human-driven vehicles, they must be able to reason about the uncertain intentions and decisions of other drivers from rich perceptual information. Towards these capabilities, we present a probabilistic forecasting model of future interactions between a variable number of agents. We perform both standard forecasting and the novel task of conditional forecasting, which reasons about how all agents will likely respond to the goal of a controlled agent (here, the AV). We train models on real and simulated data to forecast vehicle trajectories given past positions and LIDAR. Our evaluation shows that our model is substantially more accurate in multi-agent driving scenarios compared to existing state-of-the-art. Beyond its general ability to perform conditional forecasting queries, we show that our model’s predictions of all agents improve when conditioned on knowledge of the AV’s goal, further illustrating its capability to model agent interactions.

author = {Nicholas Rhinehart, Rowan McAllister, Kris M. Kitani, Sergey Levine},
title = {PRECOG: PREdiction Conditioned On Goals in Visual Multi-Agent Settings},
booktitle = {(ICCV) International Conference on Computer Vision},
year = {2019},
month = {October},
keywords = {Trajectory Forecasting, Autonomous Vehicles, Imitation Learning, Planning},
} 2019-09-27T10:28:37-04:00