Vehicle Trajectory Prediction Based on Posterior Distributions Fitting and TCN-Transformer

Research output: Contribution to journalArticlepeer-review

19 Citations (Scopus)

Abstract

Accurate and efficient prediction of future motions of surrounding vehicles plays a crucial role in navigating complex traffic scenarios for automated vehicles. To address the poor interpretability of predicted distribution boundaries and the low accuracy of distribution mean with the existing methods, this article presents a novel deep-learning-based approach for predicting future trajectory distributions of surrounding vehicles. The proposed method consists of two key parts: the posterior trajectory distribution module (PTDM) and the temporal convolutional network (TCN)-Transformer module. The PTDM module fits the posterior distribution for each recorded trajectory in the training set, while the TCN-Transformer module models the future trajectory distribution. The model training incorporates the boundaries of the posterior distribution derived from PTDM. To evaluate the proposed scheme, the NGSIM and inD datasets are used, and the results show an accuracy improvement of 8.4% compared with the state-of-the-art methods at a prediction horizon of 4 s. Furthermore, the proposed method exhibits high computational efficiency, with an average computational time of 0.05 ms and a model size of 1/6th of the GRU-based model.

Original languageEnglish
Pages (from-to)7160-7173
Number of pages14
JournalIEEE Transactions on Transportation Electrification
Volume10
Issue number3
DOIs
Publication statusPublished - 2024
Externally publishedYes

Keywords

  • Automated vehicles
  • planning-based posterior distribution fitting
  • temporal convolutional network (TCN)
  • vehicle trajectory prediction

Fingerprint

Dive into the research topics of 'Vehicle Trajectory Prediction Based on Posterior Distributions Fitting and TCN-Transformer'. Together they form a unique fingerprint.

Cite this