BiTraP: Bi-directional Pedestrian Trajectory Prediction with Multi-modal Goal Estimation

April 2021

BiTraP: Bi-directional Pedestrian Trajectory Prediction with Multi-modal Goal Estimation

Authors:

Yu Yao, Ella Atkins, M. Johnson-roberson, Ram Vasudevan, and Xiaoxiao Du

Abstract:

Pedestrian trajectory prediction is an essential task in robotic applications such as autonomous driving and robot navigation. State-of-the-art trajectory predictors use a conditional variational autoencoder (CVAE) with recurrent neural networks (RNNs) to encode observed trajectories and decode multi-modal future trajectories. This process can suffer from accumulated errors over long prediction horizons (>=2 seconds). This paper presents BiTraP, a goal-conditioned bi-directional multi-modal trajectory prediction method based on the CVAE. BiTraP estimates the goal (end-point) of trajectories and introduces a novel bi-directional decoder to improve longer-term trajectory prediction accuracy. Extensive experiments show that BiTraP generalizes to both first-person view (FPV) and bird's-eye view (BEV) scenarios and outperforms state-of-the-art results by ~10-50%. We also show that different choices of non-parametric versus parametric target models in the CVAE directly influence the predicted multi-modal trajectory distributions. These results provide guidance on trajectory predictor design for robotic applications such as collision avoidance and navigation systems. Our code is available at: https://github.com/umautobots/bidireaction-trajectory-prediction

Notes:

@article{Yao-2021-130106,
author = {Yu Yao And Ella Atkins And M. Johnson-roberson And Ram Vasudevan And Xiaoxiao Du},
title = {BiTraP: Bi-directional Pedestrian Trajectory Prediction with Multi-modal Goal Estimation},
journal = {Proceedings of IEEE Robotics and Automation Letters},
year = {2021},
month = {April},
volume = {6},
number = {2},
pages = {1463 - 1470},
}
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.