收藏切换
Trajectory Prediction Method Enhanced by Self-supervised Pretraining
收藏切换
PDF
Linhui Li, Yifan Fu, Ting Wang, Xuecheng Wang, Jing Lian
Automotive Engineering | 2024, 46(7) : 1219 - 1227
Less
收藏切换
Automotive Engineering | 2024, 46(7): 1219-1227
Trajectory Prediction Method Enhanced by Self-supervised Pretraining
Full
Linhui Li, Yifan Fu, Ting Wang, Xuecheng Wang, Jing Lian
Affiliations
  • School of Mechanical Engineering,Dalian University of Technology,Dalian 116024
Published: 2024-07-25 doi: 10.19562/j.chinasae.qcgc.2024.07.009
Outline
收藏切换

To address limitation in prediction accuracy and data utilization efficiency of supervised learning-based trajectory prediction models, a trajectory prediction model and a general self-supervised pretraining strategy are proposed. Firstly, a lightweight trajectory prediction model based on Transformer is established to extract temporal-spatial features while modeling interaction relationship. Secondly, three types of masks, namely motion information temporal mask, road information spatial mask, and interaction relationship mask, are designed for self-supervised pre-training tasks on the model to enhance the model's ability to extract general scene features. Finally, pretraining weights are used as initialization parameters for supervised learning fine-tuning in downstream tasks. Experimental results on the Argoverse2 Motion Forecasting dataset show that the model can effectively reconstruct traffic scenes in pretraining tasks. The introduction of self-supervised pretraining improves prediction accuracy and data utilization efficiency. Moreover, it exhibits universality for different prediction tasks, achieving a 3.3% and 3.7% improvement in the minFDE6 for single-agent and multi-agent trajectory prediction tasks, respectively.

autonomous driving  /  trajectory prediction  /  self-supervised pretraining
Linhui Li, Yifan Fu, Ting Wang, Xuecheng Wang, Jing Lian. Trajectory Prediction Method Enhanced by Self-supervised Pretraining[J]. Automotive Engineering, 2024 , 46 (7) : 1219 -1227 . DOI: 10.19562/j.chinasae.qcgc.2024.07.009
Year 2024 volume 46 Issue 7
PDF
276
117
Cite this Article
BibTeX
Article Info
doi: 10.19562/j.chinasae.qcgc.2024.07.009
  • Receive Date:2024-01-23
  • Online Date:2025-07-29
  • Published:2024-07-25
Article Data
Affiliations
History
  • Received:2024-01-23
  • Revised:2024-03-02
Funding
Affiliations
    School of Mechanical Engineering,Dalian University of Technology,Dalian 116024
References
Share
https://castjournals.cast.org.cn/joweb/qcygc/EN/10.19562/j.chinasae.qcgc.2024.07.009
Share to
QR

Scan QR to access full text

Cite this article
BibTeX
Citations
表12种不同金属材料的力学参数

Family
属数
Number of
genus
种数
Number of
species
占总种数比例
Percentage of
total species (%)

Genus
种数
Number of
species
占总种数比例
Percentage of total
species (%)
鹅膏菌科Amanitaceae 2 11 5.26 鹅膏菌属 Amanita 10 4.78
小菇科 Mycenaceae 2 12 5.74 丝盖伞属 Inocybe 5 2.39
多孔菌科 Polyporaceae 8 14 6.70 蜡蘑属 Laccaria 5 2.39
红菇科 Russulaceae 3 23 11.00 小皮伞属 Marasmius 6 2.87
小菇属 Mycena 11 5.26
光柄菇属 Pluteus 5 2.39
红菇属 Russula 17 8.13
栓菌属 Trametes 5 2.39
关闭全屏
  • BibTeX
  • EndNote
  • RefWorks
  • TxT