mlsanityData to 14 Oct 2026

Sharp Capacity Thresholds in Linear Associative Memory: From Top-1 Retrieval to Tail-Average Learning

arXiv 2605.05189 ↗First posted 6 May 2026stat.ML cs.IT cs.LGcited by 3

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

  1. 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,336
  2. 2022Toy Models of Superpositioncited by 971
  3. 2014Neural Turing Machinescited by 644
  4. 2020Hopfield Networks is All You Needcited by 475
  5. 2014Memory Networkscited by 363
  6. 2016Dense Associative Memory for Pattern Recognitioncited by 236
  7. 2023Birth of a Transformer: A Memory Viewpointcited by 161
  8. 2024Physics of Language Models: Part 3.3, Knowledge Capacity Scaling Lawscited by 144
  9. 2022Universal Hopfield Networks: A General Framework for Single-Shot Associative Memory Modelscited by 55
  10. 2024Understanding Factual Recall in Transformers via Associative Memoriescited by 51
  11. 2023Scaling Laws for Associative Memoriescited by 35
  12. 2025Key-value memory in the braincited by 24
5 more
  1. 2024Learning Associative Memories with Gradient Descentcited by 16
  2. 2024Do LLMs dream of elephants (when told not to)? Latent concept association and associative memory in transformerscited by 15
  3. 2026Sharp Capacity Scaling of Spectral Optimizers in Learning Associative Memorycited by 14
  4. 2026Transformers as Measure-Theoretic Associative Memory: A Statistical Perspective and Minimax Optimalitycited by 4
  5. 2026Learning to Recall with Transformers Beyond Orthogonal Embeddingscited by 2

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 3.

3 papers.