Variational Causal Dynamics: Discovering Modular World Models from Interventions. (arXiv:2206.11131v1 [cs.LG])

Web: http://arxiv.org/abs/2206.11131

Latent world models allow agents to reason about complex environments with
high-dimensional observations. However, adapting to new environments and
effectively leveraging previous knowledge remain significant challenges. We
present variational causal dynamics (VCD), a structured world model that
exploits the invariance of causal mechanisms across environments to achieve
fast and modular adaptation. By causally factorising a transition model, VCD is
able to identify reusable components across different environments. This is
achieved by combining causal discovery and variational inference to learn a …

arxiv dynamics lg models