Uniform Race: Parameter-Free Approximate Rejection Sampling
Not on the map. Why?
Abstract (arXiv)
We study approximate sampling: given N independent samples from a proposal distribution μ, the goal is to select one whose distribution is close to a target π specified only up to a normalizing constant. Block and Polyanskiy (2023) provide finite budget error bounds for approximate rejection sampling (RS) as a function of the acceptance threshold M. The threshold M giving the smallest bound, however, depends on properties of (π,μ) that are typically unavailable from the observed sample. This raises a natural question: Can one attain the best RS guarantee without taking M as input? We answer affirmatively by proposing a parameter-free sampling algorithm called uniform race (UR), based on importance weights, which are ratios of target to proposal probabilities (or densities). It divides each observed weight by an independent uniform random variable to form a score and returns the candidate with the largest score. For every budget N, its total variation error satisfies the RS upper bound for every fixed threshold M simultaneously, thereby achieving the best such bound in hindsight. We also characterize its output distribution conditional on the largest score, identifying when it is exactly the target π. Uniform race has no larger total variation error than a natural budget-calibrated RS derived from Rohatgi et al. (2025) and sampling importance resampling (SIR). In particular, we exhibit instances where UR's error is exponentially smaller in N than that of either baseline. Furthermore, we establish conditions under which attaining this RS guarantee for every (π,μ) uniquely determines the selection probabilities as those of UR. Finally, test-time scaling experiments on LLM math-reasoning tasks corroborate the theoretical comparisons and demonstrate that UR remains competitive in ground-truth accuracy without requiring threshold selection.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 14 of its 25 references that are in the map. First posted on arXiv 28 Sep 2026.
Builds on
14 of 25 references.
References · 14
- 2024The Llama 3 Herd of Modelscited by 14,192
- 2021Training Verifiers to Solve Math Word Problemscited by 9,743
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,916
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
- 2023Let's Verify Step by Stepcited by 3,598
- 2023OpenAssistant Conversations -- Democratizing Large Language Model Alignmentcited by 712
- 2016Sequence Tutor: Conservative Fine-Tuning of Sequence Generation Models with KL-controlcited by 100
- 2020Human-centric Dialog Training via Offline Reinforcement Learningcited by 92
- 2025Is Best-of-N the Best of Them? Coverage, Scaling, and Optimality in Inference-Time Alignmentcited by 91
- 2025Soft Best-of-n Sampling for Model Alignmentcited by 28
- 2025Best-of-N through the Smoothing Lens: KL Divergence and Regret Analysiscited by 15
- 2023The Sample Complexity of Approximate Rejection Sampling with Applications to Smoothed Online Learningcited by 13
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Lower Bounds for Parallel Diffusion Sampling
- 2024Multi-Draft Speculative Sampling: Canonical Decomposition and Theoretical Limits
- 2026Query Lower Bounds for Diffusion Sampling
- 2024Faster Diffusion Sampling with Randomized Midpoints: Sequential and Parallel
- 2026Limits of Confidence in Diffusion
- 2026Rank-Aware Speculative Sampling for Diffusion Draft Trees
- 2026Efficient Sampling with Discrete Diffusion Models: Sharp and Adaptive Guarantees
- 2026Bradley-Terry model under general comparison graphs