Dreamweaver: Learning Compositional World Models from Pixels

Author: Junyeob Baek, Yi-Fu Wu, Gautam Singh, Sungjin Ahn

Published to ICLR 2025

[paper] [project] [openreview]

Overview

Hightlighted Demo

Installation

git clone https://github.com/ahn-ml/dreamweaver-release.git
cd dreamweaver-release
pip install -r requirements.txt

How to Train

bash train.sh # run this first for sanity-check and change the hyperparameters appropreately

Citation

@inproceedings{
        baek2025dreamweaver,
        title={Dreamweaver: Learning Compositional World Models from Pixels},
        author={Junyeob Baek and Yi-Fu Wu and Gautam Singh and Sungjin Ahn},
        booktitle={The Thirteenth International Conference on Learning Representations},
        year={2025},
        url={https://openreview.net/forum?id=e5mTvjXG9u}
        }

Acknowledgements

Some modules in this code are referenced by singhgautam/sysbinder. We thank the authors for their great works!