Probabilistic Multi-modal Trajectory Prediction with Lane Attention for Autonomous Vehicles

Abstract

Trajectory prediction is crucial for autonomous vehicles. The planning system not only needs to know the current state of the surrounding objects but also their possible states in the future. As for vehicles, their trajectories are significantly influenced by the lane geometry and how to effectively use the lane information is of active interest. Most of the existing works use rasterized maps to explore road information, which does not distinguish different lanes. In this paper, we propose a novel instance-aware representation for lane representation. By integrating the lane features and trajectory features, a goal-oriented lane attention module is proposed to predict the future locations of the vehicle. We show that the proposed lane representation together with the lane attention module can be integrated into the widely used encoder-decoder framework to generate diverse predictions. Most importantly, each generated trajectory is associated with a probability to handle the uncertainty. Our method does not suffer from collapsing to one behavior modal and can cover diverse possibilities. Extensive experiments and ablation studies on the benchmark datasets corroborate the effectiveness of our proposed method. Notably, our proposed method ranks third place in the Argoverse motion forecasting competition at NeurIPS 2019.

IROS 2020. 3rd place solution for the Argoverse motion forecasting challenge at NeurIPS 2019. A variation of the method also won the 1st place in the Nuscenes prediction challenge 2020

References in corpus (8)

• PyTorch: An Imperative Style, High-Performance Deep Learning Library
• LSTM: A Search Space Odyssey
• Fast and Furious: Real Time End-to-End 3D Detection, Tracking and Motion Forecasting with a Single Convolutional Net
• IntentNet: Learning to Predict Intention from Raw Sensor Data
• Overcoming Limitations of Mixture Density Networks: A Sampling and Fitting Framework for Multimodal Future Prediction
• MultiPath: Multiple Probabilistic Anchor Trajectory Hypotheses for Behavior Prediction
• Multiple Futures Prediction
• StarNet: Pedestrian Trajectory Prediction using Deep Neural Network in Star Topology

Cited by in corpus (5)

• Trajectory Forecasts in Unknown Environments Conditioned on Grid-Based Plans
• THOMAS: Trajectory Heatmap Output with learned Multi-Agent Sampling
• Map-Adaptive Goal-Based Trajectory Prediction
• Multimodal Trajectory Prediction Conditioned on Lane-Graph Traversals
• PLOP: Probabilistic poLynomial Objects trajectory Planning for autonomous driving