TY - JOUR
T1 - Vehicle Trajectory Prediction Based on Posterior Distributions Fitting and TCN-Transformer
AU - Yuan, Heng
AU - Zhang, Jun
AU - Zhang, Lei
AU - Zhang, Zhiqiang
AU - Wang, Zhenpo
N1 - Publisher Copyright:
© 2023 IEEE.
PY - 2024
Y1 - 2024
N2 - 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.
AB - 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.
KW - Automated vehicles
KW - planning-based posterior distribution fitting
KW - temporal convolutional network (TCN)
KW - vehicle trajectory prediction
UR - http://www.scopus.com/inward/record.url?scp=85181567064&partnerID=8YFLogxK
U2 - 10.1109/TTE.2023.3344881
DO - 10.1109/TTE.2023.3344881
M3 - Article
AN - SCOPUS:85181567064
SN - 2332-7782
VL - 10
SP - 7160
EP - 7173
JO - IEEE Transactions on Transportation Electrification
JF - IEEE Transactions on Transportation Electrification
IS - 3
ER -