Self-Repulsive Sampling for Diffusion Language Models
On the map: Diffusion models → Discrete and text diffusion
Abstract (arXiv)
Sampling several responses and voting over their answers can improve a language model's accuracy, but repeated answers limit the benefit of additional samples. Raising temperature increases diversity at a potential cost to per-sample accuracy. We introduce Self-Repulsion (SR), a sampler for masked diffusion language models that uses peer commitments to diversify the pool. At each penalized denoising step, each path lowers a token's logit according to how many peers have committed that token at the same position. Paths share a batched forward pass and then commit in sequence, so later paths observe choices made earlier in the same step. This coupling requires no training or additional forward or backward pass and can produce distinct paths even at temperature zero. When all paths commit a position together from identical logits, the update exactly maximizes total logit minus a convex duplication cost. On LLaDA-8B-Instruct with ten paths and 128 denoising steps, deterministic SR reaches 80.38% plurality accuracy on GSM8K, compared with 70.17% for the unpenalized greedy decoder. At temperature 0.6 and matched model-evaluation budgets, the count penalty improves over self-consistency by 2.06 percentage points in blocks of 32 and 14.50 under pure diffusion. Experiments on GSM8K, MATH and TruthfulQA show that voting gains arise mainly from higher coverage of correct answers, with gains that vary by benchmark and decoding regime.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Discrete and text diffusion, one of 14 directions in Diffusion models. Builds on 42 of its 49 references that are in the map. First posted on arXiv 30 Sep 2026.
Builds on
42 of 49 references.
Closest ancestors · 10
Same direction, newest first
- 2026A Tale of Two Temperatures: Simple, Efficient, and Diverse Sampling from Diffusion Language Models10 Apr
- 2026D5P4: Partition Determinantal Point Process for Diversity in Parallel Discrete Diffusion Decoding19 Mar
- 2026Free Lunch for Pass@k? Low Cost Diverse Sampling for Diffusion Language Models5 Mar
- 2026Self-Rewarding Sequential Monte Carlo for Masked Diffusion Language Models2 Feb
- 2025Dream 7B: Diffusion Large Language Models21 Aug
- 2025A Survey on Diffusion Language Models14 Aug
- 2025Seed Diffusion: A Large-Scale Diffusion Language Model with High-Speed Inference4 Aug
- 2025Inference-Time Scaling of Diffusion Language Models via Trajectory Refinement11 Jul
- 2025Block Diffusion: Interpolating Between Autoregressive and Diffusion Language Models12 Mar
- 2025Large Language Diffusion Models14 Feb
Other references · 32
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,459
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2021Evaluating Large Language Models Trained on Codecited by 8,320
- 2022Self-Consistency Improves Chain of Thought Reasoning in Language Modelscited by 5,708
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
- 2021TruthfulQA: Measuring How Models Mimic Human Falsehoodscited by 2,975
- 2016Stein Variational Gradient Descent: A General Purpose Bayesian Inference Algorithmcited by 507
- 2016From Softmax to Sparsemax: A Sparse Model of Attention and Multi-Label Classificationcited by 341
- 2016Diverse Beam Search: Decoding Diverse Solutions from Neural Sequence Modelscited by 301
- 2025LLaDA 1.5: Variance-Reduced Preference Optimization for Large Language Diffusion Modelscited by 285
- 2025Mercury: Ultra-Fast Language Models Based on Diffusioncited by 235
- 2022A Contrastive Framework for Neural Text Generationcited by 225
20 more
- 2019Stochastic Beams and Where to Find Them: The Gumbel-Top-k Trick for Sampling Sequences Without Replacementcited by 211
- 2024Chain-of-Thought Reasoning Without Promptingcited by 210
- 2021Repulsive Deep Ensembles are Bayesiancited by 104
- 2023A Unified Theory of Diversity in Ensemble Learningcited by 71
- 2023Particle Guidance: non-I.I.D. Diverse Sampling with Diffusion Modelscited by 71
- 2025Correlated Errors in Large Language Modelscited by 45
- 2022Arithmetic Sampling: Parallel Diverse Decoding for Large Language Modelscited by 19
- 2021Determinantal Beam Searchcited by 18
- 2025Semantic-guided Diverse Decoding for Large Language Modelcited by 9
- 2023Self-Repellent Random Walks on General Graphs -- Achieving Minimal Sampling Variance via Nonlinear Markov Chainscited by 8
- 2025Test-Time Scaling in Diffusion LLMs via Hidden Semi-Autoregressive Expertscited by 8
- 2020Stein Self-Repulsive Dynamics: Benefits From Past Samplescited by 6
- 2025Think in Parallel, Answer as One: Logit Averaging for Open-Ended Reasoningcited by 6
- 2026When More Sampling Hurts: The Modal Ceiling and Correlation Ceiling of Test-Time Scalingcited by 6
- 2024Quasi-random Multi-Sample Inference for Large Language Modelscited by 4
- 2026dVoting: Fast Voting for dLLMscited by 4
- 2026QuasiMoTTo: Quasi-Monte Carlo Test-Time Scalingcited by 4
- 2025Efficient semantic uncertainty quantification in language models via diversity-steered samplingcited by 2
- 2026A Universal Avoidance Method for Diverse Multi-branch Generationcited by 1
- 2026Exploring More to Solve More: Boosting Diversity in Text Diffusion Models via Entropy-Based Guidancecited by 1
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026LESS Is More: Mutual-Stability Sampling for Diffusion Language Models
- 2026Attention-Discounted Adaptive Sampler for Masked Diffusion Language Models
- 2026Revise, Don't Freeze: Sampler-Matched Training for Self-Correcting Masked Diffusion Language Models
- 2026FastDiSS: Few-step Match Many-step Diffusion Language Model on Sequence-to-Sequence Generation--Full Version
- 2026Re-evaluating Confidence Remasking in Masked Diffusion Language Models
- 2026CoDR: Training-Free Confidence-Drift Remasking for Diffusion Language Models
- 2025Optimal Inference Schedules for Masked Diffusion Models
- 2025Diffusion Language Models are Provably Optimal Parallel Samplers