Scalable Power Sampling: Unlocking Efficient, Training-Free Reasoning for LLMs via Distribution Sharpening
Organizations: Huawei Noah’s Ark Lab · UCL Centre for Artificial Intelligence
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 compared to MCMC-based sampling.
Figures & tables
| MATH500 | HumanEval | GPQA | |
| Qwen2.5-7B | |||
| Base | 0.498 | 0.329 | 0.278 |
| Low-temperature | 0.628 | 0.524 | 0.303 |
| Best-of-N | 0.650 | 0.609 | 0.282 |
| Self-Consistency | 0.670 | 0.742 | 0.282 |
| MCMC Power Sampling | 0.706 | 0.622 | 0.318 |
| MATH500 | HumanEval | GPQA | |
|---|---|---|---|
| DeepSeek-Math-7B-RL | |||
| Base | 0.492 | 0.524 | 0.333 |
| Low-temperature | 0.412 | 0.524 | 0.303 |
| Best-of-N | 0.492 | 0.507 | 0.297 |
| Self-Consistency | 0.492 | 0.524 | 0.349 |
| MCMC Power Sampling | 0.494 | 0.530 | 0.349 |
| MATH500 | HumanEval | GPQA | |
|---|---|---|---|
| Qwen2.5-32B | |||
| Base | 0.577 | 0.392 | 0.376 |
| Low-temperature | 0.682 | 0.507 | 0.414 |
| MCMC Power Sampling | 0.738 | 0.567 | 0.383 |
| Power Sampling (ours) | 0.748 | 0.793 | 0.429 |
Appendix figures & tables9 assets
Supplementary material from the paper’s appendix.
Appendix
| Task | Qwen Prompt | DeepSeek Prompt |
|---|---|---|
| MATH500 | Can you solve the following math problem? Please reason step by step, and put your final answer within \boxed{} | Please reason step by step, and put your final answer within \boxed{} |
| HumanEval | Write a Python function to solve the following problem: | Complete the following Python function: |
| GPQA | Answer the following multiple-choice question. The last line of your response should be of the following format: ’\boxed{LETTER}’ (without quotes) where LETTER is one of ABCD (ex. ’\boxed{A}’). Think step by step before answering. {Question} A) {} B) {} C) {} D) {} | Answer the following multiple-choice question. The last line of your response should be of the following format: ’\boxed{LETTER}’ (without quotes) where LETTER is one of ABCD (ex. ’\boxed{A}’). Think step by step before answering. {Question} A) {} B) {} C) {} D) {} |
| Method | Hyperparameters |
|---|---|
| Low-temperature | temperature (with ) |
| Best-of- | ; selection score |
| MCMC power sampling | , , , : base temperature |
| Power Sampling (ours) | , , , |
| Example 2 (Task 82, MATH500, Qwen2.5-Math-7B). |
|---|
| Question. Point lies somewhere within or on the square with opposite corners and . Point lies somewhere within or on the square with opposite corners and . What is the greatest possible value of the slope of the line containing points and ? Express your answer as a common fraction. |
| GRPO (MATH). |
| To determine the greatest possible value of the slope of the line containing points and , we need to consider the coordinates of the points and within their respective squares. The slope of a line passing through points and is given by . To maximize this slope, we need to maximize the difference in the -coordinates and minimize the difference in the -coordinates. … The coordinates of point can range from to , and the coordinates of point can range from to . …This means we should consider the points for and for , as these points will give the maximum possible slope. … Let’s calculate the slope for these points: Let’s verify this using Python code to ensure accuracy. … output 3.0 Therefore, the greatest possible value of the slope of the line containing points and is: |
| Ours. |
| To determine the greatest possible value of the slope of the line containing points and , we need to consider the coordinates of the points and within their respective squares. The slope of the line passing through points and is given by: The coordinates of point can range from to , and the coordinates of point can range from to . To maximize the slope, we need to maximize the numerator and minimize the denominator . … The maximum value of occurs when and , giving . The minimum value of occurs when and , giving . Therefore, the greatest possible value of the slope is: Let’s confirm this with Python code … output 1.5 Thus, the final answer is: |
| Ground truth. |
| Example 2 (Task 19, GPQA-Diamond, Qwen2.5-Math-7B). |
|---|
| Question. A common approximation made in many-body nuclear calculations of some interaction (for example an electroweak interaction) is to suppose that nucleons in a nucleus behave in the same way as free nucleons. Which of the following four assumptions jointly imply this “impulse” approximation? 1. The interaction current only interacts with individual nucleons. |
| 2. The nucleus is transparent apart from the selected nucleon. |
| 3. The quarks internal to the selected nucleon are non-relativistic. |
| 4. The interaction proceeds as if the selected nucleon experiences no binding forces. |
| A) 2,3,4 B) 1,2,4 C) 1,2,3 D) 1,3,4 |
| GRPO (MATH). |
| MATH500 | HumanEval | GPQA | |
| Qwen2.5-7B | |||
| Base | 0.498 | 0.329 | 0.278 |
| Power Sampling (naive estimator) | 0.650 | 0.677 | 0.318 |
| Power Sampling (ours) | 0.708 | 0.756 | 0.349 |
| Qwen2.5-Math-7B | |||
| Base | 0.496 | 0.329 | 0.278 |