预测过去:基于梯度的轨迹预测中分布偏移检测方法 / Forecasting the Past: Gradient-Based Distribution Shift Detection in Trajectory Prediction
1️⃣ 一句话总结
这篇论文提出了一种自监督方法,通过训练一个解码器来‘预测过去’轨迹的后半段,并利用其损失梯度的范数作为评分,从而在不影响原有预测模型性能的前提下,有效检测自动驾驶轨迹预测中可能出现的未知场景或数据分布变化,提升系统安全性。
Trajectory prediction models often fail in real-world automated driving due to distributional shifts between training and test conditions. Such distributional shifts, whether behavioural or environmental, pose a critical risk by causing the model to make incorrect forecasts in unfamiliar situations. We propose a self-supervised method that trains a decoder in a post-hoc fashion on the self-supervised task of forecasting the second half of observed trajectories from the first half. The L2 norm of the gradient of this forecasting loss with respect to the decoder's final layer defines a score to identify distribution shifts. Our approach, first, does not affect the trajectory prediction model, ensuring no interference with original prediction performance and second, demonstrates substantial improvements on distribution shift detection for trajectory prediction on the Shifts and Argoverse datasets. Moreover, we show that this method can also be used to early detect collisions of a deep Q-Network motion planner in the Highway simulator. Source code is available at this https URL.
预测过去:基于梯度的轨迹预测中分布偏移检测方法 / Forecasting the Past: Gradient-Based Distribution Shift Detection in Trajectory Prediction
这篇论文提出了一种自监督方法,通过训练一个解码器来‘预测过去’轨迹的后半段,并利用其损失梯度的范数作为评分,从而在不影响原有预测模型性能的前提下,有效检测自动驾驶轨迹预测中可能出现的未知场景或数据分布变化,提升系统安全性。
源自 arXiv: 2604.12425