TENET: Transformer Encoding Network for Effective Temporal Flow on Motion Prediction

Yuting Wang,Hangning Zhou,Zhigang Zhang,Chen Feng,Huadong Lin,Chaofei Gao,Yizhi Tang,Zhenting Zhao,Shiyu Zhang,Jie Guo,Xuefeng Wang,Ziyao Xu,Chi Zhang
DOI: https://doi.org/10.48550/arxiv.2207.00170
2022-01-01
Abstract: This technical report presents an effective method for motion prediction in autonomous driving. We develop a Transformer-based method for input encoding and trajectory prediction. Besides, we propose the Temporal Flow Header to enhance the trajectory encoding. In the end, an efficient K-means ensemble method is used. Using our Transformer network and ensemble method, we win the first place of Argoverse 2 Motion Forecasting Challenge with the state-of-the-art brier-minFDE score of 1.90.
What problem does this paper attempt to address?