3 个月前

多类-SGCN:基于稀疏图的轨迹预测方法及其代理类别嵌入

多类-SGCN:基于稀疏图的轨迹预测方法及其代理类别嵌入

摘要

在真实场景中对道路使用者的轨迹进行预测极具挑战性,因其运动模式具有随机性和复杂性。以往面向行人的研究在建模行人之间的复杂交互方面取得了成功,但在涉及其他类型道路使用者(如机动车、骑行者等)时,轨迹预测性能显著下降,原因在于这些方法忽略了使用者的类型信息。尽管少数近期研究尝试通过引入用户标签信息构建密集连接的图结构,但其仍面临冗余的空间交互和时间依赖性问题。为解决上述挑战,本文提出一种基于稀疏图卷积网络的多类别轨迹预测方法——Multiclass-SGCN。该方法综合考虑了速度信息与智能体类型标签,并引入一种新颖的交互掩码机制,能够根据智能体间的交互得分自适应地确定其空间与时间连接关系。在Stanford Drone Dataset上的实验结果表明,所提方法显著优于当前最先进的方法,能够生成更加真实且合理的轨迹预测结果。

代码仓库

carrotsniper/multiclass-sgcn
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
trajectory-prediction-on-sddMulticlass-SGCN (ours)
mADEK @4.8s: 14.36
mF DEK @4.8s: 25.99

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
多类-SGCN:基于稀疏图的轨迹预测方法及其代理类别嵌入 | 论文 | HyperAI超神经