Suraj Nair

I am a PhD candidate in Computer Science at Stanford University, where I work at the intersection of machine learning, robotics, and computer vision. My research focuses on enabling general-purpose robotic agents through large-scale data collection and reinforcement learning. I am co-advised by Professors Chelsea Finn and Silvio Savarese, and am funded by the National Science Foundation Graduate Fellowship.

I completed my Bachelors in Computer Science at the California Institute of Technology (Caltech), where I worked with Yisong Yue on multi-agent reinforcement learning. I have also spent time at Google Brain and GE.

Email  /  CV  /  Google Scholar  /  Twitter  /  LinkedIn

profile photo

The goal of my research is to enable robotic agents which can generalize effectively across tasks, objects, and environments by learning from large datasets. To that end my research focuses on methods for scalably collecting robotic data as well as offline reinforcement learning algorithms which can learn to solve many tasks with limited supervision.

Publications & Preprints
PontTuset Learning Generalizable Robotic Reward Functions from "In-The-Wild" Human Videos
Annie S. Chen, Suraj Nair, Chelsea Finn
In Submission, 2021
project page

We propose a technique for learning multi-task reward functions from a small amount of robot data and large amounts of in-the-wild human videos. By leveraging diverse human data, the learned reward function is able to generalize to new environments and tasks.

PontTuset Greedy Hierarchical Variational Autoencoders for Large-Scale Video Prediction
Bohan Wu, Suraj Nair, Roberto Martin-Martin, Li Fei-Fei*, Chelsea Finn*
IEEE Conference on Computer Vision and Pattern Recognition (CVPR) , 2021
project page

We propose a technique for video prediction which trains a hierarchy of action-conditioned VAEs in a greedy fashion, enabling efficient training of large video prediction models.

PontTuset Model-Based Visual Planning with Self-Supervised Functional Distances
Stephen Tian, Suraj Nair, Frederik Ebert, Sudeep Dasari, Benjamin Eysenbach, Chelsea Finn, Sergey Levine
International Conference on Learning Representations (ICLR), 2021 (Spotlight)
project page

We propose a method for offline model-based RL which learns a video prediction model and a Q function based distance metric, and uses them to accomplish visually specified goals.

PontTuset Batch Exploration with Examples for Scalable Robotic Reinforcement Learning
Annie S. Chen*, HyunJi Nam*, Suraj Nair*, Chelsea Finn
Robotics and Automation Letters (RA-L) and International Conference on Robotics and Automation (ICRA), 2021
project page / code

We propose a framework for leveraging weak human superivision to enable better robotic exploration. Using just a few minutes of human supervision, the robot collects high quality data while unsupervised, providing better data for downstream offline RL.

PontTuset Recovery RL: Safe Reinforcement Learning with Learned Recovery Zones
Brijen Thananjeyan*, Ashwin Balakrishna*, Suraj Nair, Michael Luo, Krishnan Srinivasan, Minho Hwang, Joseph E. Gonzalez, Julian Ibarz, Chelsea Finn, Ken Goldberg
Robotics and Automation Letters (RA-L) and International Conference on Robotics and Automation (ICRA), 2021
project page

An algorithm for safe reinforcement learning which utilizes a set of offline data to learn about constraints before policy learning and a pair of policies which seperate the often conflicting objectives of task directed exploration and constraint satisfaction to learn contact rich and visuomotor control tasks.

PontTuset Goal-Aware Prediction: Learning to Model what Matters
Suraj Nair, Silvio Savarese, Chelsea Finn
International Conference on Machine Learning (ICML) , 2020
project page / code

We explore learning visual dynamics models which are conditioned on goals, and learn to model only goal relevant quantities.

PontTuset Hierarchical Foresight: Self-Supervised Learning of Long-Horizon Tasks via Visual Subgoal Generation.
Suraj Nair, Chelsea Finn
International Conference on Learning Representations (ICLR), 2020
project page / code

We study how we can learn long horizon vision-based tasks in self-supervised settings. Our approach, hierarchical visual foresight, can optimize for a sequence of subgoals that break down the task into easy to complete subsegments.

PontTuset Time Reversal as Self-Supervision
Suraj Nair, Mohammad Babaeizadeh, Chelsea Finn, Sergey Levine, Vikash Kumar
International Conference on Robotics and Automation (ICRA) , 2020
project page / press

We propose a technique that uses time-reversal to learn goals and provide a high level plan to reach them. In particular, our approach explores outward from a set of goal states, "unsolving" a task, which then enables solving the task from new initializations at test time.

PontTuset Causal Induction from Visual Observations for Goal-Directed Tasks
Suraj Nair, Yuke Zhu, Silvio Savarese, Li Fei-Fei
Workshop on Causal Machine Learning NeurIPS, 2019
project page / code

We explore how to effectively predict causal graphs from a small set of visual observations, and how to encorporate the learned graphs into downstream goal conditioned policy learning.

PontTuset RoboNet: Large-Scale Multi-Robot Learning
Sudeep Dasari, Frederik Ebert, Stephen Tian, Suraj Nair, Bernadette Bucher, Karl Schmeckpeper, Siddharth Singh, Sergey Levine, Chelsea Finn
Conference on Robot Learning (CoRL) , 2019
project page / code / press

We collect a dataset of robotic experience across 4 institutions and 7 robots, and demonstrate that robot learning algorithms leveraging this data can adapt to new environments faster than training from scratch.

PontTuset Neural Task Graphs: Generalizing to Unseen Tasks from a Single Video Demonstration
De-An Huang*, Suraj Nair*, Danfei Xu*, Yuke Zhu, Animesh Garg, Li Fei-Fei, Silvio Savarese, Juan Carlos Niebles
IEEE Conference on Computer Vision and Pattern Recognition (CVPR) , 2019 (Oral)

NTG learns to produce a task graph from a single video demonstration of an unseen task, and leverages it for one-shot imitation learning.

PontTuset Neural Task Programming: Learning to Generalize Across Hierarchical Tasks
Danfei Xu*, Suraj Nair*, Yuke Zhu, Julian Gao, Animesh Garg, Li Fei-Fei, Silvio Savarese
International Conference on Robotics and Automation (ICRA) , 2018
project page / code / Two Minute Papers

Neural Task Programming (NTP) is a meta-learning framework that learns to generate robot-executable neural programs from task demonstration video.

PontTuset Reliable RealTime Seismic Signal/Noise Discrimination With Machine Learning
Men-Andrin Meier, Zachary E Ross, Anshul Ramachandran, Ashwin Balakrishna, Suraj Nair, Peter Kundzicz, Zefeng Li, Jennifer Andrews, Egill Hauksson, Yisong Yue.
Journal of Geo-Physical Research: Solid Earth, 2019

Efficient prediction of real local earthquake signals from impulsive signals for earthquake early warning (EEW) alerts.

PontTuset Annotated Reconstruction of 3D Spaces Using Drones
Suraj Nair, Anshul Ramachandran, Peter Kundzicz.
MIT Undergraduate Research in Technology Conference (URTC), 2017 (Best Paper Presentation)

Reconstruct 3D voxel representations of a scene with object labels from RGB images captured from a drone, and use it for exporatory motion planning

Teaching Assistant: Stanford CS 330 [2019, 2020], Deep Multi-Task and Meta Learning
Teaching Assistant: Caltech CS/EE 155 [2017] , Machine Learning/Data Mining
Teaching Assistant: Caltech CS 121 [2016], Introduction to Relational Databases
Website Template