Dynamic Perception-Based Vehicle Trajectory Prediction Using a Memory-Enhanced Spatio-Temporal Graph Network

Zhiming Gui,Xin Wang,Wenzheng Li

DOI: https://doi.org/10.3390/ijgi13060172

IF: 3.4

2024-05-25

ISPRS International Journal of Geo-Information

Abstract:In the realm of intelligent transportation systems, accurately predicting vehicle trajectories is paramount for enhancing road safety and optimizing traffic flow management. Addressing the impacts of complex traffic environments and efficiently modeling the diverse behaviors of vehicles are the key challenges at present. To achieve precise prediction of vehicle trajectories, it is essential to fully consider the dynamic changes in traffic conditions and the long-term dependencies of time-series data. In response to these challenges, we propose the Memory-Enhanced Spatio-Temporal Graph Network (MESTGN), an innovative model that integrates a Spatio-Temporal Graph Convolutional Network (STGCN) with an attention-enhanced Long Short-Term Memory (LSTM)-based sequence to sequence (Seq2Seq) encoder–decoder structure. MESTGN utilizes STGCN to capture the complex spatial dependencies between vehicles and reflects the interactions within the traffic network through road traffic data and network topology, which significantly influences trajectory prediction. Additionally, the model focuses on historical vehicle trajectory data points using an attention-weighted mechanism under a traditional LSTM prediction architecture, calculating the importance of critical trajectory points. Finally, our experiments conducted on the urban traffic dataset ApolloSpace validate the effectiveness of our proposed model. We demonstrate that MESTGN shows a significant performance improvement in vehicle trajectory prediction compared with existing mainstream models, thereby confirming its increased prediction accuracy.

geography, physical,remote sensing,computer science, information systems

What problem does this paper attempt to address?

The paper attempts to address the problem of accurately predicting vehicle trajectories in complex traffic environments to improve road safety and optimize traffic flow management. Specifically, the main challenges of the study include: 1. **Impact of Complex Traffic Environments**: The need to effectively model the diverse behaviors of vehicles in complex traffic environments. 2. **Dynamic Changes and Long-Term Dependencies**: Fully considering the dynamic changes in traffic conditions and the long-term dependencies of time series data. To tackle these challenges, the authors propose an innovative model—Memory-Enhanced Spatio-Temporal Graph Network (MESTGN). This model combines the Spatio-Temporal Graph Convolutional Network (STGCN) with the Long Short-Term Memory (LSTM) network based on the attention mechanism, and performs trajectory prediction output under the sequence-to-sequence (Seq2Seq) framework. In this way, the model can effectively capture the complex spatial relationships between vehicles and fully consider the temporal continuity and long-term dependencies of the trajectories, thereby achieving higher prediction accuracy and reliability. Moreover, experimental results show that MESTGN outperforms existing mainstream models on the urban traffic dataset ApolloSpace, validating its effectiveness in the field of vehicle trajectory prediction.

Dynamic Perception-Based Vehicle Trajectory Prediction Using a Memory-Enhanced Spatio-Temporal Graph Network

STS-DGNN: Vehicle Trajectory Prediction via Dynamic Graph Neural Network With Spatial–Temporal Synchronization

Trajectory Prediction with Attention-Based Spatial–Temporal Graph Convolutional Networks for Autonomous Driving

GlobalInsight: an LSTM Based Model for Multi-Vehicle Trajectory Prediction.

Vehicle trajectory prediction based on cross-attention and multilevel spatio-temporal features

Group Vehicle Trajectory Prediction With Global Spatio-Temporal Graph

An Enhanced Vehicle Trajectory Prediction Model Leveraging LSTM and Social-Attention Mechanisms

Dynamic Spatial-Temporal Memory Augmentation Network for Traffic Prediction

Environment-Attention Network for Vehicle Trajectory Prediction

A multimodal vehicle trajectory prediction method with spatio-temporal feature fusion

Graph-Based Interaction-Aware Multimodal 2D Vehicle Trajectory Prediction using Diffusion Graph Convolutional Networks

DGInet: Dynamic Graph and Interaction-Aware Convolutional Network for Vehicle Trajectory Prediction.

MDSTF: a multi-dimensional spatio-temporal feature fusion trajectory prediction model for autonomous driving

Vehicle Interactive Dynamic Graph Neural Network Based Trajectory Prediction for Internet of Vehicles

NetTraj: A Network-based Vehicle Trajectory Prediction Model with Directional Representation and Spatiotemporal Attention Mechanisms

Graph-Based Spatial-Temporal Convolutional Network for Vehicle Trajectory Prediction in Autonomous Driving

MA2GCN: Multi Adjacency relationship Attention Graph Convolutional Networks for Traffic Prediction using Trajectory data

Network traffic prediction with Attention-based Spatial-Temporal Graph Network

MmSTCT: spatial–temporal convolution transformer network considering driving intention for multimodal vehicle trajectory prediction of highway

Spatial-temporal multi-factor fusion graph neural network for traffic prediction

Attention-aware Social Graph Transformer Networks for Stochastic Trajectory Prediction