让轨迹预测的置信度更真实,提升自动驾驶安全规划可靠性
Rethinking Gaussian Trajectory Predictors: Calibrated Uncertainty for Safe Planning
- 用核密度估计校准置信度,强制符合卡方分布特性
- 在真实数据集上显著提升预测置信度可靠性,降低误判风险
- 适合需要可靠概率输出的自动驾驶规划系统使用
准确的轨迹预测对拥挤环境中自动驾驶的安全导航至关重要。尽管许多轨迹预测器输出高斯分布以表征未来行人位置的多模态分布,但其置信度的可靠性常被忽视。这种局限性会导致集成不确定性感知规划器时出现不安全或过度保守的决策。现有高斯轨迹预测器主要依赖负对数似然损失,易产生过自信或欠自信的分布,损害下游规划器的安全性。本文提出一种新损失函数,利用核密度估计估算置信度的经验分布,并通过显式匹配卡方分布来保证与高斯假设的一致性。同时在最终损失中加入均方误差项以确保均值预测准确。在真实世界轨迹数据集上的实验表明,该方法显著提升了多种先进高斯轨迹预测器的置信度可靠性。我们还验证了为规划器提供可靠概率信息(即校准后的置信度)对于复杂场景下无碰撞导航的重要性。为此,将使用本方法训练的高斯轨迹预测器与不确定性感知模型预测控制结合,在真实数据集提取的场景中实现了更优的规划性能。
原文摘要 · Abstract (English)
Accurate trajectory prediction is critical for safe autonomous navigation in crowded environments. While many trajectory predictors output Gaussian distributions to represent the multi-modal distribution over future pedestrian positions, the reliability of their confidence levels often remains unaddressed. This limitation can lead to unsafe or overly conservative motion planning when the predictor is integrated with an uncertainty-aware planner. Existing Gaussian trajectory predictors primarily rely on the Negative Log-Likelihood loss, which is prone to predict over- or under-confident distributions, and may compromise downstream planner safety. This paper introduces a novel loss function for calibrating prediction uncertainty which leverages Kernel Density Estimation to estimate the empirical distribution of confidence levels. The proposed formulation enforces consistency with the properties of a Gaussian assumption by explicitly matching the estimated empirical distribution to the Chi-squared distribution. To ensure accurate mean prediction, a Mean Squared Error term is also incorporated in the final loss formulation. Experimental results on real-world trajectory datasets show that our method significantly improves the reliability of confidence levels predicted by different State-Of-The-Art Gaussian trajectory predictors. We also demonstrate the importance of providing planners with reliable probabilistic insights (i.e. calibrated confidence levels) for collision-free navigation in complex scenarios. For this purpose, we integrate Gaussian trajectory predictors trained with our loss function with an uncertainty-aware Model Predictive Control on scenarios extracted from real-world datasets, achieving improved planning performance through calibrated confidence levels.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。