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

Abstract

Forecasting the trajectories of neighbor vehicles is a crucial step for decision making and motion planning of autonomous vehicles. This paper proposes a graph-based spatial-temporal convolutional network (GSTCN) to predict future trajectory distributions of all neighbor vehicles using past trajectories. This network tackles the spatial interactions using a graph convolutional network (GCN), and captures the temporal features with a convolutional neural network (CNN). The spatial-temporal features are encoded and decoded by a gated recurrent unit (GRU) network to generate future trajectory distributions. Besides, we propose a weighted adjacency matrix to describe the intensities of mutual influence between vehicles, and the ablation study demonstrates the effectiveness of our proposed scheme. Our network is evaluated on two real-world freeway trajectory datasets: I-80 and US-101 in the Next Generation Simulation (NGSIM).Comparisons in three aspects, including prediction errors, model sizes, and inference speeds, show that our network can achieve state-of-the-art performance.

11 pages, 9 figures, 8 tables, submitted to IEEE Trans. Intelligent Transportation Systems

Cited by in corpus (10)

• Social Force Embedded Mixed Graph Convolutional Network for Multi-class Trajectory Prediction
• Trajectory Prediction for Autonomous Driving: Progress, Limitations, and Future Directions
• Heterogeneous Trajectory Forecasting via Risk and Scene Graph Learning
• A Novel Temporal Multi-Gate Mixture-of-Experts Approach for Vehicle Trajectory and Driving Intention Prediction
• Interaction-Aware Trajectory Planning for Autonomous Vehicles with Analytic Integration of Neural Networks into Model Predictive Control
• CSC-Unet: A Novel Convolutional Sparse Coding Strategy Based Neural Network for Semantic Segmentation
• FollowGen: A Scaled Noise Conditional Diffusion Model for Car-Following Trajectory Prediction
• XInsight: Revealing Model Insights for GNNs with Flow-based Explanations
• Trajectory Prediction for Autonomous Driving using Agent-Interaction Graph Embedding
• Driving Style Recognition at First Impression for Online Trajectory Prediction