LGINet: Autonomous vehicle trajectory prediction using lane graphs and imitative learning policy

Trajectory prediction is a challenging problem in autonomous driving, due to changes in scene structure and multimodal distributions of future trajectories. This study proposes a novel trajectory prediction model (LGINet). The central innovation of LGINet is the Context-Aware Graph Attention (CAGA) layer, which achieves a balance between handling scene complexity and the ability to capture information, improving prediction performance. Unlike conventional attention mechanisms, CAGA uses Sparsemax to focus on key neighbors, reducing noise and enhancing computational efficiency. The multi-head attention mechanism enhances the ability of the model to capture interactions across both local and long-distance spatial scales. Experiments with the nuScenes dataset show that the proposed LGINet model outperforms existing prediction methods. Specifically, the MinADE 5 value of LGINet is 1.23, the MR 5,2 value is 0.49, and the MinFDE 5 value is 2.39, demonstrating its effectiveness in generating diverse and accurate future trajectories for autonomous driving applications. Qualitative comparisons confirm that LGINet generates diverse trajectories aligned with scene structures and agent motion, highlighting its potential for autonomous driving.