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

Heng Yuan, Jun Zhang, Lei Zhang*, Zhiqiang Zhang, Zhenpo Wang

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

3 Citations (Scopus)
Plum Print visual indicator of research metrics
  • Citations
    • Citation Indexes: 3
  • Captures
    • Readers: 4
  • Mentions
    • News Mentions: 1
see details

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

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

Yuan, H., Zhang, J., Zhang, L., Zhang, Z., & Wang, Z. (2024). Vehicle Trajectory Prediction Based on Posterior Distributions Fitting and TCN-Transformer. IEEE Transactions on Transportation Electrification, 10(3), 7160-7173. https://doi.org/10.1109/TTE.2023.3344881