4 个月前

社会路径:利用生成对抗网络学习行人轨迹的多模态分布

社会路径:利用生成对抗网络学习行人轨迹的多模态分布

摘要

本文提出了一种新颖的方法,用于预测与其他行人互动的行人的运动。该方法利用生成对抗网络(Generative Adversarial Network, GAN)对场景中的任何代理进行合理的预测采样。由于GAN非常容易出现模式塌陷和模式丢失问题,我们展示了最近提出的Info-GAN在多模态行人轨迹预测中能够显著改善这些问题。此外,与一些先前的研究不同,我们在训练生成器时未使用L2损失,因为尽管L2损失可以加速收敛,但会导致严重的模式塌陷。通过在真实数据和合成数据上的实验,我们证明了所提出的方法能够生成更多样化的样本,并且保留预测分布的模式。特别是为了验证这一观点,我们设计了一个玩具示例轨迹数据集,可用于评估不同方法在保留预测分布模式方面的性能。

代码仓库

amiryanj/socialways
pytorch
GitHub 中提及

基准测试

基准方法指标
trajectory-prediction-on-eth-biwi-walkingSocial Ways
ADE-8/12: 0.39
trajectory-prediction-on-hotel-biwi-walkingSocial Ways
ADE-8/12: 0.39
trajectory-prediction-on-stanford-droneSocial-Ways
ADE (in world coordinates): 0.62
FDE (in world coordinates): 1.16

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
社会路径:利用生成对抗网络学习行人轨迹的多模态分布 | 论文 | HyperAI超神经