Trajectory prediction plays a crucial role in the autonomous driving stack by enabling autonomous vehicles to anticipate the motion of surrounding agents. Goal-based prediction models have gained traction in recent years for addressing the multimodal nature of future trajectories. Goal-based prediction models simplify multimodal prediction by first predicting 2D goal locations of agents and then predicting trajectories conditioned on each goal. However, a single 2D goal location serves as a weak inductive bias for predicting the whole trajectory, often leading to poor map compliance, i.e., part of the trajectory going off-road or breaking traffic rules. In this paper, we improve upon goal-based prediction by proposing the Path-based prediction (PBP) approach. PBP predicts a discrete probability distribution over reference paths in the HD map using the path features and predicts trajectories in the path-relative Frenet frame. We applied the PBP trajectory decoder on top of the HiVT scene encoder and report results on the Argoverse dataset. Our experiments show that PBP achieves competitive performance on the standard trajectory prediction metrics, while significantly outperforming state-of-the-art baselines in terms of map compliance.
翻译:轨迹预测在自动驾驶系统中扮演着关键角色,使自动驾驶车辆能够预判周围智能体的运动。近年来,基于目标的预测模型因能处理未来轨迹的多模态特性而受到广泛关注。这类模型通过先预测智能体的二维目标位置,再基于每个目标预测轨迹的方式简化多模态预测。然而,单一的二维目标位置作为预测整条轨迹的弱归纳偏置,往往导致地图合规性不足,例如部分轨迹偏离道路或违反交通规则。本文通过提出基于路径的预测(Path-based Prediction, PBP)方法对基于目标的预测进行改进。PBP利用路径特征预测高清地图中参考路径的离散概率分布,并在路径相关的Frenet坐标系中输出轨迹。我们将PBP轨迹解码器应用于HiVT场景编码器之上,并在Argoverse数据集上报告结果。实验表明,PBP在标准轨迹预测指标上达到竞争性性能,同时在地图合规性方面显著优于现有最优基线方法。