Predicting the future motion of surrounding agents is essential for autonomous vehicles (AVs) to operate safely in dynamic, human-robot-mixed environments. Context information, such as road maps and surrounding agents' states, provides crucial geometric and semantic information for motion behavior prediction. To this end, recent works explore two-stage prediction frameworks where coarse trajectories are first proposed, and then used to select critical context information for trajectory refinement. However, they either incur a large amount of computation or bring limited improvement, if not both. In this paper, we introduce a novel scenario-adaptive refinement strategy, named SmartRefine, to refine prediction with minimal additional computation. Specifically, SmartRefine can comprehensively adapt refinement configurations based on each scenario's properties, and smartly chooses the number of refinement iterations by introducing a quality score to measure the prediction quality and remaining refinement potential of each scenario. SmartRefine is designed as a generic and flexible approach that can be seamlessly integrated into most state-of-the-art motion prediction models. Experiments on Argoverse (1 & 2) show that our method consistently improves the prediction accuracy of multiple state-of-the-art prediction models. Specifically, by adding SmartRefine to QCNet, we outperform all published ensemble-free works on the Argoverse 2 leaderboard (single agent track) at submission. Comprehensive studies are also conducted to ablate design choices and explore the mechanism behind multi-iteration refinement. Codes are available at https://github.com/opendilab/SmartRefine/
翻译:预测周围智能体的未来运动对于自动驾驶车辆在动态的人机混合环境中安全运行至关重要。上下文信息(如道路地图和周围智能体状态)为运动行为预测提供了关键的几何与语义信息。为此,近年来研究者探索了两阶段预测框架:先提出粗略轨迹,再利用该轨迹筛选关键上下文信息进行轨迹精炼。然而,这类方法要么计算量巨大,要么提升效果有限,甚至两者兼有。本文提出一种新颖的场景自适应精炼策略——SmartRefine,以最小额外计算量实现预测精炼。具体而言,SmartRefine能根据每个场景的特性全面调整精炼配置,并通过引入质量分数衡量各场景的预测质量与剩余精炼潜力,智能选择精炼迭代次数。SmartRefine被设计为通用且灵活的方法,可无缝集成至大多数现有最优运动预测模型中。在Argoverse(1和2)上的实验表明,该方法能持续提升多个最优预测模型的准确性。特别地,通过将SmartRefine应用于QCNet,我们在提交时超越了Argoverse 2排行榜(单智能体赛道)中所有已发表的未使用集成方法的模型。本文还进行了全面研究以消融设计选择并探索多轮迭代精炼的机制。代码已开源至https://github.com/opendilab/SmartRefine/