Learning Real-World Action-Video Dynamics with Heterogeneous Masked Autoregression

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Wang, Lirui, Zhao, Kevin, Liu, Chaoqi, Chen, Xinlei
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866910817316765696
author Wang, Lirui
Zhao, Kevin
Liu, Chaoqi
Chen, Xinlei
author_facet Wang, Lirui
Zhao, Kevin
Liu, Chaoqi
Chen, Xinlei
contents We propose Heterogeneous Masked Autoregression (HMA) for modeling action-video dynamics to generate high-quality data and evaluation in scaling robot learning. Building interactive video world models and policies for robotics is difficult due to the challenge of handling diverse settings while maintaining computational efficiency to run in real time. HMA uses heterogeneous pre-training from observations and action sequences across different robotic embodiments, domains, and tasks. HMA uses masked autoregression to generate quantized or soft tokens for video predictions. \ourshort achieves better visual fidelity and controllability than the previous robotic video generation models with 15 times faster speed in the real world. After post-training, this model can be used as a video simulator from low-level action inputs for evaluating policies and generating synthetic data. See this link https://liruiw.github.io/hma for more information.
format Preprint
id arxiv_https___arxiv_org_abs_2502_04296
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Learning Real-World Action-Video Dynamics with Heterogeneous Masked Autoregression
Wang, Lirui
Zhao, Kevin
Liu, Chaoqi
Chen, Xinlei
Robotics
Computer Vision and Pattern Recognition
Machine Learning
We propose Heterogeneous Masked Autoregression (HMA) for modeling action-video dynamics to generate high-quality data and evaluation in scaling robot learning. Building interactive video world models and policies for robotics is difficult due to the challenge of handling diverse settings while maintaining computational efficiency to run in real time. HMA uses heterogeneous pre-training from observations and action sequences across different robotic embodiments, domains, and tasks. HMA uses masked autoregression to generate quantized or soft tokens for video predictions. \ourshort achieves better visual fidelity and controllability than the previous robotic video generation models with 15 times faster speed in the real world. After post-training, this model can be used as a video simulator from low-level action inputs for evaluating policies and generating synthetic data. See this link https://liruiw.github.io/hma for more information.
title Learning Real-World Action-Video Dynamics with Heterogeneous Masked Autoregression
topic Robotics
Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2502.04296