PowerFlow: Unlocking the Dual Nature of LLMs via Principled Distribution Matching
Not on the map. Why?
Abstract (arXiv)
Unsupervised Reinforcement Learning from Internal Feedback (RLIF) has emerged as a promising paradigm for eliciting the latent capabilities of Large Language Models (LLMs) without external supervision. However, current methods rely on heuristic intrinsic rewards, which often lack a well-defined theoretical optimization target and are prone to degenerative biases. In this work, we introduce PowerFlow, a principled framework that reformulates unsupervised fine-tuning as a distribution matching problem. By casting GFlowNet as an amortized variational sampler for unnormalized densities, we propose a length-aware Trajectory-Balance objective that explicitly neutralizes the structural length biases inherent in autoregressive generation. By targeting α-power distributions, PowerFlow enables the directional elicitation of the dual nature of LLMs: sharpening the distribution (α > 1) to intensify logical reasoning, or flattening it (α < 1) to unlock expressive creativity. Extensive experiments demonstrate that PowerFlow consistently outperforms existing RLIF methods, matching or even exceeding supervised GRPO. Furthermore, by mitigating over-sharpening in aligned models, our approach achieves simultaneous gains in diversity and quality, shifting the Pareto frontier in creative tasks.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 38 of its 46 references that are in the map. First posted on arXiv 19 Mar 2026.
Builds on
38 of 46 references.
References · 38
- 2022Training language models to follow instructions with human feedbackcited by 15,971
- 2015Distilling the Knowledge in a Neural Networkcited by 10,324
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
- 2023GPQA: A Graduate-Level Google-Proof Q&A Benchmarkcited by 2,879
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,810
- 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 2,057
- 2024OlympiadBench: A Challenging Benchmark for Promoting AGI with Olympiad-Level Bilingual Multimodal Scientific Problemscited by 1,136
- 2025Does Reinforcement Learning Really Incentivize Reasoning Capacity in LLMs Beyond the Base Model?cited by 982
- 2025DeepSeek-V3.2: Pushing the Frontier of Open Large Language Modelscited by 815
- 2024Self-Rewarding Language Modelscited by 681
- 2021Flow Network based Generative Models for Non-Iterative Diverse Candidate Generationcited by 392
26 more
- 2023Detecting Pretraining Data from Large Language Modelscited by 387
- 2022Large Language Models are Better Reasoners with Self-Verificationcited by 307
- 2025TTRL: Test-Time Reinforcement Learningcited by 275
- 2022Trajectory balance: Improved credit assignment in GFlowNetscited by 251
- 2025Learning to Reason without External Rewardscited by 210
- 2025Spurious Rewards: Rethinking Training Signals in RLVRcited by 209
- 2025Right Question is Already Half the Answer: Fully Unsupervised LLM Reasoning Incentivizationcited by 128
- 2025Verbalized Sampling: How to Mitigate Mode Collapse and Unlock LLM Diversitycited by 105
- 2024Self-Improvement in Language Models: The Sharpening Mechanismcited by 104
- 2025Maximizing Confidence Alone Improves Reasoningcited by 100
- 2025Reasoning with Sampling: Your Base Model is Smarter Than You Thinkcited by 89
- 2023EQ-Bench: An Emotional Intelligence Benchmark for Large Language Modelscited by 80
- 2025Can Large Reasoning Models Self-Train?cited by 68
- 2025Confidence Is All You Need: Few-Shot RL Fine-Tuning of Language Modelscited by 62
- 2025Diverse Preference Optimizationcited by 52
- 2025Base Models Beat Aligned Models at Randomness and Creativitycited by 51
- 2024AI as Humanity's Salieri: Quantifying Linguistic Creativity of Language Models via Systematic Attribution of Machine Text against Web Textcited by 42
- 2025FlowRL: Matching Reward Distributions for LLM Reasoningcited by 40
- 2022A Variational Perspective on Generative Flow Networkscited by 39
- 2025No Free Lunch: Rethinking Internal Feedback for LLM Reasoningcited by 37
- 2025One-shot Entropy Minimizationcited by 32
- 2026How Far Can Unsupervised RLVR Scale LLM Training?cited by 30
- 2025LLM Probability Concentration: How Alignment Shrinks the Generative Horizoncited by 26
- 2025Which LLMs Get the Joke? Probing Non-STEM Reasoning Abilities with HumorBenchcited by 8
- 2023Using Semantic Similarity and Text Embedding to Measure the Social Media Echo of Strategic Communicationscited by 4
- 2026PRISM: A Unified Framework for Post-Training LLMs Without Verifiable Rewardscited by 4
Built on by
Citing papers per year: 2026: 2.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2026Scalable Power Sampling: Unlocking Efficient, Training-Free Reasoning for LLMs via Distribution Sharpening
- 2026GFlowRL: Scaling Distribution-Matching RL to Large Language Models
- 2023Amortizing intractable inference in large language models
- 2026Power Distribution Bridges Sampling, Self-Reward RL, and Self-Distillation
- 2025Enhancing Reasoning for Diffusion LLMs via Distribution Matching Policy Optimization
- 2026LLMs as High-Dimensional Nonlinear Autoregressive Models with Attention: Training, Alignment and Inference
- 2026MaD-RL: Matching Distributions for Calibrating LLMs with Reinforcement Learning
- 2026Sparse but Critical: A Token-Level Analysis of Distributional Shifts in RLVR Fine-Tuning of LLMs