English

Power-SMC: Low-Latency Sequence-Level Power Sampling for Training-Free LLM Reasoning

Machine Learning 2026-03-24 v2 Machine Learning

Abstract

Many recent reasoning gains in large language models can be explained as distribution sharpening: biasing generation toward high-likelihood trajectories already supported by the pretrained model, rather than modifying its weights. A natural formalization is the sequence-level power distribution πα(yx)pθ(yx)α\pi_\alpha(y\mid x)\propto p_\theta(y\mid x)^\alpha (α>1\alpha>1), which concentrates mass on whole sequences instead of adjusting token-level temperature. Prior work shows that Metropolis--Hastings (MH) sampling from this distribution recovers strong reasoning performance, but at order-of-magnitude inference slowdowns. We introduce Power-SMC, a training-free Sequential Monte Carlo scheme that targets the same objective while remaining close to standard decoding latency. Power-SMC advances a small particle set in parallel, corrects importance weights token-by-token, and resamples when necessary, all within a single GPU-friendly batched decode. We prove that temperature τ=1/α\tau=1/\alpha is the unique prefix-only proposal minimizing incremental weight variance, interpret residual instability via prefix-conditioned R\'enyi entropies, and introduce an exponent-bridging schedule that improves particle stability without altering the target. On MATH500, Power-SMC matches or exceeds MH power sampling while reducing latency from 1616--28×28\times to 1.41.4--3.3×3.3\times over baseline decoding. The code is available at https://github.com/ArminAzizi98/Power-SMC.

Keywords

Cite

@article{arxiv.2602.10273,
  title  = {Power-SMC: Low-Latency Sequence-Level Power Sampling for Training-Free LLM Reasoning},
  author = {Seyedarmin Azizi and Erfan Baghaei Potraghloo and Minoo Ahmadi and Souvik Kundu and Massoud Pedram},
  journal= {arXiv preprint arXiv:2602.10273},
  year   = {2026}
}
R2 v1 2026-07-01T10:30:42.657Z