PreTR: Spatio-Temporal Non-Autoregressive Trajectory Prediction Transformer

03/17/2022
by   Lina Achaji, et al.
0

Nowadays, our mobility systems are evolving into the era of intelligent vehicles that aim to improve road safety. Due to their vulnerability, pedestrians are the users who will benefit the most from these developments. However, predicting their trajectory is one of the most challenging concerns. Indeed, accurate prediction requires a good understanding of multi-agent interactions that can be complex. Learning the underlying spatial and temporal patterns caused by these interactions is even more of a competitive and open problem that many researchers are tackling. In this paper, we introduce a model called PRediction Transformer (PReTR) that extracts features from the multi-agent scenes by employing a factorized spatio-temporal attention module. It shows less computational needs than previously studied models with empirically better results. Besides, previous works in motion prediction suffer from the exposure bias problem caused by generating future sequences conditioned on model prediction samples rather than ground-truth samples. In order to go beyond the proposed solutions, we leverage encoder-decoder Transformer networks for parallel decoding a set of learned object queries. This non-autoregressive solution avoids the need for iterative conditioning and arguably decreases training and testing computational time. We evaluate our model on the ETH/UCY datasets, a publicly available benchmark for pedestrian trajectory prediction. Finally, we justify our usage of the parallel decoding technique by showing that the trajectory prediction task can be better solved as a non-autoregressive task.

READ FULL TEXT
research
05/18/2020

Spatio-Temporal Graph Transformer Networks for Pedestrian Trajectory Prediction

Understanding crowd motion dynamics is critical to real-world applicatio...
research
07/13/2020

Multitask Non-Autoregressive Model for Human Motion Prediction

Human motion prediction, which aims at predicting future human skeletons...
research
01/27/2021

Spatial-Channel Transformer Network for Trajectory Prediction on the Traffic Scenes

Predicting motion of surrounding agents is critical to real-world applic...
research
09/15/2022

STPOTR: Simultaneous Human Trajectory and Pose Prediction Using a Non-Autoregressive Transformer for Robot Following Ahead

In this paper, we develop a neural network model to predict future human...
research
06/30/2022

Multiclass-SGCN: Sparse Graph-based Trajectory Prediction with Agent Class Embedding

Trajectory prediction of road users in real-world scenarios is challengi...
research
03/03/2023

GETNext: Trajectory Flow Map Enhanced Transformer for Next POI Recommendation

Next POI recommendation intends to forecast users' immediate future move...
research
05/21/2019

Looking to Relations for Future Trajectory Forecast

Inferring relational behavior between road users as well as road users a...

Please sign up or login with your details

Forgot password? Click here to reset