ZWM

Code for the paper "Zero-shot World Models Are Developmentally Efficient Learners"

Progress Update

We will release the code by May 2026, to enable readers to reproduce our results. The datasets used for training our BabyZWM model will also be made available.

Citation

If you use this code/model in your research, please cite it as follows:

@misc{aw2026zeroshotworldmodelsdevelopmentally,
      title={Zero-shot World Models Are Developmentally Efficient Learners}, 
      author={Khai Loong Aw and Klemen Kotar and Wanhee Lee and Seungwoo Kim and Khaled Jedoui and Rahul Venkatesh and Lilian Naing Chen and Michael C. Frank and Daniel L. K. Yamins},
      year={2026},
      eprint={2604.10333},
      archivePrefix={arXiv},
      primaryClass={cs.AI},
      url={https://arxiv.org/abs/2604.10333}, 
}