arXiv:2503.09366cs.CV2025-03被引 7

提出Pioformer模型,显式建模驾驶场景中的后交互关系,提升轨迹预测精度。

Post-interactive Multimodal Trajectory Prediction for Autonomous Driving

  • 分阶段构建粗粒度到细粒度的轨迹预测框架,融合图与超图网络提取多阶交互特征。
  • 在Argoverse 1数据集上,最小ADE6、FDE6等指标均优于基线模型,最高降低14.4%误差。
  • 特别适合复杂交互场景下的自动驾驶系统,对行为不确定性建模有显著改进。

由于智能体行为具有固有的不确定性,建模自动驾驶中各智能体间的交互关系仍具挑战性。现有方法较少关注预测轨迹中所包含的后交互(post-interactions)信息。为此,本文提出一种粗到精的Transformer架构Pioformer,显式提取后交互特征以提升预测准确性。首先,通过基于图神经网络的粗轨迹网络,利用观测轨迹和车道线生成粗略轨迹,并提取低阶交互特征;其次,设计基于超图神经网络的轨迹提案网络,学习高阶交互特征;最后,将观测轨迹与轨迹提案拼接输入提案精炼网络,结合先前交互特征与轨迹一致性特征,学习后交互特征。此外,提出三阶段训练策略以促进学习过程。在Argoverse 1数据集上的大量实验表明,相比基线方法HiVT-64,本模型在minADE6、minFDE6、MR6、brier-minFDE6指标上分别降低4.4%、8.4%、14.4%、5.7%的预测误差。

原文摘要 · Abstract (English)

Modeling the interactions among agents for trajectory prediction of autonomous driving has been challenging due to the inherent uncertainty in agents' behavior. The interactions involved in the predicted trajectories of agents, also called post-interactions, have rarely been considered in trajectory prediction models. To this end, we propose a coarse-to-fine Transformer for multimodal trajectory prediction, i.e., Pioformer, which explicitly extracts the post-interaction features to enhance the prediction accuracy. Specifically, we first build a Coarse Trajectory Network to generate coarse trajectories based on the observed trajectories and lane segments, in which the low-order interaction features are extracted with the graph neural networks. Next, we build a hypergraph neural network-based Trajectory Proposal Network to generate trajectory proposals, where the high-order interaction features are learned by the hypergraphs. Finally, the trajectory proposals are sent to the Proposal Refinement Network for further refinement. The observed trajectories and trajectory proposals are concatenated together as the inputs of the Proposal Refinement Network, in which the post-interaction features are learned by combining the previous interaction features and trajectory consistency features. Moreover, we propose a three-stage training scheme to facilitate the learning process. Extensive experiments on the Argoverse 1 dataset demonstrate the superiority of our method. Compared with the baseline HiVT-64, our model has reduced the prediction errors by 4.4%, 8.4%, 14.4%, 5.7% regarding metrics minADE6, minFDE6, MR6, and brier-minFDE6, respectively.

轨迹预测自动驾驶交互建模超图网络

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。