Log In Sign Up

CRAT-Pred: Vehicle Trajectory Prediction with Crystal Graph Convolutional Neural Networks and Multi-Head Self-Attention

by   Julian Schmidt, et al.

Predicting the motion of surrounding vehicles is essential for autonomous vehicles, as it governs their own motion plan. Current state-of-the-art vehicle prediction models heavily rely on map information. In reality, however, this information is not always available. We therefore propose CRAT-Pred, a multi-modal and non-rasterization-based trajectory prediction model, specifically designed to effectively model social interactions between vehicles, without relying on map information. CRAT-Pred applies a graph convolution method originating from the field of material science to vehicle prediction, allowing to efficiently leverage edge features, and combines it with multi-head self-attention. Compared to other map-free approaches, the model achieves state-of-the-art performance with a significantly lower number of model parameters. In addition to that, we quantitatively show that the self-attention mechanism is able to learn social interactions between vehicles, with the weights representing a measurable interaction score. The source code is publicly available.


Multi-Modal Trajectory Prediction of Surrounding Vehicles with Maneuver based LSTMs

To safely and efficiently navigate through complex traffic scenarios, au...

ReCoG: A Deep Learning Framework with Heterogeneous Graph for Interaction-Aware Trajectory Prediction

Predicting the future trajectory of surrounding vehicles is essential fo...

Multi-Head Attention with Joint Agent-Map Representation for Trajectory Prediction in Autonomous Driving

For autonomous vehicles to navigate in urban environment, the ability to...

Multi-Head Attention-based Probabilistic Vehicle Trajectory Prediction

This paper presents online-capable deep learning model for probabilistic...

SVG-Net: An SVG-based Trajectory Prediction Model

Anticipating motions of vehicles in a scene is an essential problem for ...