Generalization with World Models

This repository contains the official implementation for two papers:

  • Artem Zholus and Aleksandr I. Panov "Multitask Adaptation by Retrospective Exploration with Learned World Models", CoRR, abs/2110.13241, 2021. https://arxiv.org/abs/2110.13241
  • Artem Zholus, Yaroslav Ivchenkov, and Aleksandr Panov. "Factorized World Models for Learning Causal Relationships", In ICLR2022 Workshop on the Elements of Reasoning: Objects, Structure and Causality, 2022 https://openreview.net/pdf?id=BCGfDBOIcec

The description will be added soon.