Sharp Capacity Thresholds in Linear Associative Memory: From Top-1 Retrieval to Tail-Average Learning
Not on the map. Why?
Abstract (arXiv)
How many key-value associations can a d× d linear memory store? The answer depends not only on the d2 degrees of freedom in the memory matrix, but also on the retrieval criterion. Under isotropic Gaussian embeddings, we prove a sharp threshold for top-1 retrieval, where every signal must beat its largest distractor: the critical value of d2/(nlog n) is 2. Above the threshold, we explicitly construct a linear memory that retrieves all n associations with high probability; below it, no data-dependent linear memory can do so. The log n factor is therefore the unavoidable extreme-value cost of winner-take-all decoding. Without the logarithmic factor---that is, when n/d2→α∈(0,∞)---simultaneous top-1 retrieval is impossible. The matched target can nevertheless remain near the top of the ranking. We capture this weaker retrieval goal with the Tail-Average Margin (TAM), which, for list size k, compares each signal with the average of its k strongest competitors; a positive TAM margin certifies that the target belongs to the top-k candidate list. When k/n→ r∈(0,1), we learn the memory by empirical risk minimization with a smoothed TAM objective and derive an exact high-dimensional characterization through a two-parameter scalar variational problem. The result gives limiting laws for signal and competitor scores, margins, and percentile ranks. Sending the ridge parameter to zero after the high-dimensional limit yields a closed-form critical load αc(r) separating vanishing from positive average loss. The analysis in this work also develops a coupled leave-one-out method for matrix-valued empirical risk problems in which each sample enters many dependent comparisons, a tool that may be useful beyond associative memory.
Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Builds on 17 of its 60 references that are in the map. First posted on arXiv 6 May 2026.
Builds on
17 of 60 references.
References · 17
- 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,336
- 2022Toy Models of Superpositioncited by 971
- 2014Neural Turing Machinescited by 644
- 2020Hopfield Networks is All You Needcited by 475
- 2014Memory Networkscited by 363
- 2016Dense Associative Memory for Pattern Recognitioncited by 236
- 2023Birth of a Transformer: A Memory Viewpointcited by 161
- 2024Physics of Language Models: Part 3.3, Knowledge Capacity Scaling Lawscited by 144
- 2022Universal Hopfield Networks: A General Framework for Single-Shot Associative Memory Modelscited by 55
- 2024Understanding Factual Recall in Transformers via Associative Memoriescited by 51
- 2023Scaling Laws for Associative Memoriescited by 35
- 2025Key-value memory in the braincited by 24
5 more
- 2024Learning Associative Memories with Gradient Descentcited by 16
- 2024Do LLMs dream of elephants (when told not to)? Latent concept association and associative memory in transformerscited by 15
- 2026Sharp Capacity Scaling of Spectral Optimizers in Learning Associative Memorycited by 14
- 2026Transformers as Measure-Theoretic Associative Memory: A Statistical Perspective and Minimax Optimalitycited by 4
- 2026Learning to Recall with Transformers Beyond Orthogonal Embeddingscited by 2
Built on by
Citing papers per year: 2026: 3.
3 papers.