Inference Pipeline
The cache starts as a decoding speedup and turns into a bandwidth tax as the working set grows.
This page treats the paper as a systems story in pictures: the KV cache grows with context, memory traffic starts to dominate, and IndexMem splits the fix into two distinct jobs, predicting what to keep and compensating for what gets evicted.
The selector estimates future token importance before those future queries fully exist. The latent memory handles the harder failure mode: once a token is evicted, can a compact state still return a useful correction later instead of pretending deletion was harmless?
Switch workload size to separate the part the paper measures from the part it mainly motivates.
The cache starts as a decoding speedup and turns into a bandwidth tax as the working set grows.
As the prompt grows, the need becomes more urgent while the paper’s direct evidence covers a narrower slice.
Illustrative log-scale trend: the systems motivation grows faster than the measured region.
IndexMem’s selector is small on purpose: shared key projection, normalized scores, and max pooling so one future query can keep a quiet but crucial token alive.
Hover cells to inspect future-query spikes. Compression acts on non-sink tokens while the first four sink positions stay pinned.
The important design choice is not “learned” by itself. It is the max pooling over queries, which preserves rare-but-critical evidence.
Evicted keys and values write into a fixed-size state. Later queries read a summary vector back, and that vector only contributes a residual correction.
Use the step buttons to walk through the memory path. Toggle the correction off to see how much of the picture the selector alone must carry.
The episode’s reading is narrow by design: useful bounded-budget evidence, not a blanket proof of million-token readiness.
Switch between RULER stress points, LongBench degradation curves, and the one efficiency pass cited in the transcript.
The episode calls out the visual pattern: fewer empty regions across needle positions when compression gets harsh.
Primary links only. arXiv links are rendered in canonical abs form when IDs were provided.