Apr, 2022

HDGT: 基于场景编码的多智能体轨迹预测的异质驱动图变压器

TL;DR针对自动驾驶中编码驾驶场景以进行轨迹预测的任务,本文提出了一种新颖的骨干网络,称为异构驾驶图变换器 (Heterogeneous Driving Graph Transformer, HDGT),将驾驶场景建模为具有不同类型节点和边的异构图,并在图神经网络 (GNN) 的聚合模块中采用分层的变换器结构来适应输入的异构性,实验结果表明,该方法在 INTERACTION Prediction Challenge 和 Waymo Open Motion Challenge 中取得了新的最优结果。