Cloning Deterministic Worlds: The Critical Role of Latent Geometry in Long-Horizon World Models
Abstract
We study long-horizon prediction in deterministic environments and show that the main bottleneck is often not the dynamics model itself, but the geometry of the learned latent representation. Many world models can generate plausible futures, but here we focus on a stricter setting where a given initial condition and action sequence should lead to one correct future. In this setting, rollout fidelity becomes a direct test of whether the model has actually captured the environment. We find that even strong predictive models fail when operating on poorly structured latent spaces. In contrast, when the same predictive pipeline is given true underlying states, long-horizon prediction becomes much easier. This isolates representation geometry as a central issue.
Main Contributions
- We identify representation geometry as a primary bottleneck for long-horizon fidelity in deterministic world modeling.
- We propose GRWM, a simple plug-in regularization method that improves latent structure without changing the dynamics backbone.
- We show that better latent geometry leads to more faithful rollouts across multiple world-modeling backbones.
Results
Our results suggest that representation quality is a first-order issue for long-horizon prediction in deterministic worlds. By improving the geometry of the latent space, we obtain more faithful rollouts and reduce the gap to an oracle model with access to true states.
Citation
@article{xia2025cloning,
title = {Cloning Deterministic Worlds: The Critical Role of Latent Geometry in Long-Horizon World Models},
author = {Zaishuo Xia and Yukuan Lu and Xinyi Li and Yifan Xu and Yubei Chen},
journal = {arXiv preprint arXiv:2510.26782},
year = {2025}
}