Buckets:
How to read the memory-selection heatmaps
Each PNG is one rollout episode.
- X axis = policy call (time →). One call ≈ 16 env-steps (the model acts on a 16-step chunk, then re-plans). The first few calls are the demo-watch/priming phase.
- Y axis = the 8 memory slots of the log-spaced window, labelled by how far back they look:
current(0),-16,-32,-64,-128,-256,-512,-1024env-steps. Early in an episode the deep slots don't exist yet, so they clamp to the oldest available frame. - Color = the gate weight on that slot at that moment. Black = discarded (~0), bright = kept. This is literally "how much the model trusts that past observation right now."
What to look for — the core finding
Permanence (VideoUnmask_episode_5_success) — recall WORKS.
The current row stays warm (always anchor on now), and as the episode proceeds the gate
progressively lights up specific deeper slots, reaching all the way back to -1024 (the
demo / pre-occlusion region) by late episode. The task is "remember which container hides the
green cube" — and you can see the gate reach back to the observation from before the cube was
covered. This is the mechanism behind Permanence ≈ HAMLET (20.0 vs 19.5).
Counting (BinFill_episode_*) — integration FAILS.
Compare the same layout: the gate is almost entirely stuck on the current row, with only
faint, sparse deep activation. Counting ("put N cubes in the bin, then stop") needs the model to
retain every placement event and add them up — but the gate's job is selective recall, so
it throws most of the history away. The result is below even no-memory (5.0 vs vanilla 14): the
answer is partly visible in the current frame, and the sparse gated memory adds noise instead of
a reliable count.
Imitation (MoveCube_episode_*).
The demo is ~15 snapshots (n_demo≈246 env-steps). At exec time the log-window can only sample a
few of them, front-loaded — so the fine demo trajectory is under-represented. The gate primes and
attends, but the window resolution over a long demo is the bottleneck (5.0 vs HAMLET 14).
One-line summary
Selective gating helps "recall one past thing" and hurts "integrate all past things." That trade-off is the whole story of this checkpoint, and it is visible directly in these images.
Xet Storage Details
- Size:
- 2.36 kB
- Xet hash:
- cea4bd206f48f538f5b9e3145d512a2c429f61c4463d0a8cf84e144b5a0cde0f
Xet efficiently stores files, intelligently splitting them into unique chunks and accelerating uploads and downloads. More info.