Remembering Distinct Items, Not Tokens: A Learnable Dirichlet-Process Cache Between State-Space Models and Attention
arXiv:2607.09889
2026
Memory
1 ideas extracted · analyzed Aug 30, 2026
What the math gives to ML
The paper provides a concrete online clustering operator that turns a recurrent model's unbounded token history into a cache whose capacity scales with distinct key patterns rather than sequence length. Its transferable asset is the hard novelty decision: cosine distance to the nearest occupied slot determines whether to allocate memory or merge with an existing prototype, giving an explicit memory-growth control knob instead of heuristic eviction. The most promising integration is a write-side cache after an SSM or recurrent backbone, retaining representative key-value pairs and using the sparse table for associative recall. This should first be tested against full attention and fixed-budget eviction on redundant streams, measuring recall, cache size, and inference cost at equal sequence length.
Ideas from this paper
△ Mechanism confirmed, baseline not beaten
2026
Replace token-by-token KV storage after an SSM or recurrent encoder with an online allocate-on-novelty cache. A new slot is created only when the incoming key is sufficiently dissimilar from every stored key; otherwise the incoming value is merged into its nearest slot, so repeated or redundant content does not grow the cache.
Useful8/10
Difficulty4/10
Novelty6/10