Controlling the World by Sleight of Hand
Sruthi Sudhakar, Ruoshi Liu, Basile Van Hoorick, Carl Vondrick, Richard Zemel
TL;DR
The paper addresses enabling machines to predict object-state changes caused by actions by learning from unlabeled hand-object video data. It presents CosHand, a diffusion-based model finetuned from a pretrained image model and conditioned on the input image, a hand mask, and a target interaction mask to synthesize future scenes. The method demonstrates strong generalization to unseen objects, backgrounds, and even robot-arm embodiments, and can sample multiple futures to reflect uncertainty in forces and dynamics. The work suggests that leveraging hand-conditioned priors with large-scale video data can provide scalable, versatile world models for robotic planning, controllable image editing, and AR/VR applications.
Abstract
Humans naturally build mental models of object interactions and dynamics, allowing them to imagine how their surroundings will change if they take a certain action. While generative models today have shown impressive results on generating/editing images unconditionally or conditioned on text, current methods do not provide the ability to perform object manipulation conditioned on actions, an important tool for world modeling and action planning. Therefore, we propose to learn an action-conditional generative models by learning from unlabeled videos of human hands interacting with objects. The vast quantity of such data on the internet allows for efficient scaling which can enable high-performing action-conditional models. Given an image, and the shape/location of a desired hand interaction, CosHand, synthesizes an image of a future after the interaction has occurred. Experiments show that the resulting model can predict the effects of hand-object interactions well, with strong generalization particularly to translation, stretching, and squeezing interactions of unseen objects in unseen environments. Further, CosHand can be sampled many times to predict multiple possible effects, modeling the uncertainty of forces in the interaction/environment. Finally, method generalizes to different embodiments, including non-human hands, i.e. robot hands, suggesting that generative video models can be powerful models for robotics.
