← Search

Deepali Jain

19 accepted papers

2025

Achieving Human Level Competitive Robot Table Tennis

ICRA 2025

Achieving human-level performance on real world tasks is a north star for the robotics community. We present the first learned robot agent that reaches amateur humanlevel performance in competitive table tennis. Table tennis is a physically demanding sport that takes humans years to master. We contr

Cited by 43SourceScholar
2025

Learning the RoPEs: Better 2D and 3D Position Encodings with STRING

ICML 2025spotlight

We introduce $\textbf{STRING}$: Separable Translationally Invariant Position Encodings. STRING extends Rotary Position Encodings, a recently proposed and widely used algorithm in large language models, via a unifying theoretical framework. Importantly, STRING still provides $\textbf{exact}$ translat…

Cited by 1SourcePDFScholar
2025

Linear Transformer Topological Masking with Graph Random Features

ICLR 2025poster

When training transformers on graph-structured data, incorporating information about the underlying topology is crucial for good performance. Topological masking, a type of relative position encoding, achieves this by upweighting or downweighting attention depending on the relationship between the q…

Cited by 1SourcePDFScholar
2024

Embodied AI with Two Arms: Zero-shot Learning, Safety and Modularity

IROS 2024poster

We present an embodied AI system which receives open-ended natural language instructions from a human, and controls two arms to collaboratively accomplish potentially long-horizon tasks over a large workspace. Our system is modular: it deploys state of the art Large Language Models for task planning…

Cited by 15SourceScholar
2024

Modeling the Real World with High-Density Visual Particle Dynamics

CoRL 2024poster

We present High-Density Visual Particle Dynamics (HD-VPD), a learned world model that can emulate the physical dynamics of real scenes by processing massive latent point clouds containing 100K+ particles. To enable efficiency at this scale, we introduce a novel family of Point Cloud Transformers (P…

Cited by 1SourceScholar
2024

SARA-RT: Scaling up Robotics Transformers with Self-Adaptive Robust Attention

ICRA 2024poster

We present Self-Adaptive Robust Attention for Robotics Transformers (SARA-RT): a new paradigm for addressing the emerging challenge of scaling up Robotics Transformers (RT) for on-robot deployment. SARA-RT relies on the new method of fine-tuning proposed by us, called up-training. It converts pre-tr…

Cited by 10SourceScholar
2024

Structured Unrestricted-Rank Matrices for Parameter Efficient Finetuning

NeurIPS 2024poster

Recent efforts to scale Transformer models have demonstrated rapid progress across a wide range of tasks (Wei at. al 2022). However, fine-tuning these models for downstream tasks is quite expensive due to their large parameter counts. Parameter-efficient fine-tuning (PEFT) approaches have emerged as…

2024

The Design of the Barkour Benchmark for Robot Agility

IROS 2024poster

In this paper, we describe the design of the Barkour benchmark for measuring robot agility in navigating complex environments. Despite the growing interest in developing agile robot locomotion skills, the field lacks systematic benchmarks to measure the performance of robotic control systems and har…

Cited by 1SourceScholar
2023

Efficient Graph Field Integrators Meet Point Clouds

ICML 2023poster

We present two new classes of algorithms for efficient field integration on graphs encoding point cloud data. The first class, $\mathrm{SeparatorFactorization}$ (SF), leverages the bounded genus of point cloud mesh graphs, while the second class, $\mathrm{RFDiffusion}$ (RFD), uses popular $\epsilon$…

2023

Mnemosyne: Learning to Train Transformers with Transformers

NeurIPS 2023poster

In this work, we propose a new class of learnable optimizers, called Mnemosyne. It is based on the novel spatio-temporal low-rank implicit attention Transformers that can learn to train entire neural network architectures, including other Transformers, without any task-specific optimizer tuning. We…

Cited by 8SourcePDFScholar
2023

Robotic Table Tennis: A Case Study into a High Speed Learning System

RSS 2023poster

We present a deep-dive into a real-world robotic learning system that, in previous work, was shown to be capable of hundreds of table tennis rallies with a human and has the ability to precisely return the ball to desired targets. This system puts together a highly optimized perception subsystem, a…

2022

Hybrid Random Features

ICLR 2022poster

We propose a new class of random feature methods for linearizing softmax and Gaussian kernels called hybrid random features (HRFs) that automatically adapt the quality of kernel estimation to provide most accurate approximation in the defined regions of interest. Special instantiations of HRFs lead…

2022

i-Sim2Real: Reinforcement Learning of Robotic Policies in Tight Human-Robot Interaction Loops

CoRL 2022oral

Sim-to-real transfer is a powerful paradigm for robotic reinforcement learning. The ability to train policies in simulation enables safe exploration and large-scale data collection quickly at low cost. However, prior works in sim-to-real transfer of robotic policies typically do not involve any huma…

Cited by 66SourceScholar
2021

Learning Agile Locomotion Skills with a Mentor

ICRA 2021poster

Developing agile behaviors for legged robots re-mains a challenging problem. While deep reinforcement learning is a promising approach, learning truly agile behaviors typically requires tedious reward shaping and careful curriculum design. We formulate agile locomotion as a multi-stage learning prob…

Cited by 22SourceScholar
2021

Reward Machines for Vision-Based Robotic Manipulation

ICRA 2021poster

Deep Q learning (DQN) has enabled robot agents to accomplish vision based tasks that seemed out of reach. Despite recent success stories, there are still several sources of computational complexity that challenge the performance of DQN. We place the focus on vision manipulation tasks, where the corr…

Cited by 32SourceScholar
2021

Visual-Locomotion: Learning to Walk on Complex Terrains with Vision

CoRL 2021poster

Vision is one of the most important perception modalities for legged robots to safely and efficiently navigate uneven terrains, such as stairs and stepping stones. However, training robots to effectively understand high-dimensional visual input for locomotion is a challenging problem. In this work,…

Cited by 86SourceScholar
2019

Provably Robust Blackbox Optimization for Reinforcement Learning

CoRL 2019

Interest in derivative-free optimization (DFO) and “evolutionary strategies” (ES) has recently surged in the Reinforcement Learning (RL) community, with growing evidence that they can match state of the art methods for policy optimization problems in Robotics. However, it is well known that DFO meth