mlsanityData to 7 Oct 2026

Attention is All You Need Until You Need Retention

arXiv 2501.09166 ↗First posted 15 Jan 2025cs.LG cs.AIcited by 1

Not on the map. Why?

Abstract (arXiv)

Pretrained Transformers keep what they learned in their weights and lose what they observe once a session ends. The first version of this paper proposed a Retention Layer, a persistent memory that a Transformer block reads with attention and writes during use. Because most of what a deployed model could retain is produced by other agents, this revision treats deciding what to keep as a social learning problem: when to rely on observed behaviour, whom to learn from and how much independent agreement to require. We give a corrected specification of the layer, which reduces exactly to the base Transformer when its memory is empty. We derive the memory's lifecycle from social learning strategies: encoding gated by surprise, observed outcomes and earned credibility; consolidation by a credibility weighted quorum of distinct, recent sources that must also outweigh every rival behaviour; and reconsolidation by the outcomes of reproduction. We prove that raising the quorum lowers the risk of consolidating a coordinated false template exponentially while delaying true templates only linearly, and that relative consolidation protects only while credible honest evidence arrives faster than adversarial evidence. In a simulation with world drift and three memory-poisoning attacks, the lifecycle reached accuracies of 0.989 to 0.996, against 0.62 to 0.63 for the ungated first version design, and kept attack success at or below 0.07 when 30% of the observations about a target were adversarial. As predicted, it amplified attacks once adversarial evidence outpaced honest evidence. Experience with a long running assistant adds two rules: a model's own outputs must not count as support, and a user's testimony should be kept after one mention. We close with an evaluation protocol for language models.

Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 33 of its 81 references that are in the map. First posted on arXiv 15 Jan 2025.

Builds on

33 of 81 references.

References · 33

  1. 2017Attention Is All You Needcited by 59,932
  2. 2020Language Models are Few-Shot Learnerscited by 32,199
  3. 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,160
  4. 2016Overcoming catastrophic forgetting in neural networkscited by 4,718
  5. 2022Flamingo: a Visual Language Model for Few-Shot Learningcited by 4,337
  6. 2023Lost in the Middle: How Language Models Use Long Contextscited by 3,383
  7. 2024Gemini 1.5: Unlocking multimodal understanding across millions of tokens of contextcited by 3,175
  8. 2023MemGPT: Towards LLMs as Operating Systemscited by 1,295
  9. 2021Improving language models by retrieving from trillions of tokenscited by 1,278
  10. 2024Evaluating Very Long-Term Conversational Memory of LLM Agentscited by 800
  11. 2019Generalization through Memorization: Nearest Neighbor Language Modelscited by 788
  12. 2023LLaMA-Adapter: Efficient Fine-tuning of Language Models with Zero-init Attentioncited by 776
21 more
  1. 2022Transformers learn in-context by gradient descentcited by 714
  2. 2014Neural Turing Machinescited by 630
  3. 2024A Survey on the Memory Mechanism of Large Language Model based Agentscited by 622
  4. 2023MemoryBank: Enhancing Large Language Models with Long-Term Memorycited by 539
  5. 2021Linear Transformers Are Secretly Fast Weight Programmerscited by 471
  6. 2014Memory Networkscited by 354
  7. 2024Learning to (Learn at Test Time): RNNs with Expressive Hidden Statescited by 329
  8. 2024BABILong: Testing the Limits of LLMs with Long Context Reasoning-in-a-Haystackcited by 211
  9. 2023Augmenting Language Models with Long-Term Memorycited by 186
  10. 2016Using Fast Weights to Attend to the Recent Pastcited by 161
  11. 2019Large Memory Layers with Product Keyscited by 120
  12. 2020Memformer: A Memory-Augmented Transformer for Sequence Modelingcited by 63
  13. 2024Human-inspired Episodic Memory for Infinite Context LLMscited by 62
  14. 2016One-shot Learning with Memory-Augmented Neural Networkscited by 60
  15. 2017Dynamic Evaluation of Neural Sequence Modelscited by 53
  16. 2024Larimar: Large Language Models with Episodic Memory Controlcited by 53
  17. 2020Emergent Social Learning via Multi-agent Reinforcement Learningcited by 44
  18. 2024CAMELoT: Towards Large Language Models with Training-Free Consolidated Associative Memorycited by 24
  19. 2024TransformerFAM: Feedback attention is working memorycited by 21
  20. 2023Social Learning: Towards Collaborative Learning with Large Language Modelscited by 15
  21. 2024Social Learning through Interactions with Other Agents: A Surveycited by 3

Built on by

Cited by year

Citing papers per year: 2025: 1.

1 paper.