AdaWorld: Learning Adaptable World Models with Latent Actions

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Gao, Shenyuan, Zhou, Siyuan, Du, Yilun, Zhang, Jun, Gan, Chuang
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915316814053376
author Gao, Shenyuan
Zhou, Siyuan
Du, Yilun
Zhang, Jun
Gan, Chuang
author_facet Gao, Shenyuan
Zhou, Siyuan
Du, Yilun
Zhang, Jun
Gan, Chuang
contents World models aim to learn action-controlled future prediction and have proven essential for the development of intelligent agents. However, most existing world models rely heavily on substantial action-labeled data and costly training, making it challenging to adapt to novel environments with heterogeneous actions through limited interactions. This limitation can hinder their applicability across broader domains. To overcome this limitation, we propose AdaWorld, an innovative world model learning approach that enables efficient adaptation. The key idea is to incorporate action information during the pretraining of world models. This is achieved by extracting latent actions from videos in a self-supervised manner, capturing the most critical transitions between frames. We then develop an autoregressive world model that conditions on these latent actions. This learning paradigm enables highly adaptable world models, facilitating efficient transfer and learning of new actions even with limited interactions and finetuning. Our comprehensive experiments across multiple environments demonstrate that AdaWorld achieves superior performance in both simulation quality and visual planning.
format Preprint
id arxiv_https___arxiv_org_abs_2503_18938
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle AdaWorld: Learning Adaptable World Models with Latent Actions
Gao, Shenyuan
Zhou, Siyuan
Du, Yilun
Zhang, Jun
Gan, Chuang
Artificial Intelligence
Computer Vision and Pattern Recognition
Machine Learning
Robotics
World models aim to learn action-controlled future prediction and have proven essential for the development of intelligent agents. However, most existing world models rely heavily on substantial action-labeled data and costly training, making it challenging to adapt to novel environments with heterogeneous actions through limited interactions. This limitation can hinder their applicability across broader domains. To overcome this limitation, we propose AdaWorld, an innovative world model learning approach that enables efficient adaptation. The key idea is to incorporate action information during the pretraining of world models. This is achieved by extracting latent actions from videos in a self-supervised manner, capturing the most critical transitions between frames. We then develop an autoregressive world model that conditions on these latent actions. This learning paradigm enables highly adaptable world models, facilitating efficient transfer and learning of new actions even with limited interactions and finetuning. Our comprehensive experiments across multiple environments demonstrate that AdaWorld achieves superior performance in both simulation quality and visual planning.
title AdaWorld: Learning Adaptable World Models with Latent Actions
topic Artificial Intelligence
Computer Vision and Pattern Recognition
Machine Learning
Robotics
url https://arxiv.org/abs/2503.18938