3 个月前

基于异质图卷积循环网络的动态场景表征学习用于运动预测

基于异质图卷积循环网络的动态场景表征学习用于运动预测

摘要

由于动态场景中交互关系复杂且不断变化,运动预测在自动驾驶领域是一项极具挑战性的问题。现有大多数方法依赖静态道路图谱来表征场景,难以有效建模动态场景中不断演变的时空依赖关系。本文提出采用动态异构图来建模场景,联合编码包括车辆(智能体)与车道在内的多种场景要素、多类型交互关系及其随时间的变化过程。此外,我们设计了一种新型的异构图卷积循环网络,能够聚合多样化的交互信息并捕捉其动态演化过程,从而学习动态图中内在的时空依赖关系,获得对动态场景的有效表征。最后,结合运动预测解码器,模型能够预测出真实且具有多模态特性的智能体未来轨迹,在多个运动预测基准测试中均优于现有的最先进方法。

基准测试

基准方法指标
trajectory-prediction-on-argoverseHeteroGCN
MR (K=6): 0.12
brier-minFDE (K=6): 1.75
minADE (K=6): 0.79
minFDE (K=6): 1.16
trajectory-prediction-on-argoverse2HeteroGCN
MR (K=6): 0.18
brier-minFDE (K=6): 1.90
minADE (K=6): 0.69
minFDE (K=6): 1.34

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于异质图卷积循环网络的动态场景表征学习用于运动预测 | 论文 | HyperAI超神经