Scalable Power Sampling: Unlocking Efficient, Training-Free Reasoning for LLMs via Distribution Sharpening
cs.LG, cs.AI
Submitted: 2026-01-29
Updated: 2026-10-12
Comments: Accepted at ICML 2026
License: http://creativecommons.org/licenses/by/4.0/
The gist: Reinforcement learning (RL) post-training is a dominant approach for improving the reasoning performance of large language models (LLMs), yet growing evidence suggests that its gains arise primarily
Terminology
Abstract
Reinforcement learning (RL) post-training is a dominant approach for improving the reasoning performance of large language models (LLMs), yet growing evidence suggests that its gains arise primarily from distribution sharpening rather than the acquisition of new capabilities. Recent work has shown that sampling from the power distribution of LLMs using Markov chain Monte Carlo (MCMC) can recover performance comparable to RL post-training without relying on external rewards; however, the high computational cost of MCMC makes such approaches impractical for widespread adoption. In this work, we propose a theoretically grounded alternative that eliminates the need for iterative MCMC. We derive a token-level decomposition of the autoregressive power distribution, showing that the global power distribution can be approximated by a token-level scaled low-temperature one, where the scaling factor captures future trajectory quality. Leveraging this insight, we introduce a training-free and verifier-free algorithm that sharpens the base model's generative distribution autoregressively. Empirically, we evaluate our method on math, QA, and code tasks, and show that our method matches or surpasses one-shot GRPO without relying on any external rewards, while reducing inference latency by over 10 times compared to MCMC-based sampling.
Sources
- Large Language Models for Mathematical Reasoning: Progresses and Challenges
- Reasoning on a Budget: A Survey of Adaptive and Controllable Test-Time Compute in LLMs
- Large Language Monkeys: Scaling Inference Compute with Repeated Sampling
- Large Language Models for Planning: A Comprehensive and Systematic Survey
- Accelerating Large Language Model Decoding with Speculative Sampling
- Evaluating Large Language Models Trained on Code
- Inference-Aware Fine-Tuning for Best-of-N Sampling in Large Language Models
- Training Verifiers to Solve Math Word Problems
- Break the Sequential Dependency of LLM Inference Using Lookahead Decoding
- Differential Smoothing Mitigates Sharpening and Improves LLM Reasoning
- Is Best-of-N the Best of Them? Coverage, Scaling, and Optimality in Inference-Time Alignment
- Let's Verify Step by Step
- Statistical Rejection Sampling Improves Preference Optimization
- The AI Scientist: Towards Fully Automated Open-Ended Scientific Discovery
- The Jackknife Estimation Method
- Can GRPO Help LLMs Transcend Their Pretraining Origin?
- Code Llama: Open Foundation Models for Code
- Spurious Rewards: Rethinking Training Signals in RLVR
- DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Models
- Outcome-based Exploration for LLM Reasoning
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks