arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2202.13772cs.AIcs.LGcs.RO
收藏

基于路径感知的HD图运动预测

Path-Aware Graph Attention for HD Maps in Motion Prediction

QCraft Inc(QCraft公司)

Fang Da, Yu Zhang

摘要:

自动驾驶运动预测的成功依赖于高清地图信息的整合。由于地图是自然的图形结构,近年来对用于编码HD地图的图神经网络(GNN)的研究正在蓬勃发展。然而,与直接部署GNN的许多其他应用程序不同,HD地图是异构图,其中顶点(车道)由各种性质的边缘(车道-车道交互关系)连接,并且大多数基于图的模型并没有设计用于理解各种边缘类型,这些边缘类型为预测智能体将如何行驶车道提供关键线索。为了克服这一挑战,我们提出了路径感知图注意力,一种新的注意力架构,通过解析形成连接它们的路径的边缘序列来推断两个顶点之间的注意力。我们的分析说明了所提出的注意力机制如何促进学习的教学问题,现有的图形网络,如GCN的斗争。通过改进地图编码,该模型在Argoverse运动预测数据集上超越了先前的最新技术水平,并在2021年Argoverse运动预测竞赛中获得第一名。

英文摘要:

The success of motion prediction for autonomous driving relies on integration of information from the HD maps. As maps are naturally graph-structured, investigation on graph neural networks (GNNs) for encoding HD maps is burgeoning in recent years. However, unlike many other applications where GNNs have been straightforwardly deployed, HD maps are heterogeneous graphs where vertices (lanes) are connected by edges (lane-lane interaction relationships) of various nature, and most graph-based models are not designed to understand the variety of edge types which provide crucial cues for predicting how the agents would travel the lanes. To overcome this challenge, we propose Path-Aware Graph Attention, a novel attention architecture that infers the attention between two vertices by parsing the sequence of edges forming the paths that connect them. Our analysis illustrates how the proposed attention mechanism can facilitate learning in a didactic problem where existing graph networks like GCN struggle. By improving map encoding, the proposed model surpasses previous state of the art on the Argoverse Motion Forecasting dataset, and won the first place in the 2021 Argoverse Motion Forecasting Competition.

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

↑