Papers › Predictive Statistics Shape Emergent World Representations of Grid Walkers

Predictive Statistics Shape Emergent World Representations of Grid Walkers

17 Mar 2026arXiv:2603.16689added by Syntology

Sasha Brenner, Thomas R. Knösche, Nico Scherf

Title, abstract, authors and date from arXiv's metadata (CC0); this paper is not in the Papers with Code archive (frozen 2025-07-28).

Next-token predictors often appear to develop internal representations of the latent world and its rules. The probabilistic nature of these models suggests a deep connection between the structure of the world and the geometry of probability distributions. In order to understand this link more precisely, we use a minimal stochastic process as a controlled setting: constrained random walks on a two-dimensional lattice that must reach a fixed endpoint after a predetermined number of steps. Optimal prediction of this process solely depends on a sufficient vector determined by the walker's position relative to the target and the remaining time horizon; in other words, the probability distributions are parametrized by the world's grid geometry. We train decoder-only transformers and recurrent networks on prefixes sampled from the exact distribution of these walks and compare their hidden activations to sufficient statistics of prediction, by measuring alignment and linear readability across layers. We find that the transformer's computation factors into two stages: the first attention block extracts the sufficient statistic from the input, and later layers transform it into the next-step predictive geometry. Across constraint variants the post-attention representation is universal: a shared world-state of the lattice that can be read directly as a world model, traced to the predictive geometry of the data. Later layers then specialize it to each variant's next-step distribution. Recurrent networks reach the same Bayes-optimal loss but do not isolate this world-state as a separate stage, showing that the world-model geometry also depends on architecture. Although demonstrated in a toy system, the results suggest that the geometry of the predictive distribution is a useful lens on how neural networks internalize the structure of their data.

PaperPDFCode Syntology ran

In Syntology Open this paper in Syntology's Atlas, the map of the papers in Syntology's graph and their citations.

For agents, Syntology's MCP tool lists every function and class Syntology harvested from this paper and whether it ran (how to connect): get_harvested_code_for_paper(arxiv_id="2603.16689")

Code

Syntology Ran 0 of 20 code samples harvested from 1 repository linked to this paper; 20 have no recorded run.

By repository: found in paper text by Syntology: 20 samples from 1 repository, 0 ran. The run record, sample by sample. “Ran” means executed on a synthesized input, not that the code is correct or reproduces the paper.

allshaks/grid-walkers found in paper text by Syntology report

Reachability, where shown, is from one Syntology probe window (2026-09-16 to 2026-09-18); repositories not probed show nothing. GitHub stars are not tracked.

Code Syntology ran Syntology

20 samples harvested; 0 ran; 0 honoured the contract we drafted; 20 have no recorded run. Read from Syntology's graph 2026-09-24; that is when this build read the record, not when the samples ran.

20unverified

Licence: 0 of the 20 samples are pointer only, meaning Syntology does not serve that copy's text. This page shows no code text for any sample; each one links to its file in the repository.

Harvested from allshaks/grid-walkers. “Ran” means the sample executed on a synthesized input. It does not mean the output is correct, and nothing here reproduces the paper's results. “Honoured” and “violated” refer to a contract Syntology drafted from the code itself; “our draft was wrong” and “fixture could not drive it” are failures of Syntology's instrument, not of the code.

Each sample ends with its code_sha256, Syntology's identity for that exact code. An agent fetches the stored sample with Syntology's MCP tool get_code(code_sha256="…") (how to connect); click an identity to copy that call.

Repository labels, per sample. official repository: The archive marks this repository official for the paper. named in the paper: The archive records that the paper mentions this repository; it is not marked official. community (archive-listed): In the archive's code links for this paper, not marked official and not recorded as mentioned in the paper. found in paper text by Syntology: Syntology found this repository in the paper's own text; whether it is the authors' implementation is not asserted. community: Not in the archive's code links for this paper; a community repository Syntology harvested. Samples from a repository marked official are listed first. Licence labels name the repository's licence as recorded at harvest. “Pointer only” means Syntology does not serve that copy's text, for one of four reasons: no licence file was found; the licence was not identified; the licence is recorded as permissive but that copy's record is not marked cleared; or the licence is outside the permissive list Syntology serves text under (MIT, Apache-2.0, BSD and similar). Some licences outside that list permit redistribution, such as WTFPL, and GPL-3.0 under its conditions; they are simply not on the list. Hover a licence label for the reason. File links open the file on GitHub at the default branch, which may have changed since the harvest.

align_pca_signs_to_reference allshaks/grid-walkers/lattice_walk_repro/analysis/geometry.py found in paper text by Syntology unverified MIT (permissive) · 0bb79bf086db6bff · report
build_features allshaks/grid-walkers/lattice_walk_repro/analysis/features.py found in paper text by Syntology unverified MIT (permissive) · 6333e683d4df57fb · report
build_recurrent_model allshaks/grid-walkers/lattice_walk_repro/rnn_models.py found in paper text by Syntology unverified MIT (permissive) · 906417f366c52e3a · report
build_state_ids allshaks/grid-walkers/lattice_walk_repro/analysis/features.py found in paper text by Syntology unverified MIT (permissive) · 493ca622d0af3180 · report
centered_logits allshaks/grid-walkers/lattice_walk_repro/analysis/features.py found in paper text by Syntology unverified MIT (permissive) · 10f796c0ef4bb599 · report
ckpt_step allshaks/grid-walkers/lattice_walk_repro/analysis/model_loading.py found in paper text by Syntology unverified MIT (permissive) · c9c70f5d632fe199 · report
compute_walk_displacements allshaks/grid-walkers/lattice_walk_repro/walk_engine/compute_walk_grid.py found in paper text by Syntology unverified MIT (permissive) · 1b3c9e4f937bf7f3 · report
equiv_weights_disp_len allshaks/grid-walkers/lattice_walk_repro/analysis/lattice_metrics.py found in paper text by Syntology unverified MIT (permissive) · a2b9ecac75cd2778 · report
find_final_checkpoint allshaks/grid-walkers/lattice_walk_repro/analysis/model_loading.py found in paper text by Syntology unverified MIT (permissive) · 75a219d5c0cf7bfa · report
fit_and_evaluate allshaks/grid-walkers/lattice_walk_repro/analysis/lattice_metrics.py found in paper text by Syntology unverified MIT (permissive) · 28cbaf3749c57845 · report
gt_cache_key allshaks/grid-walkers/lattice_walk_repro/analysis/ground_truth.py found in paper text by Syntology unverified MIT (permissive) · c3387a5ce2a33bbd · report
infer_model_type allshaks/grid-walkers/lattice_walk_repro/training_core.py found in paper text by Syntology unverified MIT (permissive) · 898cee532c619b78 · report
joint_ols allshaks/grid-walkers/lattice_walk_repro/analysis/r2.py found in paper text by Syntology unverified MIT (permissive) · 3a390bc8cbaff1d6 · report
linear_cka allshaks/grid-walkers/lattice_walk_repro/analysis/lattice_metrics.py found in paper text by Syntology unverified MIT (permissive) · 9fa1ca8f048f4513 · report
load_median_table allshaks/grid-walkers/lattice_walk_repro/analysis/effective_field.py found in paper text by Syntology unverified MIT (permissive) · 04255c9b90a1c706 · report
medians_vec allshaks/grid-walkers/lattice_walk_repro/analysis/effective_field.py found in paper text by Syntology unverified MIT (permissive) · a78fafb45c68f764 · report
pca_3d allshaks/grid-walkers/lattice_walk_repro/analysis/geometry.py found in paper text by Syntology unverified MIT (permissive) · 48fa29145cc812ae · report
regularized_q allshaks/grid-walkers/lattice_walk_repro/analysis/effective_field.py found in paper text by Syntology unverified MIT (permissive) · b547ef5df5602dc5 · report
torch_dtype allshaks/grid-walkers/lattice_walk_repro/training_core.py found in paper text by Syntology unverified MIT (permissive) · 2f18a7f275206a0a · report
variant_label allshaks/grid-walkers/lattice_walk_repro/analysis/model_loading.py found in paper text by Syntology unverified MIT (permissive) · fbd28415ffca2bad · report

Results from the paper

The Papers with Code archive ends with its 2025-07-28 snapshot. This paper's arXiv identifier, 2603.16689, was issued in March 2026, after that date, so the archive has no leaderboard rows for it.

Report a problem or propose a change · a person checks every report against the paper or source before anything changes; decisions are listed on /corrections