Improving Neural Language Models with a Continuous Cache
Not on the map. Why?
Abstract (arXiv)
We propose an extension to neural network language models to adapt their prediction to the recent history. Our model is a simplified version of memory augmented networks, which stores past hidden activations as memory and accesses them through a dot product with the current hidden activation. This mechanism is very efficient and scales to very large memory sizes. We also draw a link between the use of external memory in neural network and cache models used with count based language models. We demonstrate on several language model datasets that our approach performs significantly better than recent memory augmented networks.
Key facts. Cited by 81 papers in the map (data through 7 Oct 2026). First posted on arXiv 13 Dec 2016.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 21; 2022: 24; 2023: 16; 2024: 12; 2025: 3; 2026: 5.
81 papers.
Sort:
By area:
- 2021Improving language models by retrieving from trillions of tokenscited by 1,278
- 2022Atlas: Few-shot Learning with Retrieval Augmented Language Modelscited by 940
- 2021Retrieval Augmentation Reduces Hallucination in Conversationcited by 691
- 2024A Survey on RAG Meeting LLMs: Towards Retrieval-Augmented Large Language Modelscited by 450
- 2023Augmented Language Models: a Surveycited by 398
- 2023Think before you speak: Training Language Models With Pause Tokenscited by 254
- 2022Memorizing Transformerscited by 219
- 2021Internet-Augmented Dialogue Generationcited by 217
- 2023Domain Specialization as the Key to Make Large Language Models Disruptive: A Comprehensive Surveycited by 134
- 2021Between words and characters: A Brief History of Open-Vocabulary Modeling and Tokenization in NLPcited by 130
- 2021How much do language models copy from their training data? Evaluating linguistic novelty in text generation using RAVENcited by 124
- 2022Training Language Models with Memory Augmentationcited by 112
38 more
- 2023Lift Yourself Up: Retrieval-augmented Text Generation with Self Memorycited by 103
- 2022Language Models that Seek for Knowledge: Modular Search & Generation for Dialogue and Prompt Completioncited by 102
- 2021Adaptive Semiparametric Language Modelscited by 83
- 2022A Survey of Knowledge Enhanced Pre-trained Language Modelscited by 76
- 2022Your Transformer May Not be as Powerful as You Expectcited by 55
- 2021On the Evaluation of Neural Code Summarizationcited by 47
- 2021Paradigm Shift in Natural Language Processingcited by 39
- 2024Recent Advances in Generative AI and Large Language Models: Current Status, Challenges, and Perspectivescited by 30
- 2021Selfish Sparse RNN Trainingcited by 26
- 2022Efficient recurrent architectures through activity sparsity and sparse back-propagation through timecited by 26
- 2024RAG and RAU: A Survey on Retrieval-Augmented Language Model in Natural Language Processingcited by 25
- 2024ALR2: A Retrieve-then-Reason Framework for Long-context Question Answeringcited by 25
- 2022Relational Memory Augmented Language Modelscited by 24
- 2022Knowledge-in-Context: Towards Knowledgeable Semi-Parametric Language Modelscited by 24
- 2024Fuzz-Testing Meets LLM-Based Agents: An Automated and Efficient Framework for Jailbreaking Text-To-Image Generation Modelscited by 24
- 2025Pre-training under infinite computecited by 24
- 2022You can't pick your neighbors, or can you? When and how to rely on retrieval in the kNN-LMcited by 18
- 2024KBLaM: Knowledge Base augmented Language Modelcited by 16
- 2024Frozen Feature Augmentation for Few-Shot Image Classificationcited by 13
- 2023Recurrent Attention Networks for Long-text Modelingcited by 12
- 2021Beyond Short Clips: End-to-End Video-Level Learning with Collaborative Memoriescited by 11
- 2024Retrieval-Enhanced Machine Learning: Synthesis and Opportunitiescited by 11
- 2021Low-Resource Language Modelling of South African Languagescited by 10
- 2023KNN-LM Does Not Improve Open-ended Text Generationcited by 10
- 2024Efficient Prompt Caching via Embedding Similaritycited by 9
- 2021Capturing Structural Locality in Non-parametric Language Modelscited by 8
- 2021Faster Nearest Neighbor Machine Translationcited by 8
- 2022Zemi: Learning Zero-Shot Semi-Parametric Language Models from Multiple Taskscited by 8
- 2021∞-former: Infinite Memory Transformercited by 7
- 2024Vocabulary-Defined Semantics: Latent Space Clustering for Improving In-Context Learningcited by 7
- 2022Semi-Parametric Inducing Point Networks and Neural Processescited by 6
- 2022A Natural Bias for Language Generation Modelscited by 6
- 2023HistAlign: Improving Context Dependency in Language Generation by Aligning with Historycited by 6
- 2022Better Language Model with Hypernym Class Predictioncited by 5
- 2022Embedding Recycling for Language Modelscited by 5
- 2023TAPIR: Learning Adaptive Revision for Incremental Natural Language Understanding with a Two-Pass Modelcited by 5
- 2023Approximate Caching for Efficiently Serving Diffusion Modelscited by 5
- 2024NNOSE: Nearest Neighbor Occupational Skill Extractioncited by 5
50 most cited of 81 All 81 (JSON)
- 2026Forking: Sudden Overfitting Under Replay30 Sep 2026
- 2026Warranted Attention: Learning What to Pass from Attention to Prediction29 Jun 2026
- 2026Parallel Causal Associative Fields: Gated Sparse Memory for Long-Context Language Modeling9 Jun 2026
- 2026On the Importance and Evaluation of Narrativity in Natural Language AI Explanations20 Apr 2026
- 2026TempoFit: Plug-and-Play Layer-Wise Temporal KV Memory for Long-Horizon Vision-Language-Action Manipulation8 Mar 2026
- 2025GPT and Prejudice: A Sparse Approach to Understanding Learned Representations in Large Language Models24 Sep 2025
- 2025Pre-training under infinite compute18 Sep 2025
- 2025Foundations of GenIR6 Jan 2025
- 2024KBLaM: Knowledge Base augmented Language Model14 Oct 2024
- 2024ALR2: A Retrieve-then-Reason Framework for Long-context Question Answering4 Oct 2024
- 2024Fuzz-Testing Meets LLM-Based Agents: An Automated and Efficient Framework for Jailbreaking Text-To-Image Generation Models1 Aug 2024
- 2024Recent Advances in Generative AI and Large Language Models: Current Status, Challenges, and Perspectives20 Jul 2024
38 more
- 2024Retrieval-Enhanced Machine Learning: Synthesis and Opportunities17 Jul 2024
- 2024A Survey on RAG Meeting LLMs: Towards Retrieval-Augmented Large Language Models10 May 2024
- 2024Weight Sparsity Complements Activity Sparsity in Neuromorphic Language Models1 May 2024
- 2024RAG and RAU: A Survey on Retrieval-Augmented Language Model in Natural Language Processing30 Apr 2024
- 2024Frozen Feature Augmentation for Few-Shot Image Classification15 Mar 2024
- 2024Efficient Prompt Caching via Embedding Similarity2 Feb 2024
- 2024NNOSE: Nearest Neighbor Occupational Skill Extraction30 Jan 2024
- 2024Vocabulary-Defined Semantics: Latent Space Clustering for Improving In-Context Learning29 Jan 2024
- 2023Cached Transformers: Improving Transformers with Differentiable Memory Cache20 Dec 2023
- 2023Approximate Caching for Efficiently Serving Diffusion Models7 Dec 2023
- 2023Activity Sparsity Complements Weight Sparsity for Efficient RNN Inference13 Nov 2023
- 2023Style Locality for Controllable Generation with kNN Language Models1 Nov 2023
- 2023Think before you speak: Training Language Models With Pause Tokens3 Oct 2023
- 2023A Framework for Inference Inspired by Human Memory Mechanisms1 Oct 2023
- 2023Recurrent Attention Networks for Long-text Modeling12 Jun 2023
- 2023Domain Specialization as the Key to Make Large Language Models Disruptive: A Comprehensive Survey30 May 2023
- 2023A Quantitative Review on Language Model Efficiency Research28 May 2023
- 2023KNN-LM Does Not Improve Open-ended Text Generation24 May 2023
- 2023Extending Memory for Language Modelling19 May 2023
- 2023TAPIR: Learning Adaptive Revision for Incremental Natural Language Understanding with a Two-Pass Model18 May 2023
- 2023HistAlign: Improving Context Dependency in Language Generation by Aligning with History8 May 2023
- 2023Lift Yourself Up: Retrieval-augmented Text Generation with Self Memory3 May 2023
- 2023FIANCEE: Faster Inference of Adversarial Networks via Conditional Early Exits20 Apr 2023
- 2023Augmented Language Models: a Survey15 Feb 2023
- 2022A Natural Bias for Language Generation Models19 Dec 2022
- 2022A Survey of Knowledge Enhanced Pre-trained Language Models11 Nov 2022
- 2022Knowledge-in-Context: Towards Knowledgeable Semi-Parametric Language Models28 Oct 2022
- 2022You can't pick your neighbors, or can you? When and how to rely on retrieval in the kNN-LM28 Oct 2022
- 2022Nearest Neighbor Language Models for Stylistic Controllable Generation27 Oct 2022
- 2022N-gram Is Back: Residual Learning of Neural Text Generation with n-gram Language Model26 Oct 2022
- 2022Zemi: Learning Zero-Shot Semi-Parametric Language Models from Multiple Tasks1 Oct 2022
- 2022Recurrent Convolutional Neural Networks Learn Succinct Learning Algorithms1 Sep 2022
- 2022Atlas: Few-shot Learning with Retrieval Augmented Language Models5 Aug 2022
- 2022Fusing Sentence Embeddings Into LSTM-based Autoregressive Language Models4 Aug 2022
- 2022Learning Sequence Representations by Non-local Recurrent Neural Memory20 Jul 2022
- 2022Embedding Recycling for Language Models11 Jul 2022
- 2022Efficient recurrent architectures through activity sparsity and sparse back-propagation through time13 Jun 2022
- 2022Syntactic Inductive Biases for Deep Learning Methods8 Jun 2022
50 newest of 81 All 81 (JSON)
- 2022Atlas: Few-shot Learning with Retrieval Augmented Language Modelscited by 940
- 2024A Survey on RAG Meeting LLMs: Towards Retrieval-Augmented Large Language Modelscited by 450
- 2023Lift Yourself Up: Retrieval-augmented Text Generation with Self Memorycited by 103
- 2024RAG and RAU: A Survey on Retrieval-Augmented Language Model in Natural Language Processingcited by 25
- 2023KNN-LM Does Not Improve Open-ended Text Generationcited by 10
- 2024A Survey on RAG Meeting LLMs: Towards Retrieval-Augmented Large Language Models10 May 2024
- 2024RAG and RAU: A Survey on Retrieval-Augmented Language Model in Natural Language Processing30 Apr 2024
- 2023KNN-LM Does Not Improve Open-ended Text Generation24 May 2023
- 2023Lift Yourself Up: Retrieval-augmented Text Generation with Self Memory3 May 2023
- 2022Atlas: Few-shot Learning with Retrieval Augmented Language Models5 Aug 2022
- 2023Augmented Language Models: a Surveycited by 398
- 2023Think before you speak: Training Language Models With Pause Tokenscited by 254
- 2024ALR2: A Retrieve-then-Reason Framework for Long-context Question Answeringcited by 25
- 2021Improving language models by retrieving from trillions of tokenscited by 1,278
- 2021Retrieval Augmentation Reduces Hallucination in Conversationcited by 691
- 2022Memorizing Transformerscited by 219
- 2021Internet-Augmented Dialogue Generationcited by 217
- 2023Domain Specialization as the Key to Make Large Language Models Disruptive: A Comprehensive Surveycited by 134
- 2021Between words and characters: A Brief History of Open-Vocabulary Modeling and Tokenization in NLPcited by 130
- 2021How much do language models copy from their training data? Evaluating linguistic novelty in text generation using RAVENcited by 124
- 2022Training Language Models with Memory Augmentationcited by 112
- 2022Language Models that Seek for Knowledge: Modular Search & Generation for Dialogue and Prompt Completioncited by 102
- 2021Adaptive Semiparametric Language Modelscited by 83
- 2022A Survey of Knowledge Enhanced Pre-trained Language Modelscited by 76
- 2022Your Transformer May Not be as Powerful as You Expectcited by 55
12 most cited of 68 All 68 (JSON)
- 2026Forking: Sudden Overfitting Under Replay30 Sep 2026
- 2026Warranted Attention: Learning What to Pass from Attention to Prediction29 Jun 2026
- 2026Parallel Causal Associative Fields: Gated Sparse Memory for Long-Context Language Modeling9 Jun 2026
- 2026On the Importance and Evaluation of Narrativity in Natural Language AI Explanations20 Apr 2026
- 2025GPT and Prejudice: A Sparse Approach to Understanding Learned Representations in Large Language Models24 Sep 2025
- 2025Pre-training under infinite compute18 Sep 2025
- 2025Foundations of GenIR6 Jan 2025
- 2024KBLaM: Knowledge Base augmented Language Model14 Oct 2024
- 2024Recent Advances in Generative AI and Large Language Models: Current Status, Challenges, and Perspectives20 Jul 2024
- 2024Retrieval-Enhanced Machine Learning: Synthesis and Opportunities17 Jul 2024
- 2024Weight Sparsity Complements Activity Sparsity in Neuromorphic Language Models1 May 2024
- 2024Frozen Feature Augmentation for Few-Shot Image Classification15 Mar 2024
12 newest of 68 All 68 (JSON)