Sumeet Batra

USC Viterbi School of Engineering. ssbatra[at]usc[dot]edu.

profile_pic.jpg

I am a 5th year PhD candidate at the University of Southern California advised by Prof. Gaurav Sukhatme in the Robotics Embedded Systems Lab (RESL). My research lies at the intersection of curiosity-driven exploration, perception, and decision-making for robots – focused on how embodied agents can explore, understand, and act in the world in a generalizable manner. I’m particularly interested in how data-driven models can achieve extrapolative generalization: enabling robots to generalize beyond their training data by learning world models and representations that support robust decision-making in novel environments. To this end, I’ve published work on Reinforcement Learning (RL), open-ended learning algorithms such as Quality Diversity (QD) for self-supervised exploration, Generative models such as Diffusion models for robotics, and self-supervised representation learning from RGB inputs for embodied agents.

Previously, I’ve had the privilege of working with the Autonomous Vehicles team at NVIDIA as a research scientist intern in 2022 and 2023 on Reinforcement Learning for autonomous driving and Diffusion models for scenario generation. Before that, I was an intern at the National Institute of Standards and Technology (NIST), where I worked on Generative Adversarial Networks (GANs).

news

Jul 15, 2024 Invited to a panel with The Walking Dead showrunner Glen Mazzara to discuss the influence of AI and robotics on pop culture.
May 01, 2023 Joined NVIDIA as a Research Scientist Intern in the Autonomous Vehicles team, working on Diffusion models for scenario generation.
May 10, 2022 Joined NVIDIA as a Research Scientist Intern in the Autonomous Vehicles team, working on Reinforcement Learning for autonomous driving.

selected publications

  1. arXiv
    vis_gen_robots.gif
    Zero-Shot Visual Generalization in Robot Manipulation
    Sumeet Batra and Gaurav Sukhatme
    arXiv preprint arXiv:2505.11719, 2025
  2. ICML
    alda.gif
    Zero-Shot Generalization of Vision-Based RL Without Data Augmentation
    Sumeet Batra and Gaurav S Sukhatme
    International Conference on Machine Learning, 2025
  3. ICLR
    ppga_optimized.gif
    Proximal Policy Gradient Arborescence for Quality Diversity Reinforcement Learning
    Sumeet Batra, Bryon Tjanaka, Matthew C Fontaine, and 3 more authors
    International Conference on Learning Representations (Spotlight, <10%), 2023
  4. NeurIPS
    policy_diff.png
    Generating Behaviorally Diverse Policies with Latent Diffusion Models
    Shashank Hegde*, Sumeet Batra*, KR Zentner, and 1 more author
    Advances in Neural Information Processing Systems, 2023
  5. GECCO
    qd_position.png
    Quality Diversity for Robot Learning: Limitations and Future Directions
    Sumeet Batra, Bryon Tjanaka, Stefanos Nikolaidis, and 1 more author
    Proceedings of the Genetic and Evolutionary Computation Conference Companion, 2024
  6. ICRA
    drones_2024.png
    Collision Avoidance and Navigation for a Quadrotor Swarm Using End-to-end Deep Reinforcement Learning
    Zhehui Huang, Zhaojing Yang, Rahul Krupani, and 3 more authors
    2024 IEEE International Conference on Robotics and Automation (ICRA), 2024
  7. CoRL
    swarm-rl-optim.gif
    Decentralized Control of Quadrotor Swarms with End-to-end Deep Reinforcement Learning
    Sumeet Batra*, Zhehui Huang*, Aleksei Petrenko*, and 3 more authors
    Conference on Robot Learning, 2022