BiTraP: Bi-Directional Pedestrian Trajectory Prediction With Multi-Modal Goal Estimation

Yu Yao,Ella Atkins,Matthew Johnson-Roberson,Ram Vasudevan,Xiaoxiao Du
DOI: https://doi.org/10.1109/lra.2021.3056339
IF: 5.2
2021-04-01
IEEE Robotics and Automation Letters
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 ($\geq\! 2$ seconds). This letter 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 $\sim\! 10\text{--}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.
robotics
What problem does this paper attempt to address?