Conditional Latent ODEs for Motion Prediction in Autonomous Driving

Khang Truong Giang,Yongjae Kim,Andrea Finazzi
2024-05-29
Abstract:This paper addresses imitation learning for motion prediction problem in autonomous driving, especially in multi-agent setting. Different from previous methods based on GAN, we present the conditional latent ordinary differential equation (cLODE) to leverage both the generative strength of conditional VAE and the continuous representation of neural ODE. Our network architecture is inspired from the Latent-ODE model. The experiment shows that our method outperform the baseline methods in the simulation of multi-agent driving and is very efficient in term of GPU memory consumption. Our code and docker image are publicly available: <a class="link-external link-https" href="https://github.com/TruongKhang/cLODE;" rel="external noopener nofollow">this https URL</a> <a class="link-external link-https" href="https://hub.docker.com/r/kim4375731/clode" rel="external noopener nofollow">this https URL</a>.
Robotics
What problem does this paper attempt to address?