Power Distribution Bridges Sampling, Self-Reward RL, and Self-Distillation
On the map: LLM reasoning → Reinforcement learning for reasoning
also in 1 more
Abstract (arXiv)
Recent analyses question whether reinforcement learning (RL) is responsible for strong reasoning in large language models (LLMs). At the same time, distillation and inference-time sampling, including power sampling, have emerged as effective ways to improve LLM performance. However, the relationship among RL, distillation, and sampling remains unclear. In this study, we focus on the power distribution, the target distribution of power sampling, and show that the power distribution bridges sampling, self-reward KL-regularized RL, and self-distillation. From the sampling perspective, we show that inexpensive local approximations cannot reproduce sequence-level power without information about possible suffixes. From the RL perspective, the power distribution is the closed-form optimizer of KL-regularized RL when the model's sequence-level log-probabilities are used as the reward. This identification leads to power self-distillation, an offline distillation surrogate that shares the same target distribution and amortizes the cost of power sampling into supervised training on teacher samples. We further show that power self-distillation can achieve self-reward sharpening, while improvement in a downstream true reward is governed by the covariance between true reward and self-reward under the power distribution. Experiments on reasoning tasks support our analysis: power sampling raises self-reward, true-reward gains depend on alignment with self-reward, and power self-distillation can match or exceed the performance of power sampling at much lower inference cost.
Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Placed in Reinforcement learning for reasoning, one of 17 directions in LLM reasoning. Builds on 40 of its 51 references that are in the map. First posted on arXiv 6 May 2026.
Builds on
40 of 51 references.
Closest ancestors · 8
Same direction, newest first
- 2026Scalable Power Sampling: Unlocking Efficient, Training-Free Reasoning for LLMs via Distribution Sharpening29 Jan
- 2025Spurious Rewards: Rethinking Training Signals in RLVR12 Jun
- 2025Rewarding the Unlikely: Lifting GRPO Beyond Distribution Sharpening3 Jun
- 2025Maximizing Confidence Alone Improves Reasoning28 May
- 2025Learning to Reason without External Rewards26 May
- 2025Does Reinforcement Learning Really Incentivize Reasoning Capacity in LLMs Beyond the Base Model?18 Apr
- 2024Tulu 3: Pushing Frontiers in Open Language Model Post-Training22 Nov
- 2023Let's Verify Step by Step31 May
Other references · 32
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
- 2015Distilling the Knowledge in a Neural Networkcited by 10,076
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2021Evaluating Large Language Models Trained on Codecited by 8,053
- 2024Qwen2.5 Technical Reportcited by 4,713
- 2021Program Synthesis with Large Language Modelscited by 3,509
- 2023GPQA: A Graduate-Level Google-Proof Q&A Benchmarkcited by 2,748
- 2022Self-Instruct: Aligning Language Models with Self-Generated Instructionscited by 2,674
- 2020Learning to summarize from human feedbackcited by 2,562
- 2024Scaling LLM Test-Time Compute Optimally can be More Effective than Scaling Model Parameterscited by 1,928
- 2024Phi-3 Technical Report: A Highly Capable Language Model Locally on Your Phonecited by 1,883
20 more
- 2021WebGPT: Browser-assisted question-answering with human feedbackcited by 1,722
- 2024Qwen2.5-Math Technical Report: Toward Mathematical Expert Model via Self-Improvementcited by 1,057
- 2022Scaling Laws for Reward Model Overoptimizationcited by 1,056
- 2022Large Language Models Can Self-Improvecited by 656
- 2018Reinforcement Learning and Control as Probabilistic Inference: Tutorial and Reviewcited by 466
- 2015Policy Distillationcited by 414
- 2017Distral: Robust Multitask Reinforcement Learningcited by 231
- 2023Helping or Herding? Reward Model Ensembles Mitigate but do not Eliminate Reward Hackingcited by 195
- 2022In-context Reinforcement Learning with Algorithm Distillationcited by 186
- 2024From Decoding to Meta-Generation: Inference-time Algorithms for Large Language Modelscited by 136
- 2023Controlled Decoding from Language Modelscited by 133
- 2024BoNBoN Alignment for Large Language Models and the Sweetness of Best-of-n Samplingcited by 120
- 2024Self-Improvement in Language Models: The Sharpening Mechanismcited by 102
- 2024Probabilistic Inference in Language Models via Twisted Sequential Monte Carlocited by 85
- 2016Reward Augmented Maximum Likelihood for Neural Structured Predictioncited by 80
- 2024BOND: Aligning LLMs with Best-of-N Distillationcited by 73
- 2024Variational Best-of-N Alignmentcited by 49
- 2024InfAlign: Inference-aware language model alignmentcited by 34
- 2026Power-SMC: Low-Latency Sequence-Level Power Sampling for Training-Free LLM Reasoningcited by 17
- 2024Faster WIND: Accelerating Iterative Best-of-N Distillation for LLM Alignmentcited by 9
Built on by
Citing papers per year: 2026: 3.
3 papers.
Similar, not cited
Close in wording, no citation either way
- 2026The Two-Stage Decision-Sampling Hypothesis: Understanding the Emergence of Self-Reflection in RL-Trained LLMs
- 2026Sparse but Critical: A Token-Level Analysis of Distributional Shifts in RLVR Fine-Tuning of LLMs
- 2026Sample Where You Struggle: Sharpening Base Model Reasoning via Entropy-Guided Power Sampling
- 2026Rethinking RL for LLM Reasoning: It's Sparse Policy Selection, Not Capability Learning
- 2026Adaptive Power Sampling for LLM Reasoning
- 2026More Correct Mass, Worse Answers: Why Power Sampling Can Fail and How to Fix It
- 2026Sharpen Without Search: On-Policy Distillation of Sequence-Level Power Distribution
- 2026PowerOPD: Stabilizing On-Policy Distillation with Bounded Power Transformation