Learning Abstract World Models with a Group-Structured Latent Space
Fuente:
arXiv
Saved in:
| Main Authors: | , , , , |
|---|---|
| Format: | Preprint |
| Published: |
2025
|
| Subjects: | |
| Online Access: | |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866918510549008384 |
|---|---|
| author | Delliaux, Thomas Vu, Nguyen-Khanh François-Lavet, Vincent van der Pol, Elise Rachelson, Emmanuel |
| author_facet | Delliaux, Thomas Vu, Nguyen-Khanh François-Lavet, Vincent van der Pol, Elise Rachelson, Emmanuel |
| contents | Learning meaningful abstract models of Markov Decision Processes (MDPs) is crucial for improving generalization from limited data. In this work, we show how geometric priors can be imposed on the low-dimensional representation manifold of a learned transition model. We incorporate known symmetric structures via appropriate choices of the latent space and the associated group actions, which encode prior knowledge about invariances in the environment. In addition, our framework allows the embedding of additional unstructured information alongside these symmetries. We show experimentally that this leads to better predictions of the latent transition model than fully unstructured approaches, as well as better learning on downstream RL tasks, in environments with rotational and translational features, including in first-person views of 3D environments. Additionally, our experiments show that this leads to simpler and more disentangled representations. The full code is available on GitHub to ensure reproducibility. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2506_01529 |
| institution | arXiv |
| publishDate | 2025 |
| record_format | arxiv |
| spellingShingle | Learning Abstract World Models with a Group-Structured Latent Space Delliaux, Thomas Vu, Nguyen-Khanh François-Lavet, Vincent van der Pol, Elise Rachelson, Emmanuel Machine Learning Learning meaningful abstract models of Markov Decision Processes (MDPs) is crucial for improving generalization from limited data. In this work, we show how geometric priors can be imposed on the low-dimensional representation manifold of a learned transition model. We incorporate known symmetric structures via appropriate choices of the latent space and the associated group actions, which encode prior knowledge about invariances in the environment. In addition, our framework allows the embedding of additional unstructured information alongside these symmetries. We show experimentally that this leads to better predictions of the latent transition model than fully unstructured approaches, as well as better learning on downstream RL tasks, in environments with rotational and translational features, including in first-person views of 3D environments. Additionally, our experiments show that this leads to simpler and more disentangled representations. The full code is available on GitHub to ensure reproducibility. |
| title | Learning Abstract World Models with a Group-Structured Latent Space |
| topic | Machine Learning |
| url | https://arxiv.org/abs/2506.01529 |