← Search

Ayzaan Wahid

18 accepted papers

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

Robot Data Curation with Mutual Information Estimators

RSS 2025poster

The performance of imitation learning policies often hinges on the datasets with which they are trained. Consequently, investment in data collection for robotics has grown across both industrial and academic labs. However, despite the marked increase in the quantity of demonstrations collected, litt…

Cited by 3PDFScholar
2025

Self-Improving Embodied Foundation Models

NeurIPS 2025poster

Foundation models trained on web-scale data have revolutionized robotics, but their application to low-level control remains largely limited to behavioral cloning. Drawing inspiration from the success of the reinforcement learning stage in fine-tuning large language models, we propose a two-stage po…

Cited by 0SourceScholar
2024

ALOHA Unleashed: A Simple Recipe for Robot Dexterity

CoRL 2024poster

Recent work has shown promising results for learning end-to-end robot policies using imitation learning. In this work we address the question of how far can we push imitation learning for challenging dexterous manipulation tasks. We show that a simple recipe of large scale data collection on the ALO…

Cited by 83SourceScholar
2024

Learning to Learn Faster from Human Feedback with Language Model Predictive Control

RSS 2024poster

Large language models (LLMs) have been shown to exhibit a wide range of capabilities, such as writing robot code from language commands -- enabling non-experts to direct robot behaviors, modify them based on feedback, or compose them to perform new tasks. However, these capabilities (driven by in-co…

2024

Open X-Embodiment: Robotic Learning Datasets and RT-X Models : Open X-Embodiment Collaboration

ICRA 2024

Large, high-capacity models trained on diverse datasets have shown remarkable successes on efficiently tackling downstream applications. In domains from NLP to Computer Vision, this has led to a consolidation of pretrained models, with general pretrained backbones serving as a starting point for man

Cited by 910SourcecodeScholar
2024

Open X-Embodiment: Robotic Learning Datasets and RT-X Models : Open X-Embodiment Collaboration0

ICRA 2024poster

Large, high-capacity models trained on diverse datasets have shown remarkable successes on efficiently tackling downstream applications. In domains from NLP to Computer Vision, this has led to a consolidation of pretrained models, with general pretrained backbones serving as a starting point for man…

Cited by 259SourcecodeScholar
2024

PIVOT: Iterative Visual Prompting Elicits Actionable Knowledge for VLMs

ICML 2024poster

Vision language models (VLMs) have shown impressive capabilities across a variety of tasks, from logical reasoning to visual understanding. This opens the door to richer interaction with the world, for example robotic control. However, VLMs produce only textual outputs, while robotic control and oth…

Cited by 95SourcePDFScholar
2024

Vid2Robot: End-to-end Video-conditioned Policy Learning with Cross-Attention Transformers

RSS 2024poster

Large-scale multi-task robotic manipulation systems often rely on text to specify the task. In this work, we explore whether a robot can learn by observing humans. To do so, the robot must understand a person's intent and perform the inferred task despite differences in the embodiments and environme…

2023

PaLM-E: An Embodied Multimodal Language Model

ICML 2023poster

Large language models excel at a wide range of complex tasks. However, enabling general inference in the real world, e.g. for robotics problems, raises the challenge of grounding. We propose embodied language models to directly incorporate real-world continuous sensor modalities into language models…

Cited by 1902SourcePDFScholar
2023

RT-2: Vision-Language-Action Models Transfer Web Knowledge to Robotic Control

CoRL 2023poster

We study how vision-language models trained on Internet-scale data can be incorporated directly into end-to-end robotic control to boost generalization and enable emergent semantic reasoning. Our goal is to enable a single end-to-end trained model to both learn to map robot observations to actions a…

Cited by 1068SourceScholar
2023

Robotic Skill Acquisition via Instruction Augmentation with Vision-Language Models

RSS 2023poster

Robotic manipulation policies that follow natural language instructions are typically trained from corpora of robot-language data that were either collected with specific tasks in mind or expensively relabeled by humans with varied language descriptions in hindsight. Recently, large-scale pretrained…

Cited by 81SourcePDFScholar
2023

Visuomotor Control in Multi-Object Scenes Using Object-Aware Representations

ICRA 2023poster

Perceptual understanding of the scene and the relationship between its different components is important for successful completion of robotic tasks. Representation learning has been shown to be a powerful technique for this, but most of the current methodologies learn task specific representations t…

Cited by 20SourceScholar
2020

Learning Object-conditioned Exploration using Distributed Soft Actor Critic

CoRL 2020

Object navigation is defined as navigating to an object of a given label in a complex, unexplored environment. In its general form, this problem poses several challenges for Robotics: semantic exploration of unknown environments in search of an object and low-level control. In this work we study obj

Cited by 0SourcePDFScholar
2019

Long Range Neural Navigation Policies for the Real World

IROS 2019poster

Learned Neural Network based policies have shown promising results for robot navigation. However, most of these approaches fall short of being used on a real robot due to the extensive simulated training they require. These simulations lack the visuals and dynamics of the real world, which makes it…

Cited by 22SourceScholar
2019

Visual Representations for Semantic Target Driven Navigation

ICRA 2019poster

What is a good visual representation for navigation? We study this question in the context of semantic visual navigation, which is the problem of a robot finding its way through a previously unseen environment to a target object, e.g. go to the refrigerator. Instead of acquiring a metric semantic ma…

Cited by 256SourcecodeScholar