TENET: Transformer Encoding Network for Effective Temporal Flow on Motion Prediction
Yuting Wang
Hangning Zhou
Zhigang Zhang
Chen Feng
H. Lin
Chaofei Gao
Yizhi Tang
Zhenting Zhao
Shiyu Zhang
Jie-Ru Guo
Xuefeng Wang
Ziyao Xu
Chi Zhang

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.
View on arXivComments on this paper