Trajectory Prediction with Linguistic Representations

TitleTrajectory Prediction with Linguistic Representations
Publication TypeConference Paper
Year of Publication2022
AuthorsKuo, Y-L, Huang, X, Barbu, A, McGill, SG, Katz, B, Leonard, JJ, Rosman, G
Conference Name2022 IEEE International Conference on Robotics and Automation (ICRA)
Conference LocationPhiladelphia, PA, USA
Abstract

Language allows humans to build mental models that interpret what is happening around them resulting in more accurate long-term predictions. We present a novel trajectory prediction model that uses linguistic intermediate representations to forecast trajectories, and is trained using trajectory samples with partially-annotated captions. The model learns the meaning of each of the words without direct per-word supervision. At inference time, it generates a linguistic description of trajectories which captures maneuvers and interactions over an extended time interval. This generated description is used to refine predictions of the trajectories of multiple agents. We train and validate our model on the Argoverse dataset, and demonstrate improved accuracy results in trajectory prediction. In addition, our model is more interpretable: it presents part of its reasoning in plain language as captions, which can aid model development and can aid in building confidence in the model before deploying it.

URLhttps://ieeexplore.ieee.org/document/9811928
DOI10.1109/ICRA46639.2022.9811928

Associated Module: 

CBMM Relationship: 

  • CBMM Funded