Power Distribution Bridges Sampling, Self-Reward RL, and Self-Distillation
cs.LG
Submitted: 2026-05-06
Updated: 2026-10-19
Comments: NeurIPS 2026
Code: https://github.com/tom4649/power-self-distillation
License: http://creativecommons.org/licenses/by/4.0/
The gist: Recent analyses question whether reinforcement learning (RL) is responsible for strong reasoning in large language models (LLMs).
Terminology
Abstract
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, that downstream true-reward gains are governed by the covariance between true reward and self-reward under the power distribution, and that this ideal power-teacher gain transfers to a distilled student when the student is close to the power teacher. 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. The code is available at https://github.com/tom4649/power-self-distillation.
Sources
- Phi-3 Technical Report: A Highly Capable Language Model Locally on Your Phone
- Program Synthesis with Large Language Models
- Power-SMC: Low-Latency Sequence-Level Power Sampling for Training-Free LLM Reasoning
- Evaluating Large Language Models Trained on Code
- Helping or Herding? Reward Model Ensembles Mitigate but do not Eliminate Reward Hacking
- DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learning
- Distilling the Knowledge in a Neural Network
- Scalable Power Sampling: Unlocking Efficient, Training-Free Reasoning for LLMs via Distribution Sharpening
- Tulu 3: Pushing Frontiers in Open Language Model Post-Training
- Reinforcement Learning and Control as Probabilistic Inference: Tutorial and Review
- Decoupled Weight Decay Regularization
- WebGPT: Browser-assisted question-answering with human feedback
- Maximizing Confidence Alone Improves Reasoning
- Spurious Rewards: Rethinking Training Signals in RLVR
- DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Models
- Scaling LLM Test-Time Compute Optimally can be More Effective than Scaling Model Parameters
- Llama 2: Open Foundation and Fine-Tuned Chat Models
- Qwen2.5 Technical Report
- Qwen2.5-Math Technical Report: Toward Mathematical Expert Model via Self-Improvement
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