Freeway Trajectory Prediction Via SpatiotemporalTransformers

Jingtao Peng,Ronghui Zhang,Zhifa Yang,Shengru Chen,Cong Ye,Hao Wu,Mengran Li
DOI: https://doi.org/10.1145/3654823.3654871
2024-01-01
Abstract:For autonomous driving, accurate trajectory prediction is paramount, necessitating effective harnessing of spatiotemporal data. This study proposes an innovative Spatiotemporal Transformer-based model, enhancing trajectory prediction precision by leveraging a multi-head self-attention mechanism. This mechanism intricately captures both inter-vehicular interactions and temporal trajectory dependencies. The model, structured around an LSTM-based encoder-decoder framework, innovatively considers spatial interactions among observed and future trajectories during the decoding process. Evaluated on the NGSIM dataset, our model demonstrates superior predictive accuracy over existing methodologies, underscoring the efficacy of Spatiotemporal Transformers in complex dynamic environments.
What problem does this paper attempt to address?