Joint Interaction and Trajectory Prediction for Autonomous Driving using Graph Neural Networks
arXiv:1912.07882
Abstract
In this work, we aim to predict the future motion of vehicles in a traffic scene by explicitly modeling their pairwise interactions. Specifically, we propose a graph neural network that jointly predicts the discrete interaction modes and 5-second future trajectories for all agents in the scene. Our model infers an interaction graph whose nodes are agents and whose edges capture the long-term interaction intents among the agents. In order to train the model to recognize known modes of interaction, we introduce an auto-labeling function to generate ground truth interaction labels. Using a large-scale real-world driving dataset, we demonstrate that jointly predicting the trajectories along with the explicit interaction types leads to significantly lower trajectory error than baseline methods. Finally, we show through simulation studies that the learned interaction modes are semantically meaningful.
Accepted in Machine Learning for Autonomous Driving NeurIPS 2019 Workshop
References in corpus (2)
Cited by in corpus (9)
- Choose Your Simulator Wisely: A Review on Open-source Simulators for Autonomous Driving
- Trajectory Prediction for Autonomous Driving: Progress, Limitations, and Future Directions
- Beyond RMSE: Do machine-learned models of road user interaction produce human-like behavior?
- Optimal Solving of Constrained Path-Planning Problems with Graph Convolutional Networks and Optimized Tree Search
- Grounded Relational Inference: Domain Knowledge Driven Explainable Autonomous Driving
- Physically Feasible Vehicle Trajectory Prediction
- Multi-modal anticipation of stochastic trajectories in a dynamic environment with Conditional Variational Autoencoders
- Disentangled Neural Relational Inference for Interpretable Motion Prediction
- Lifelong Vehicle Trajectory Prediction Framework Based on Generative Replay