Goal-based Neural Physics Vehicle Trajectory Prediction Model
Rui Gan, Haotian Shi, Pei Li, Keshu Wu, Bocheng An, Linheng Li, Junyi Ma, Chengyuan Ma, Bin Ran
TL;DR
The paper presents the Goal-based Neural Physics Vehicle Trajectory Prediction Model (GNP), a two-stage framework that first infers multiple driving goals using a mode-aware transformer and then completes trajectories with a neural differential equations model embedded in a physics-based neural social force. This design balances long-term predictive accuracy with interpretability by exposing goal distributions and the forces shaping motion, including goal attraction and inter-vehicle/lane repulsion learned through neural networks. Empirical evaluation on NGSIM and HighD demonstrates state-of-the-art performance for long-horizon predictions, complemented by interpretable visualizations of forces and intention modes. Ablation studies confirm the importance of intention-mode modeling and the attraction-dominated trajectory completion, underscoring the value of combining neural learning with physics-informed dynamics for autonomous driving applications.
Abstract
Vehicle trajectory prediction plays a vital role in intelligent transportation systems and autonomous driving, as it significantly affects vehicle behavior planning and control, thereby influencing traffic safety and efficiency. Numerous studies have been conducted to predict short-term vehicle trajectories in the immediate future. However, long-term trajectory prediction remains a major challenge due to accumulated errors and uncertainties. Additionally, balancing accuracy with interpretability in the prediction is another challenging issue in predicting vehicle trajectory. To address these challenges, this paper proposes a Goal-based Neural Physics Vehicle Trajectory Prediction Model (GNP). The GNP model simplifies vehicle trajectory prediction into a two-stage process: determining the vehicle's goal and then choosing the appropriate trajectory to reach this goal. The GNP model contains two sub-modules to achieve this process. The first sub-module employs a multi-head attention mechanism to accurately predict goals. The second sub-module integrates a deep learning model with a physics-based social force model to progressively predict the complete trajectory using the generated goals. The GNP demonstrates state-of-the-art long-term prediction accuracy compared to four baseline models. We provide interpretable visualization results to highlight the multi-modality and inherent nature of our neural physics framework. Additionally, ablation studies are performed to validate the effectiveness of our key designs.
