arrow
Return

Explainable multimodal trajectory prediction using attention models

delete2022-10-01
delete21
PRE
AI
K
Kunpeng Zhang
李
李丽 (Li Li) *
DOI:10.1016/j.trc.2022.103829delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Automated vehicles are expected to navigate complex urban environments safely along with several non-cooperating agents. Therefore, accurate trajectory prediction is crucial for safe nav-igation and receives much interest. However, most existing studies mainly focus on models' prediction accuracy rather than their explainability. In this paper, we propose a Multimodal Trajectory Prediction Transformer (MTPT) model to retrieve the influencing factors of prediction and help understand the intrinsic mechanism of prediction. Especially in the MTPT model, we use a modified Swin Transformer with multiple prediction heads to carry out multimodal trajectory prediction. Numerical experiments confirm the MTPT model can capture the most critical input factors with the help of the attention method and thus improve prediction accuracy. The proposed model obtains state-of-the-art results even with limited training data. Moreover, the identified input factors are also in agreement with the human driving experience. This agreement indicates that the proposed model appropriately learns how to predict.
Keywords:
Automated vehicles
Trajectory prediction
Transformer
Explainable AI

Journal

Transportation Research Part C-Emerging Technologies cover
Transportation Research Part C-Emerging Technologies
IF:
7.9
Papers:
4.9K
Citations:
3.2W

Organization

T
tsinghua university
Scholars:
11.9W
Papers: 10.0W
Citations: 137
Cited Papers

Cited Papers

A decade of tyrosine kinases: from gene discovery to therapeutics
err2003-07-01
err0
PREAI
errRolf J. Craven; Harry Lightfoot; William G. Cance
errShare
errSave
Guillain-Barré as an Autoimmune Disease
err2009-09-04
err0
PREAI
errYehuda Shoenfeld; Jacob George; James B. Peter
errShare
errSave
Safety and Efficacy of Rituximab: Experience of a Single Multiple Sclerosis Center
err2018-03-01
err0
PREAI
errBrett Alldredge; Allison Jordan; Jaime Imitola; Michael K. Racke
errShare
errSave
errShare
errSave
researcher View more