Saved in:
Bibliographic Details
Main Authors: He, Zizhan, Daigle, Maxime, Bashivan, Pouya
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2505.13696
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915770747846656
author He, Zizhan
Daigle, Maxime
Bashivan, Pouya
author_facet He, Zizhan
Daigle, Maxime
Bashivan, Pouya
contents Many animals possess a remarkable capacity to rapidly construct flexible cognitive maps of their environments. These maps are crucial for ethologically relevant behaviors such as navigation, exploration, and planning. Existing computational models typically require long sequential trajectories to build accurate maps, but neuroscience evidence suggests maps can also arise from integrating disjoint experiences governed by consistent spatial rules. We introduce the Episodic Spatial World Model (ESWM), a novel framework that constructs spatial maps from sparse, disjoint episodic memories. Across environments of varying complexity, ESWM predicts unobserved transitions from minimal experience, and the geometry of its latent space aligns with that of the environment. Because it operates on episodic memories that can be independently stored and updated, ESWM is inherently adaptive, enabling rapid adjustment to environmental changes. Furthermore, we demonstrate that ESWM readily enables near-optimal strategies for exploring novel environments and navigating between arbitrary points, all without the need for additional training. Our work demonstrates how neuroscience-inspired principles of episodic memory can advance the development of more flexible and generalizable world models.
format Preprint
id arxiv_https___arxiv_org_abs_2505_13696
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Building spatial world models from sparse transitional episodic memories
He, Zizhan
Daigle, Maxime
Bashivan, Pouya
Artificial Intelligence
Many animals possess a remarkable capacity to rapidly construct flexible cognitive maps of their environments. These maps are crucial for ethologically relevant behaviors such as navigation, exploration, and planning. Existing computational models typically require long sequential trajectories to build accurate maps, but neuroscience evidence suggests maps can also arise from integrating disjoint experiences governed by consistent spatial rules. We introduce the Episodic Spatial World Model (ESWM), a novel framework that constructs spatial maps from sparse, disjoint episodic memories. Across environments of varying complexity, ESWM predicts unobserved transitions from minimal experience, and the geometry of its latent space aligns with that of the environment. Because it operates on episodic memories that can be independently stored and updated, ESWM is inherently adaptive, enabling rapid adjustment to environmental changes. Furthermore, we demonstrate that ESWM readily enables near-optimal strategies for exploring novel environments and navigating between arbitrary points, all without the need for additional training. Our work demonstrates how neuroscience-inspired principles of episodic memory can advance the development of more flexible and generalizable world models.
title Building spatial world models from sparse transitional episodic memories
topic Artificial Intelligence
url https://arxiv.org/abs/2505.13696