论文标题

Get-Dipp:用于图形的变压器用于可区分的集成预测和计划

GET-DIPP: Graph-Embedded Transformer for Differentiable Integrated Prediction and Planning

论文作者

Sun, Jiawei, Yuan, Chengran, Sun, Shuo, Liu, Zhiyang, Goh, Terence, Wong, Anthony, Tee, Keng Peng, Ang Jr, Marcelo H.

论文摘要

准确地预测互动道路代理的未来轨迹,并因此规划具有社会符合性和人类的轨迹,因此对于自动驾驶汽车来说很重要。在本文中,我们提出了一个以计划为中心的预测神经网络,该网络将围绕代理的历史状态和将上下文信息映射为输入,并通过模仿学习来输出周围代理的共同多模式预测轨迹,以及对自动车辆的控制命令的顺序。在我们的网络体系结构中提出了沿时间轴的代理商交互模块,以更好地理解道路上所有其他智能代理之间的关系。为了结合地图的拓扑信息,采用动态图卷积神经网络(DGCNN)来处理道路网络拓扑。此外,通过提供准确的预测结果和初始计划命令,整个架构可以用作与计划(DIPP)方法进行可区分的集成运动预测的骨干。实验是在现实世界数据集上进行的,以证明与先前最新方法相比,我们提出的方法在计划和预测准确性方面的改进。

Accurately predicting interactive road agents' future trajectories and planning a socially compliant and human-like trajectory accordingly are important for autonomous vehicles. In this paper, we propose a planning-centric prediction neural network, which takes surrounding agents' historical states and map context information as input, and outputs the joint multi-modal prediction trajectories for surrounding agents, as well as a sequence of control commands for the ego vehicle by imitation learning. An agent-agent interaction module along the time axis is proposed in our network architecture to better comprehend the relationship among all the other intelligent agents on the road. To incorporate the map's topological information, a Dynamic Graph Convolutional Neural Network (DGCNN) is employed to process the road network topology. Besides, the whole architecture can serve as a backbone for the Differentiable Integrated motion Prediction with Planning (DIPP) method by providing accurate prediction results and initial planning commands. Experiments are conducted on real-world datasets to demonstrate the improvements made by our proposed method in both planning and prediction accuracy compared to the previous state-of-the-art methods.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源