arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2503.09366cs.CV

面向自动驾驶的后交互多模态轨迹预测

Post-interactive Multimodal Trajectory Prediction for Autonomous Driving

College of Computer Science and Electronic Engineering, Hunan University(湖南大学计算机与通信工程学院);College of Mechanical and Vehicle Engineering, Hunan University(湖南大学机械与运载工程学院);Department of Computer Science, The University of Hong Kong(香港大学计算机科学系);Department of Civil Engineering, Monash University(莫纳什大学土木工程系)

Ziyi Huang, Yang Li, Dushuai Li, Yao Mu, Hongmao Qin, Nan Zheng

AI总结:

本文提出Pioformer,一种由粗到细的Transformer模型,通过显式提取后交互特征并结合三阶段训练,在Argoverse 1数据集上显著降低多模态轨迹预测误差。

摘要:

由于智能体行为固有的不确定性,对自动驾驶轨迹预测中的智能体间交互进行建模一直具有挑战性。预测轨迹中所涉及的交互,也称为后交互,在轨迹预测模型中很少被考虑。为此,我们提出了一种用于多模态轨迹预测的由粗到细的Transformer,即Pioformer,它显式提取后交互特征以提高预测精度。具体而言,我们首先构建一个粗轨迹网络,基于观测轨迹和车道段生成粗轨迹,其中利用图神经网络提取低阶交互特征。接下来,我们构建一个基于超图神经网络的轨迹提议网络来生成轨迹提议,其中通过超图学习高阶交互特征。最后,将轨迹提议送入提议细化网络进行进一步细化。观测轨迹和轨迹提议被拼接在一起作为提议细化网络的输入,其中通过结合先前的交互特征和轨迹一致性特征来学习后交互特征。此外,我们提出了一种三阶段训练方案以促进学习过程。在Argoverse 1数据集上的大量实验证明了我们方法的优越性。与基线HiVT-64相比,我们的模型在minADE6、minFDE6、MR6和brier-minFDE6指标上的预测误差分别降低了4.4%、8.4%、14.4%和5.7%。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑