twanghcmut's picture
|
download
raw
2.36 kB

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, -1024 env-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.