Self-Aware Trajectory Prediction for Safe Autonomous Driving

Wenbo Shao,Jun Li,Hong Wang
DOI: https://doi.org/10.48550/arXiv.2305.09147
IF: 3.7
2023-05-16
Robotics
Abstract:Trajectory prediction is one of the key components of the autonomous driving software stack. Accurate prediction for the future movement of surrounding traffic participants is an important prerequisite for ensuring the driving efficiency and safety of intelligent vehicles. Trajectory prediction algorithms based on artificial intelligence have been widely studied and applied in recent years and have achieved remarkable results. However, complex artificial intelligence models are uncertain and difficult to explain, so they may face unintended failures when applied in the real world. In this paper, a self-aware trajectory prediction method is proposed. By introducing a self-awareness module and a two-stage training process, the original trajectory prediction module's performance is estimated online, to facilitate the system to deal with the possible scenario of insufficient prediction function in time, and create conditions for the realization of safe and reliable autonomous driving. Comprehensive experiments and analysis are performed, and the proposed method performed well in terms of self-awareness, memory footprint, and real-time performance, showing that it may serve as a promising paradigm for safe autonomous driving.
What problem does this paper attempt to address?
The problem this paper attempts to address is that in autonomous driving systems, existing AI-based trajectory prediction models, despite achieving significant results in prediction performance, have uncertainties, are difficult to interpret, and may face unexpected failures in practical applications. Specifically, these models may lead to unacceptable decision errors when dealing with unknown long-tail scenarios, potentially causing severe traffic accidents. To solve this problem, the paper proposes a self-aware trajectory prediction method. By introducing a self-aware module and a two-stage training process, this method can estimate the performance of the original trajectory prediction module during online operation, thereby helping the system to timely respond to potential prediction deficiencies, creating conditions for safe and reliable autonomous driving. The main contributions include: 1. **Novel self-aware trajectory prediction framework**: A new, computationally efficient self-aware module is introduced, which can diagnose prediction results by obtaining real-time information from the intermediate and output layers of the original trajectory prediction module. 2. **New training and inference process**: A two-stage training process is proposed to optimize the self-aware module without affecting the performance of the original trajectory prediction module. The training of the self-aware module does not require modifying the parameters of the already trained trajectory prediction module, only obtaining part of its information, thus having strong flexibility and practicality. 3. **Comprehensive experiments and analysis**: By comparing with various baseline methods, the proposed method's superiority in self-awareness capability, memory usage, and real-time performance is demonstrated. Additionally, detailed ablation studies and quantitative and qualitative analyses provide in-depth exploration. In summary, this paper aims to improve the safety, reliability, and robustness of autonomous driving systems by endowing trajectory prediction models with self-awareness capabilities.