The paper proposes a Graph Recurrent Attentive Neural Process (GRANP) model for vehicle trajectory prediction. GRANP consists of an encoder with deterministic and latent paths, and a decoder for prediction.
The key highlights are:
The encoder utilizes Graph Attention Networks (GAT) to capture the social interactions among traffic participants, LSTM to extract temporal features, and 1D convolutional layers to encode the contextual information.
The deterministic path in the encoder generates a deterministic representation for contextual information, while the latent path learns a latent distribution, enabling GRANP to quantify prediction uncertainty.
Experiments on the highD dataset show that GRANP outperforms state-of-the-art models in both prediction accuracy (RMSE) and uncertainty quantification (NLL).
A case study demonstrates GRANP's ability to accurately predict trajectories and visualize uncertainties in different driving scenarios, such as lane changing and going straight.
Sensitivity analysis indicates that GRANP's performance is not sensitive to the number of attention heads but improves with increased model complexity (hidden dimensions).
Overall, GRANP provides a robust and efficient solution for vehicle trajectory prediction, with the unique capability of directly quantifying and visualizing prediction uncertainties.
Egy másik nyelvre
a forrásanyagból
arxiv.org
Mélyebb kérdések