Applied Mathematics and Nonlinear Sciences
Journal license

Journal

Applied Mathematics and Nonlinear Sciences


Volume
& Issue

Volume 8, Issue 1


Published
on

June 30, 2023


Pages


DOI

Article

A trajectory prediction method based on graph attention mechanism

Check for updates


Authors

Hejun Zhou Affiliation:
Department of Electromechanical and Information Engineering, Changde Vocational Technical College, Changde, 415000, China.
, Ting Zhao Affiliation:
School of Communication and Information Engineering, Chongqing University of Posts and Telecommunications, Chongqing, 400065, China.
, Yang Fang Affiliation:
School of Computer Science and Technology, Chongqing University of Posts and Telecommunications, Chongqing, 400065, China.
and Qilie liu Affiliation:
School of Communication and Information Engineering, Chongqing University of Posts and Telecommunications, Chongqing, 400065, China.


Abstract

Vehicle trajectory prediction is one of the key technologies to realize autonomous driving, which provides an important guarantee for the safety of vehicles in the process of autonomous driving. In this paper, with this as the starting point, a graph convolutional neural network is introduced through a graph attention mechanism to obtain scene features by modeling the temporal Transformer model of surrounding information. Based on the temporal convolutional model to obtain scene features, new feature vectors are calculated by aggregating the weights for the features of nodes and neighboring nodes. Then the input feature dimensions are transformed into the weight matrix of the output feature dimensions, and the output feature vector corresponding to the attention coefficients is calculated by using weighted summation. Then the effect of multiple training of the model is evaluated by taking the mean value and defining its structural relationship. The experimental results show that the prediction error of the proposed method is significantly smaller than that of the comparison method in scenarios with speeds less than or equal to 5m/s and greater than 5m/s. The prediction error based on target detection is reduced by 58.95%, indicating that the proposed method is more consistent with the operation scenarios of autonomous driving.


Keywords

Trajectory prediction, Graph attention mechanism, Temporal Transformer model, Convolutional neural network, Feature vector, 68T05


Citation

Zhou, H., Zhao, T., Fang, Y., & liu, Q. (2023). A trajectory prediction method based on graph attention mechanism. Applied Mathematics and Nonlinear Sciences, 8(1). https://doi.org/10.2478/amns.2023.1.00481

Published by: Engineering Journals

Engineering Journals Logo