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