arrow
Return

Semantically controllable augmentations for generalizable robot learning

delete2024-11-01
delete0
delete
OA
AI
H
Homanga Bharadhwaj *
M
Mohit Sharma
S
Shuran Song
A
Abhishek Gupta
V
Vikash Kumar
DOI:10.1177/02783649241273686delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Generalization to unseen real-world scenarios for robot manipulation requires exposure to diverse datasets during training. However, collecting large real-world datasets is intractable due to high operational costs. For robot learning to generalize despite these challenges, it is essential to leverage sources of data or priors beyond the robot's direct experience. In this work, we posit that image-text generative models, which are pre-trained on large corpora of web-scraped data, can serve as such a data source. These generative models encompass a broad range of real-world scenarios beyond a robot's direct experience and can synthesize novel synthetic experiences that expose robotic agents to additional world priors aiding real-world generalization at no extra cost. In particular, our approach leverages pre-trained generative models as an effective tool for data augmentation. We propose a generative augmentation framework for semantically controllable augmentations and rapidly multiplying robot datasets while inducing rich variations that enable real-world generalization. Based on diverse augmentations of robot data, we show how scalable robot manipulation policies can be trained and deployed both in simulation and in unseen real-world environments such as kitchens and table-tops. By demonstrating the effectiveness of image-text generative models in diverse real-world robotic applications, our generative augmentation framework provides a scalable and efficient path for boosting generalization in robot learning at no extra human cost.
Keywords:
Generative models
data augmentation
robot learning

Journal

International Journal of Robotics Research cover
International Journal of Robotics Research
IF:
5
Papers:
2.4K
Citations:
1.5W

Organization

U
University of Washington
Scholars:
8.0W
Papers: 7.0W
Citations: 12.5W
S
Stanford University
Scholars:
9.6W
Papers: 8.2W
Citations: 17.0W
Cited Papers

Cited Papers

RLBench: The Robot Learning Benchmark & Learning Environment
err2020-04-01
err0
PREAI
errStephen James; Zicong Ma; David Rovick Arrojo; Andrew J. Davison
errShare
errSave
Learning Fine-Grained Bimanual Manipulation with Low-Cost Hardware
err2023-07-10
err0
errOAAI
errTony Zhao; Vikash Kumar; Sergey Levine; Chelsea Finn
errShare
errSave
AutoAugment: Learning Augmentation Strategies From Data
err2019-06-01
err0
PREAI
errEkin D. Cubuk; Barret Zoph; Dandelion Mane; Vijay Vasudevan; Quoc V. Le
errShare
errSave
RL-CycleGAN: Reinforcement Learning Aware Simulation-to-Real
err2020-06-01
err0
errOAAI
errKanishka Rao; Chris Harris; Alex Irpan; Sergey Levine; Julian Ibarz; Mohi Khansari
errShare
errSave
High-Resolution Image Synthesis with Latent Diffusion Models
err2022-06-01
err0
errOAAI
errRobin Rombach; Andreas Blattmann; Dominik Lorenz; Patrick Esser; Bjorn Ommer
errShare
errSave
errShare
errSave
errShare
errSave
Verbs in Action: Improving verb understanding in video-language models
err2023-10-01
err0
errOAAI
errLiliane Momeni; Mathilde Caron; Arsha Nagrani; Andrew Zisserman; Cordelia Schmid
errShare
errSave
Manipulator-Independent Representations for Visual Imitation
err2021-07-12
err0
errOAAI
errYuxiang Zhou; Yusuf Aytar; Konstantinos Bousmalis
errShare
errSave
researcher View more