RoboVIP: Multi-View Video Generation with Visual Identity Prompting Augments Robot Manipulation
Abstract
The diversity, quantity, and quality of manipulation data are critical for training effective robot policies. However, due to hardware and setup constraints, collecting large-scale real-world manipulation data remains difficult to scale across diverse environments. Recent work uses text-prompt conditioned image diffusion models to augment manipulation data by altering the backgrounds and tabletop objects in the visual observations. However, these approaches often overlook the practical need for multi-view and temporally coherent observations required by state-of-the-art policy models. Further, text prompts alone cannot reliably specify the scene setup. To provide the diffusion model with explicit visual guidance, we introduce visual identity prompting, which supplies exemplar images as conditioning inputs to guide the generation of the desired scene setup. To this end, we also build a scalable pipeline to curate a visual identity pool from large robotics datasets. Using our augmented manipulation data to train downstream vision-language-action and visuomotor policy models yields consistent performance gains in both simulation and real-robot settings.
Real-World Robot Section
Vanilla
Original
Successful Case
RoboEngine
Successful Case
Cosmos-Transfer2.5
Successful Case
Our RoboVIP
Successful Case
Cluttered
Original
Failure Case
RoboEngine
Failure Case
Cosmos-Transfer2.5
Failure Case
Our RoboVIP
Successful Case
Video Generation Section
Droid Augmented by Our RoboVIP
Ground Truth
Cosmos-Transfer2.5
RoboEngine
Our RoboVIP
Case 1
Case 2
Case 3
Case 1
Case 2
Case 3
Case 1
Case 2
Case 3
Case 1
Case 2
Case 3
BridgeData V2 Augmented by Our RoboVIP
Our Case 1
Our Case 2
Our Case 3
Our Case 4
Real-World Robot Trajectories Augmented by Our RoboVIP
Our Case 1
(with 30 FPS)
Our Case 2
(with 30 FPS)
Our Case 3
(with 30 FPS)
Our Case 4
(with 30 FPS)
Simulation Section
Pi0 Roll-out in SimplerEnv with Our RoboVIP
Put Spoon on Tablecloth
Put Carrot on Plate
Stack Green Block on Yellow Block
Put Eggplant in Basket
Octo Roll-out in SimplerEnv with Our RoboVIP
Put Spoon on Tablecloth
Put Carrot on Plate
Stack Green Block on Yellow Block
Put Eggplant in Basket