Projects · PyTorch · Diffusion · Deep Learning · Autonomous Vehicle

Motion Prediction with Guided Diffusion

Researched and developed a classifier-free guidance-based latent diffusion model for autonomous vehicle motion forecasting using UNet and Transformer as backbones

2023.03.04 · 1 min read · by Zhenlin Wang · updated 2023-04-12

Introduction

We proposed a guided diffusion based method for Motion Forecasting task. The diffusion process uses the standard UNet architecture with 1D-convolution conditioned on past locations. We addressed the problem of a long-tailed data distribution using a max-norm scaling. Our model outperformed baseline methods in experiments using ArgoVerse 2 dataset.

Members

Andrew Shen, Zhenlin Wang, Yilong Qin

[Code][Report]

Results Prediction Plots