Skip to content

WorldDreamer: Towards General World Models for Video Generation via Predicting Masked Tokens

License

Notifications You must be signed in to change notification settings

JeffWang987/WorldDreamer

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 

Repository files navigation

WorldDreamer: Towards General World Models for Video Generation via Predicting Masked Tokens

Abstract

World models play a crucial role in understanding and predicting the dynamics of the world, which is essential for video generation. However, existing world models are confined to specific scenarios such as gaming or driving, limiting their ability to capture the complexity of general world dynamic environments. Therefore, we introduce WorldDreamer, a pioneering world model to foster a comprehensive comprehension of general world physics and motions, which significantly enhances the capabilities of video generation. Drawing inspiration from the success of large language models, WorldDreamer frames world modeling as an unsupervised visual sequence modeling challenge. This is achieved by mapping visual inputs to discrete tokens and predicting the masked ones. During this process, we incorporate multi-modal prompts to facilitate interaction within the world model. Our experiments show that WorldDreamer excels in generating videos across different scenarios, including natural scenes and driving environments. WorldDreamer showcases versatility in executing tasks such as text-to-video conversion, imageto-video synthesis, and video editing. These results underscore WorldDreamer’s effectiveness in capturing dynamic elements within diverse general world environments.

abs

News

  • [2024/01/18] Repository Initialization.

WorldDreamer Framework method

Bibtex

If this work is helpful for your research, please consider citing the following BibTeX entry.

@article{wang2023worlddreamer,
      title={WorldDreamer: Towards General World Models for Video Generation via Predicting Masked Tokens}, 
      author={Xiaofeng Wang and Zheng Zhu and Guan Huang and Boyuan Wang and Xinze Chen and Jiwen Lu},
      journal={arXiv preprint arXiv:2401.09985},
      year={2024}
}

About

WorldDreamer: Towards General World Models for Video Generation via Predicting Masked Tokens

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published