Vaibhav Saxena

I am a Ph.D. student in Machine Learning at the Georgia Institute of Technology advised by Prof. Danfei Xu. My research revolves around generative models for decision-making, memory, and computer vision, with the goal of generalizable robot learning from offline demonstrations!

 In 2019, I graduated with a Master of Science in Applied Computing from the Department of Computer Science, University of Toronto, where I was advised by Prof. Jimmy Ba and Prof. Frank Rudzicz, and worked closely with Danijar Hafner. My research involved learning state-space models for long-horizon planning, deep reinforcement learning, and multi-modal representation learning.

See more about my professional experience here.

News

Research  (representative works highlighted)

C3DM: Constrained-Context Conditional Diffusion Models for Imitation Learning

Vaibhav Saxena, Yotto Koga, Danfei Xu

Workshop on Deployable Robotics, CoRL 2023

arXiv

Query image

Pose estimation using i-σSRN

Generalizable Pose Estimation using Implicit Scene Representations

Vaibhav Saxena, Kamal Rahimi Malekshan, Linh Tran, Yotto Koga

ICRA, 2023

📄 Paper
🌐 Project Webpage
🖥️ Code
🔗 Blog

Guiding Exploration Towards Impactful Actions

Vaibhav Saxena, Jimmy Ba, Danijar Hafner

Deep Reinforcement Learning Workshop
NeurIPS, 2022

📄 Paper
🔗 Tweet

Clockwork Variational Autoencoders

Vaibhav Saxena, Jimmy Ba, Danijar Hafner

NeurIPS, 2021


📄 Paper
🖥️ Code
🌐 Project Webpage
🔗 Tweet

Dyna-AIL : Adversarial Imitation Learning by Planning

Vaibhav Saxena, Srinivasan Sivanandan, Pulkit Mathur

Beyond "Tabula Rasa" in Reinforcement Learning Workshop
ICLR, 2020

📄 Paper