Arithmetic Sampling: Parallel Diverse Decoding for Large Language Models
Not on the map. Why?
Abstract (arXiv)
Decoding methods for large language models often trade-off between diversity of outputs and parallelism of computation. Methods such as beam search and Gumbel top-k sampling can guarantee a different output for each element of the beam, but are not easy to parallelize. Alternatively, methods such as temperature sampling and its modifications (top-k sampling, nucleus sampling, typical decoding, and others), are embarrassingly parallel, but have no guarantees about duplicate samples. We present a framework for sampling according to an arithmetic code book implicitly defined by a large language model, compatible with common sampling variations, with provable beam diversity under certain conditions, as well as being embarrassingly parallel and providing unbiased and consistent expectations from the original model. We demonstrate the effectiveness of our approach on WMT machine translation, more than halving the standard deviation when estimating expected BLEU score reward, and closing the BLEU score gap between independent sampling and beam search by up to 63%.
Key facts. Cited by 19 papers in the map (data through 14 Oct 2026). Builds on 20 of its 30 references that are in the map. First posted on arXiv 18 Oct 2022.
Builds on
20 of 30 references.
References · 20
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2022Training language models to follow instructions with human feedbackcited by 15,971
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,599
- 2017Deep reinforcement learning from human preferencescited by 3,912
- 2019The Curious Case of Neural Text Degenerationcited by 2,916
- 2020Learning to summarize from human feedbackcited by 2,588
- 2016Google's Neural Machine Translation System: Bridging the Gap between Human and Machine Translationcited by 1,475
- 2018Hierarchical Neural Story Generationcited by 1,193
- 2015Sequence Level Training with Recurrent Neural Networkscited by 576
- 2019Neural Text Generation with Unlikelihood Trainingcited by 503
- 2019Stochastic Beams and Where to Find Them: The Gumbel-Top-k Trick for Sampling Sequences Without Replacementcited by 211
- 2022Scaling Up Models and Data with textttt5x and textttseqiocited by 187
8 more
- 2014Quasi-Monte Carlo Feature Maps for Shift-Invariant Kernelscited by 42
- 2018Quasi-Monte Carlo Variational Inferencecited by 24
- 2015Language Models for Image Captioning: The Quirks and What Workscited by 19
- 2016Noisy Parallel Approximate Decoding for Conditional Recurrent Language Modelcited by 18
- 2021Determinantal Beam Searchcited by 18
- 2021Implicit Unlikelihood Training: Improving Neural Text Generation with Reinforcement Learningcited by 16
- 2021Conditional Poisson Stochastic Beam Searchcited by 11
- 2021Attention-based Contextual Language Model Adaptation for Speech Recognitioncited by 4
Built on by
Citing papers per year: 2024: 7; 2025: 3; 2026: 9.
19 papers.
Sort:
By area:
- 2024Probabilistic Inference in Language Models via Twisted Sequential Monte Carlocited by 90
- 2024Detecting Machine-Generated Texts by Multi-Population Aware Optimization for Maximum Mean Discrepancycited by 20
- 2024Training LLMs over Neurally Compressed Textcited by 17
- 2024Conformal Generative Modeling with Improved Sample Efficiency through Sequential Greedy Filteringcited by 10
- 2025Semantic-guided Diverse Decoding for Large Language Modelcited by 9
- 2024Priority Sampling of Large Language Models for Compilerscited by 8
- 2025p-less Sampling: A Robust Hyperparameter-Free Approach for LLM Decodingcited by 5
- 2026Annotations Mitigate Post-Training Mode Collapsecited by 5
- 2024Unused information in token probability distribution of generative LLM: improving LLM reading comprehension through calculation of expected valuescited by 4
- 2024Quasi-random Multi-Sample Inference for Large Language Modelscited by 4
- 2025Advancing Decoding Strategies: Enhancements in Locally Typical Sampling for LLMscited by 4
- 2026When are likely answers right? On Sequence Probability and Correctness in LLMscited by 4
7 more
- 2026Decoding as Optimisation on the Probability Simplex: From Top-K to Top-P (Nucleus) to Best-of-K Samplerscited by 2
- 2026Efficient Test-Time Inference via Deterministic Exploration of Truncated Decoding Treescited by 1
- 2026DecompressionLM: Deterministic, Diagnostic, and Zero-Shot Concept Graph Extraction from Language Modelscited by 0
- 2026Improving the Diversity of LLM Outputs without a Trade-offcited by 0
- 2026Composable Decoding on the Probability Simplex: Theory and Implementationcited by 0
- 2026Self-Repulsive Sampling for Diffusion Language Modelscited by 0
- 2026Truncation Samplers Are Nested: Comparing Decoding Rules Is a Measurement Problemcited by 0
- 2026Truncation Samplers Are Nested: Comparing Decoding Rules Is a Measurement Problem11 Oct 2026
- 2026Self-Repulsive Sampling for Diffusion Language Models30 Sep 2026
- 2026Composable Decoding on the Probability Simplex: Theory and Implementation28 Sep 2026
- 2026Improving the Diversity of LLM Outputs without a Trade-off27 Sep 2026
- 2026When are likely answers right? On Sequence Probability and Correctness in LLMs25 Jun 2026
- 2026Annotations Mitigate Post-Training Mode Collapse11 May 2026
- 2026Efficient Test-Time Inference via Deterministic Exploration of Truncated Decoding Trees22 Apr 2026
- 2026Decoding as Optimisation on the Probability Simplex: From Top-K to Top-P (Nucleus) to Best-of-K Samplers20 Feb 2026
- 2026DecompressionLM: Deterministic, Diagnostic, and Zero-Shot Concept Graph Extraction from Language Models30 Jan 2026
- 2025p-less Sampling: A Robust Hyperparameter-Free Approach for LLM Decoding27 Sep 2025
- 2025Semantic-guided Diverse Decoding for Large Language Model30 Jun 2025
- 2025Advancing Decoding Strategies: Enhancements in Locally Typical Sampling for LLMs3 Jun 2025
7 more
- 2024Quasi-random Multi-Sample Inference for Large Language Models9 Nov 2024
- 2024Conformal Generative Modeling with Improved Sample Efficiency through Sequential Greedy Filtering2 Oct 2024
- 2024Unused information in token probability distribution of generative LLM: improving LLM reading comprehension through calculation of expected values11 Jun 2024
- 2024Probabilistic Inference in Language Models via Twisted Sequential Monte Carlo26 Apr 2024
- 2024Training LLMs over Neurally Compressed Text4 Apr 2024
- 2024Priority Sampling of Large Language Models for Compilers28 Feb 2024
- 2024Detecting Machine-Generated Texts by Multi-Population Aware Optimization for Maximum Mean Discrepancy25 Feb 2024
- 2026Self-Repulsive Sampling for Diffusion Language Modelscited by 0
- 2026Self-Repulsive Sampling for Diffusion Language Models30 Sep 2026
- 2024Detecting Machine-Generated Texts by Multi-Population Aware Optimization for Maximum Mean Discrepancycited by 20
- 2024Training LLMs over Neurally Compressed Textcited by 17
- 2024Conformal Generative Modeling with Improved Sample Efficiency through Sequential Greedy Filteringcited by 10
- 2025Semantic-guided Diverse Decoding for Large Language Modelcited by 9
- 2024Priority Sampling of Large Language Models for Compilerscited by 8
- 2025p-less Sampling: A Robust Hyperparameter-Free Approach for LLM Decodingcited by 5
- 2026Annotations Mitigate Post-Training Mode Collapsecited by 5
- 2024Unused information in token probability distribution of generative LLM: improving LLM reading comprehension through calculation of expected valuescited by 4
- 2024Quasi-random Multi-Sample Inference for Large Language Modelscited by 4
- 2025Advancing Decoding Strategies: Enhancements in Locally Typical Sampling for LLMscited by 4
- 2026When are likely answers right? On Sequence Probability and Correctness in LLMscited by 4
- 2026Decoding as Optimisation on the Probability Simplex: From Top-K to Top-P (Nucleus) to Best-of-K Samplerscited by 2
12 most cited of 16 All 16 (JSON)
- 2026Truncation Samplers Are Nested: Comparing Decoding Rules Is a Measurement Problem11 Oct 2026
- 2026Composable Decoding on the Probability Simplex: Theory and Implementation28 Sep 2026
- 2026Improving the Diversity of LLM Outputs without a Trade-off27 Sep 2026
- 2026When are likely answers right? On Sequence Probability and Correctness in LLMs25 Jun 2026
- 2026Annotations Mitigate Post-Training Mode Collapse11 May 2026
- 2026Decoding as Optimisation on the Probability Simplex: From Top-K to Top-P (Nucleus) to Best-of-K Samplers20 Feb 2026
- 2026DecompressionLM: Deterministic, Diagnostic, and Zero-Shot Concept Graph Extraction from Language Models30 Jan 2026
- 2025p-less Sampling: A Robust Hyperparameter-Free Approach for LLM Decoding27 Sep 2025
- 2025Semantic-guided Diverse Decoding for Large Language Model30 Jun 2025
- 2025Advancing Decoding Strategies: Enhancements in Locally Typical Sampling for LLMs3 Jun 2025
- 2024Quasi-random Multi-Sample Inference for Large Language Models9 Nov 2024
- 2024Conformal Generative Modeling with Improved Sample Efficiency through Sequential Greedy Filtering2 Oct 2024
12 newest of 16 All 16 (JSON)