Nearest Neighbor Machine Translation
Not on the map. Why?
Cited by 39 in Retrieval-augmented generation, but placed by its own abstract, which didn’t fit.
Abstract (arXiv)
We introduce k-nearest-neighbor machine translation (kNN-MT), which predicts tokens with a nearest neighbor classifier over a large datastore of cached examples, using representations from a neural translation model for similarity search. This approach requires no additional training and scales to give the decoder direct access to billions of examples at test time, resulting in a highly expressive model that consistently improves performance across many settings. Simply adding nearest neighbor search improves a state-of-the-art German-English translation model by 1.5 BLEU. kNN-MT allows a single model to be adapted to diverse domains by using a domain-specific datastore, improving results by an average of 9.2 BLEU over zero-shot transfer, and achieving new state-of-the-art results -- without training on these domains. A massively multilingual model can also be specialized for particular language pairs, with improvements of 3 BLEU for translating from English into German and Chinese. Qualitatively, kNN-MT is easily interpretable; it combines source and target context to retrieve highly relevant examples.
Key facts. Cited by 209 papers in the map (data through 7 Oct 2026). First posted on arXiv 1 Oct 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 34; 2022: 51; 2023: 54; 2024: 41; 2025: 19; 2026: 10.
209 papers.
Sort:
By area:
- 2021What Makes Good In-Context Examples for GPT-3?cited by 1,184
- 2021Retrieval Augmentation Reduces Hallucination in Conversationcited by 691
- 2021Learning To Retrieve Prompts for In-Context Learningcited by 641
- 2024Retrieval-Augmented Generation for AI-Generated Content: A Surveycited by 352
- 2021BASE Layers: Simplifying Training of Large, Sparse Modelscited by 315
- 2023Text-to-image Diffusion Models in Generative AI: A Surveycited by 264
- 2021Internet-Augmented Dialogue Generationcited by 217
- 2022In-context Examples Selection for Machine Translationcited by 172
- 2022A Survey on Retrieval-Augmented Text Generationcited by 157
- 2022Prompting PaLM for Translation: Assessing Strategies and Performancecited by 144
- 2021End-to-End Training of Multi-Document Reader and Retriever for Open-Domain Question Answeringcited by 133
- 2022Training Language Models with Memory Augmentationcited by 112
38 more
- 2021Case-based Reasoning for Natural Language Queries over Knowledge Basescited by 107
- 2023Text Classification via Large Language Modelscited by 104
- 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
- 2021Efficient Nearest Neighbor Language Modelscited by 93
- 2021Adaptive Semiparametric Language Modelscited by 83
- 2023Learning to Filter Context for Retrieval-Augmented Generationcited by 82
- 2022Efficient Methods for Natural Language Processing: A Surveycited by 78
- 2023Dictionary-based Phrase-level Prompting of Large Language Models for Machine Translationcited by 74
- 2024Reliable, Adaptable, and Attributable Language Models with Retrievalcited by 73
- 2024Retrieval-Augmented Generation for Natural Language Processing: A Surveycited by 70
- 2023Self-Knowledge Guided Retrieval Augmentation for Large Language Modelscited by 69
- 2022Text-Guided Synthesis of Artistic Images with Retrieval-Augmented Diffusion Modelscited by 63
- 2021Neural Machine Translation with Monolingual Translation Memorycited by 61
- 2022Neuro-Symbolic Language Modeling with Automaton-augmented Retrievalcited by 60
- 2021Adaptive Nearest Neighbor Machine Translationcited by 58
- 2021Personalized Federated Learning through Local Memorizationcited by 57
- 2023Benchmarking Machine Translation with Cultural Awarenesscited by 53
- 2023kNN Prompting: Beyond-Context Learning with Calibration-Free Nearest Neighbor Inferencecited by 47
- 2021Domain Adaptation and Multi-Domain Adaptation for Neural Machine Translation: A Surveycited by 46
- 2022Retrieval-Enhanced Machine Learningcited by 46
- 2025A Survey of Test-Time Compute: From Intuitive Inference to Deliberate Reasoningcited by 40
- 2021Fast Nearest Neighbor Machine Translationcited by 37
- 2022Semi-Parametric Neural Image Synthesiscited by 36
- 2022Decoupling Knowledge from Memorization: Retrieval-augmented Prompt Learningcited by 36
- 2021Controllable Semantic Parsing via Retrieval Augmentationcited by 35
- 2024MeaCap: Memory-Augmented Zero-shot Image Captioningcited by 35
- 2022Knowledge Base Question Answering by Case-based Reasoning over Subgraphscited by 31
- 2021How BPE Affects Memorization in Transformerscited by 29
- 2023Why do Nearest Neighbor Language Models Work?cited by 29
- 2024PipeRAG: Fast Retrieval-Augmented Generation via Algorithm-System Co-designcited by 28
- 2021GNN-LM: Language Modeling based on Global Contexts via GNNcited by 27
- 2022kNN-Prompt: Nearest Neighbor Zero-Shot Inferencecited by 27
- 2021Learning Kernel-Smoothed Machine Translation with Retrieved Examplescited by 25
- 2024RAG and RAU: A Survey on Retrieval-Augmented Language Model in Natural Language Processingcited by 25
- 2022Relational Memory Augmented Language Modelscited by 24
- 2022Efficient Cluster-Based k-Nearest-Neighbor Machine Translationcited by 24
- 2022Improving Few-Shot Performance of Language Models via Nearest Neighbor Calibrationcited by 24
50 most cited of 209 All 209 (JSON)
- 2026Compression Beyond the Uncompressed: A Two-Stage Training Recipe for Soft Context Compression in RAG4 Sep 2026
- 2026P2E-VQ: ECG-linked representation augmentation for PPG via discrete patch retrieval31 Jul 2026
- 2026Encode Errors: Representational Retrieval of In-Context Demonstrations for Multilingual Grammatical Error Correction13 Jun 2026
- 2026Differentially Private Datastore Generation for Retrieval-Augmented Inference31 May 2026
- 2026Worlds Within Words: Translating Culture in Ancient Chinese Texts with Multi-Agent Coordination31 May 2026
- 2026Nearest-Neighbor Radii under Dependent Sampling14 May 2026
- 2026Incentivizing Parametric Knowledge via Reinforcement Learning with Verifiable Rewards for Cross-Cultural Entity Translation18 Apr 2026
- 2026Retrieval-Augmented Anatomical Guidance for Text-to-CT Generation9 Mar 2026
- 2026Representation Collapse in Machine Translation Through the Lens of Angular Dispersion19 Feb 2026
- 2026RASST: Retrieval-Augmented Simultaneous Speech Translation30 Jan 2026
- 2025Retrieval-augmented Prompt Learning for Pre-trained Foundation Models23 Dec 2025
- 2025Mined Prompting and Metadata-Guided Generation for Wound Care Visual Question Answering13 Nov 2025
38 more
- 2025TreePrompt: Leveraging Hierarchical Few-Shot Example Selection for Improved English-Persian and English-German Translation4 Oct 2025
- 2025Low-Resource English-Tigrinya MT: Leveraging Multilingual Models, Custom Tokenizers, and Clean Evaluation Benchmarks24 Sep 2025
- 2025Angular Dispersion Accelerates k-Nearest Neighbors Machine Translation20 Sep 2025
- 2025Case-Based Decision-Theoretic Decoding with Quality Memories16 Sep 2025
- 2025Enhancing RAG Efficiency with Adaptive Context Compression24 Jul 2025
- 2025Change of Thought: Adaptive Test-Time Computation17 Jul 2025
- 2025OntoRAG: Enhancing Question-Answering through Automated Ontology Derivation from Unstructured Knowledge Bases31 May 2025
- 2025Nearest Neighbor Multivariate Time Series Forecasting16 May 2025
- 2025A Multimodal Recaptioning Framework to Account for Perceptual Diversity Across Languages in Vision-Language Modeling19 Apr 2025
- 2025Bridging the Linguistic Divide: A Survey on Leveraging Large Language Models for Machine Translation2 Apr 2025
- 2025Long-Tail Crisis in Nearest Neighbor Language Models28 Mar 2025
- 2025RAGO: Systematic Performance Optimization for Retrieval-Augmented Generation Serving18 Mar 2025
- 2025Leveraging Domain Knowledge at Inference Time for LLM Translation: Retrieval versus Generation6 Mar 2025
- 2025Label Distribution Learning-Enhanced Dual-KNN for Text Classification6 Mar 2025
- 2025ExaGPT: Example-Based Machine-Generated Text Detection for Human Interpretability17 Feb 2025
- 2025Explanation based In-Context Demonstrations Retrieval for Multilingual Grammatical Error Correction12 Feb 2025
- 2025A Survey of Test-Time Compute: From Intuitive Inference to Deliberate Reasoning5 Jan 2025
- 2024Retrieval-Augmented Machine Translation with Unstructured Knowledge5 Dec 2024
- 2024Long Context RAG Performance of Large Language Models5 Nov 2024
- 2024On the Loss of Context-awareness in General Instruction Fine-tuning5 Nov 2024
- 2024LLM-based Translation Inference with Iterative Bilingual Understanding16 Oct 2024
- 2024Optimizing Rare Word Accuracy in Direct Speech Translation with a Retrieval-and-Demonstration Approach13 Sep 2024
- 2024Great Memory, Shallow Reasoning: Limits of kNN-LMs21 Aug 2024
- 2024kNN Retrieval for Simple and Effective Zero-Shot Multi-speaker Text-to-Speech20 Aug 2024
- 2024Introducing the NewsPaLM MBR and QE Dataset: LLM-Generated High-Quality Parallel Data Outperforms Traditional Web-Crawled Data13 Aug 2024
- 2024Simply Trainable Nearest Neighbour Machine Translation with GPU Inference29 Jul 2024
- 2024Exploiting Pre-trained Models for Drug Target Affinity Prediction with Nearest Neighbors21 Jul 2024
- 2024Retrieval-Augmented Generation for Natural Language Processing: A Survey18 Jul 2024
- 2024Retrieval-Enhanced Machine Learning: Synthesis and Opportunities17 Jul 2024
- 2024Enhancing Low-Resource NMT with a Multilingual Encoder and Knowledge Distillation: A Case Study9 Jul 2024
- 2024Vector Retrieval with Similarity and Diversity: How Hard Is It?5 Jul 2024
- 2024Efficient Nearest Neighbor based Uncertainty Estimation for Natural Language Processing Tasks2 Jul 2024
- 2024Languages Transferred Within the Encoder: On Representation Transfer in Zero-Shot Multilingual Translation12 Jun 2024
- 2024Efficient k-Nearest-Neighbor Machine Translation with Dynamic Retrieval10 Jun 2024
- 2024Memorization in deep learning: A survey6 Jun 2024
- 2024Improving Zero-Shot Chinese-English Code-Switching ASR with kNN-CTC and Gated Monolingual Datastores6 Jun 2024
- 2024Text Generation: A Systematic Literature Review of Tasks, Evaluation, and Challenges24 May 2024
- 2024RAEE: A Robust Retrieval-Augmented Early Exit Framework for Efficient Inference24 May 2024
50 newest of 209 All 209 (JSON)
- 2024Retrieval-Augmented Generation for AI-Generated Content: A Surveycited by 352
- 2022A Survey on Retrieval-Augmented Text Generationcited by 157
- 2021End-to-End Training of Multi-Document Reader and Retriever for Open-Domain Question Answeringcited by 133
- 2023Lift Yourself Up: Retrieval-augmented Text Generation with Self Memorycited by 103
- 2023Learning to Filter Context for Retrieval-Augmented Generationcited by 82
- 2024Reliable, Adaptable, and Attributable Language Models with Retrievalcited by 73
- 2024Retrieval-Augmented Generation for Natural Language Processing: A Surveycited by 70
- 2022Text-Guided Synthesis of Artistic Images with Retrieval-Augmented Diffusion Modelscited by 63
- 2022Decoupling Knowledge from Memorization: Retrieval-augmented Prompt Learningcited by 36
- 2023Why do Nearest Neighbor Language Models Work?cited by 29
- 2024PipeRAG: Fast Retrieval-Augmented Generation via Algorithm-System Co-designcited by 28
- 2022kNN-Prompt: Nearest Neighbor Zero-Shot Inferencecited by 27
12 most cited of 39 All 39 (JSON)
- 2026Compression Beyond the Uncompressed: A Two-Stage Training Recipe for Soft Context Compression in RAG4 Sep 2026
- 2026Retrieval-Augmented Anatomical Guidance for Text-to-CT Generation9 Mar 2026
- 2026RASST: Retrieval-Augmented Simultaneous Speech Translation30 Jan 2026
- 2025Retrieval-augmented Prompt Learning for Pre-trained Foundation Models23 Dec 2025
- 2025Enhancing RAG Efficiency with Adaptive Context Compression24 Jul 2025
- 2025Long-Tail Crisis in Nearest Neighbor Language Models28 Mar 2025
- 2025RAGO: Systematic Performance Optimization for Retrieval-Augmented Generation Serving18 Mar 2025
- 2024Retrieval-Augmented Machine Translation with Unstructured Knowledge5 Dec 2024
- 2024Retrieval-Augmented Generation for Natural Language Processing: A Survey18 Jul 2024
- 2024Vector Retrieval with Similarity and Diversity: How Hard Is It?5 Jul 2024
- 2024RAEE: A Robust Retrieval-Augmented Early Exit Framework for Efficient Inference24 May 2024
- 2024RAG and RAU: A Survey on Retrieval-Augmented Language Model in Natural Language Processing30 Apr 2024
12 newest of 39 All 39 (JSON)
- 2025A Survey of Test-Time Compute: From Intuitive Inference to Deliberate Reasoningcited by 40
- 2024Great Memory, Shallow Reasoning: Limits of kNN-LMscited by 8
- 2024ARAIDA: Analogical Reasoning-Augmented Interactive Data Annotationcited by 6
- 2022Analogical Math Word Problems Solving with Enhanced Problem-Solution Associationcited by 4
- 2025A Survey of Test-Time Compute: From Intuitive Inference to Deliberate Reasoning5 Jan 2025
- 2024Great Memory, Shallow Reasoning: Limits of kNN-LMs21 Aug 2024
- 2024ARAIDA: Analogical Reasoning-Augmented Interactive Data Annotation20 May 2024
- 2022Analogical Math Word Problems Solving with Enhanced Problem-Solution Association1 Dec 2022
- 2023Text-to-image Diffusion Models in Generative AI: A Surveycited by 264
- 2024Nearest Neighbour Score Estimators for Diffusion Generative Modelscited by 10
- 2021What Makes Good In-Context Examples for GPT-3?cited by 1,184
- 2021Retrieval Augmentation Reduces Hallucination in Conversationcited by 691
- 2021Learning To Retrieve Prompts for In-Context Learningcited by 641
- 2021BASE Layers: Simplifying Training of Large, Sparse Modelscited by 315
- 2021Internet-Augmented Dialogue Generationcited by 217
- 2022In-context Examples Selection for Machine Translationcited by 172
- 2022Prompting PaLM for Translation: Assessing Strategies and Performancecited by 144
- 2022Training Language Models with Memory Augmentationcited by 112
- 2021Case-based Reasoning for Natural Language Queries over Knowledge Basescited by 107
- 2023Text Classification via Large Language Modelscited by 104
- 2022Language Models that Seek for Knowledge: Modular Search & Generation for Dialogue and Prompt Completioncited by 102
- 2021Efficient Nearest Neighbor Language Modelscited by 93
12 most cited of 162 All 162 (JSON)
- 2026P2E-VQ: ECG-linked representation augmentation for PPG via discrete patch retrieval31 Jul 2026
- 2026Encode Errors: Representational Retrieval of In-Context Demonstrations for Multilingual Grammatical Error Correction13 Jun 2026
- 2026Differentially Private Datastore Generation for Retrieval-Augmented Inference31 May 2026
- 2026Worlds Within Words: Translating Culture in Ancient Chinese Texts with Multi-Agent Coordination31 May 2026
- 2026Nearest-Neighbor Radii under Dependent Sampling14 May 2026
- 2026Incentivizing Parametric Knowledge via Reinforcement Learning with Verifiable Rewards for Cross-Cultural Entity Translation18 Apr 2026
- 2026Representation Collapse in Machine Translation Through the Lens of Angular Dispersion19 Feb 2026
- 2025Mined Prompting and Metadata-Guided Generation for Wound Care Visual Question Answering13 Nov 2025
- 2025TreePrompt: Leveraging Hierarchical Few-Shot Example Selection for Improved English-Persian and English-German Translation4 Oct 2025
- 2025Low-Resource English-Tigrinya MT: Leveraging Multilingual Models, Custom Tokenizers, and Clean Evaluation Benchmarks24 Sep 2025
- 2025Angular Dispersion Accelerates k-Nearest Neighbors Machine Translation20 Sep 2025
- 2025Case-Based Decision-Theoretic Decoding with Quality Memories16 Sep 2025
12 newest of 162 All 162 (JSON)