← Search

Annie Xie

21 accepted papers

2025

Bidirectional Decoding: Improving Action Chunking via Guided Test-Time Sampling

ICLR 2025poster

Predicting and executing a sequence of actions without intermediate replanning, known as action chunking, is increasingly used in robot learning from human demonstrations. Yet, its effects on the learned policy remain inconsistent: some studies find it crucial for achieving strong results, while oth…

2025

Exploiting Policy Idling for Dexterous Manipulation

IROS 2025

Learning based methods for dexterous manipulation have made notable progress in recent years, and they can now produce solutions to complex tasks. However, learned policies often still lack reliability and exhibit limited robustness to important factors of variation. One failure pattern that can be

Cited by 1SourceScholar
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
2024

DROID: A Large-Scale In-The-Wild Robot Manipulation Dataset

RSS 2024poster

The creation of large, diverse, high-quality robot manipulation datasets is an important stepping stone on the path toward more capable and robust robotic manipulation policies. However, creating such datasets is challenging: collecting robot manipulation data in diverse environments poses logistica…

Cited by 216SourcePDFScholar
2024

Decomposing the Generalization Gap in Imitation Learning for Visual Robotic Manipulation

ICRA 2024poster

What makes generalization hard for imitation learning in visual robotic manipulation? This question is difficult to approach at face value, but the environment from the perspective of a robot can often be decomposed into enumerable factors of variation, such as the lighting conditions or the placeme…

Cited by 59SourceScholar
2024

Efficient Data Collection for Robotic Manipulation via Compositional Generalization

RSS 2024poster

Data collection has become an increasingly important problem in robotic manipulation, yet there still lacks much understanding of how to effectively collect data to facilitate broad generalization. Recent works on large-scale robotic data collection typically vary many environmental factors of varia…

Cited by 15SourcePDFScholar
2024

Learning to Explore in POMDPs with Informational Rewards

ICML 2024poster

Standard exploration methods typically rely on random coverage of the state space or coverage-promoting exploration bonuses. However, in partially observed settings, the biggest exploration challenge is often posed by the need to discover information-gathering strategies---e.g., an agent that has to…

Cited by 3SourcePDFScholar
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
2023

Supervised Pretraining Can Learn In-Context Reinforcement Learning

NeurIPS 2023spotlight

Large transformer models trained on diverse datasets have shown a remarkable ability to learn in-context, achieving high few-shot performance on tasks they were not explicitly trained to solve. In this paper, we study the in-context learning capabilities of transformers in decision-making problems,…

Cited by 80SourcePDFScholar
2022

Robust Policy Learning over Multiple Uncertainty Sets

ICML 2022spotlight

Reinforcement learning (RL) agents need to be robust to variations in safety-critical environments. While system identification methods provide a way to infer the variation from online experience, they can fail in settings where fast identification is not possible. Another dominant approach is robus…

Cited by 18SourcePDFScholar
2022

When to Ask for Help: Proactive Interventions in Autonomous Reinforcement Learning

NeurIPS 2022accept

A long-term goal of reinforcement learning is to design agents that can autonomously interact and learn in the world. A critical challenge to such autonomy is the presence of irreversible states which require external assistance to recover from, such as when a robot arm has pushed an object off of a…

2021

Deep Reinforcement Learning amidst Continual Structured Non-Stationarity

ICML 2021spotlight

As humans, our goals and our environment are persistently changing throughout our lifetime based on our experiences, actions, and internal and external drives. In contrast, typical reinforcement learning problem set-ups consider decision processes that are stationary across episodes. Can we develop…

Cited by 47SourcePDFScholar
2020

Learning Latent Representations to Influence Multi-Agent Interaction

CoRL 2020

Seamlessly interacting with humans or robots is hard because these agents are non-stationary. They update their policy in response to the ego agent’s behavior, and the ego agent must anticipate these changes to co-adapt. Inspired by humans, we recognize that robots do not need to explicitly model ev

Cited by 0SourcePDFScholar
2020

Learning Predictive Models from Observation and Interaction

ECCV 2020poster

Learning predictive models from interaction with the world allows an agent, such as a robot, to learn about how the world works, and then use this learned model to plan coordinated sequences of actions to bring about desired outcomes. However, learning a model that captures the dynamics of complex s…

Cited by 65SourcePDFScholar
2019

Improvisation through Physical Understanding: Using Novel Objects As Tools with Visual Foresight

RSS 2019poster

Machine learning has enabled robots to perform complex tasks in narrowly-scoped settings, and to perform simple tasks with high generalization. However, learning a model that can both perform complex tasks and generalize to previously unseen objects and goals remains a significant challenge. We stud…

Cited by 105SourcePDFScholar
2018

One-Shot Imitation from Observing Humans via Domain-Adaptive Meta-Learning

RSS 2018poster

Humans and animals are capable of learning a new behavior by observing others perform the skill just once. We consider the problem of allowing a robot to do the same -- learning from a video of a human, even when there is domain shift in the perspective, environment, and embodiment between the robot…

Cited by 435SourcePDFScholar