mlsanityData to 14 Oct 2026

LLM Probability Concentration: How Alignment Shrinks the Generative Horizon

arXiv 2506.17871 ↗First posted 22 Jun 2025cs.CL cs.AI cs.LGcited by 26

On the map: LLM alignment / RLHF / preference optimisation → LLM preference optimization methods

Abstract (arXiv)

Despite their impressive capabilities, aligned large language models (LLMs) often generate outputs that lack diversity. What drives this consistency in the generation? We investigate this phenomenon through the lens of probability concentration in the model's output distribution. To quantify it, we use the Branching Factor (BF)--the exponentiated length-averaged entropy of the output distribution, interpreted as the effective number of plausible next steps during generation. Our empirical analysis reveals two key findings: (1) BF often decreases as generation progresses, suggesting that LLMs become more predictable; a controlled intervention indicates that this decline is largely a task-independent property of autoregressive self-conditioning, distinct from alignment, and can be locally reversed by unexpected context. (2) Alignment tuning sharpens the output distribution from the outset, reducing BF by a factor of 2--5 overall and up to an order of magnitude (e.g., from 12 to 1.2) at early positions. This reduction helps explain why aligned models are less sensitive to decoding strategies. It also has implications for complex reasoning: aligned Chain-of-Thought (CoT) models (e.g., DeepSeek-distilled models) generate longer reasoning chains that reach later, more deterministic (lower-BF) stages, yielding more stable outputs. We hypothesize that alignment does not fundamentally change model behavior, but instead steers the model toward stylistic tokens (e.g., ``Sure'') that unlock low-entropy trajectories already present in the base model. Nudging experiments support this view: prompting base models with such tokens similarly reduces BF. Together, our findings establish BF as a diagnostic for understanding and controlling LLM outputs, clarifying how alignment reduces variability, CoT stabilizes generation, and base models can be steered away from diversity.

Key facts. Cited by 26 papers in the map (data through 14 Oct 2026). Placed in LLM preference optimization methods, one of 10 directions in LLM alignment / RLHF / preference optimisation. Builds on 70 of its 95 references that are in the map. First posted on arXiv 22 Jun 2025.

Builds on

70 of 95 references.

Closest ancestors · 7

Same direction, newest first

  1. 2025Base Models Beat Aligned Models at Randomness and Creativity30 Apr
  2. 2025Diverse Preference Optimization30 Jan
  3. 2024GDPO: Learning to Directly Align Language Models with Diversity Using GFlowNets19 Oct
  4. 2024From Distributional to Overton Pluralism: Investigating Large Language Model Alignment25 Jun
  5. 2023The Unlocking Spell on Base LLMs: Rethinking Alignment via In-Context Learning4 Dec
  6. 2023Beyond Reverse KL: Generalizing Direct Preference Optimization with Diverse Divergence Constraints28 Sep
  7. 2022Training language models to follow instructions with human feedback4 Mar

Other references · 63

  1. 2024The Llama 3 Herd of Modelscited by 14,715
  2. 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 13,028
  3. 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,459
  4. 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
  5. 2020Measuring Massive Multitask Language Understandingcited by 7,963
  6. 2022Training a Helpful and Harmless Assistant with Reinforcement Learning from Human Feedbackcited by 3,720
  7. 2023Tree of Thoughts: Deliberate Problem Solving with Large Language Modelscited by 3,211
  8. 2019The Curious Case of Neural Text Degenerationcited by 2,916
  9. 2024Qwen2 Technical Reportcited by 2,120
  10. 2022Language Models (Mostly) Know What They Knowcited by 1,856
  11. 2018Don't Give Me the Details, Just the Summary! Topic-Aware Convolutional Neural Networks for Extreme Summarizationcited by 1,317
  12. 2015A Diversity-Promoting Objective Function for Neural Conversation Modelscited by 1,077
38 more
  1. 2022Red Teaming Language Models with Language Modelscited by 1,026
  2. 2023Can LLMs Express Their Uncertainty? An Empirical Evaluation of Confidence Elicitation in LLMscited by 974
  3. 2023Reasoning with Language Model is Planning with World Modelcited by 861
  4. 2024Medusa: Simple LLM Inference Acceleration Framework with Multiple Decoding Headscited by 843
  5. 2023Semantic Uncertainty: Linguistic Invariances for Uncertainty Estimation in Natural Language Generationcited by 819
  6. 2025Beyond the 80/20 Rule: High-Entropy Minority Tokens Drive Effective Reinforcement Learning for LLM Reasoningcited by 559
  7. 2025The Entropy Mechanism of Reinforcement Learning for Reasoning Language Modelscited by 485
  8. 2020How Can We Know When Language Models Know? On the Calibration of Language Models for Question Answeringcited by 445
  9. 2023Understanding the Effects of RLHF on LLM Generalisation and Diversitycited by 396
  10. 2023Detecting Pretraining Data from Large Language Modelscited by 387
  11. 20242 OLMo 2 Furiouscited by 340
  12. 2022Language Models Are Greedy Reasoners: A Systematic Formal Analysis of Chain-of-Thoughtcited by 337
  13. 2020Calibration of Pre-trained Transformerscited by 315
  14. 2023Alphazero-like Tree-Search can Guide Large Language Model Decoding and Trainingcited by 295
  15. 2025Reasoning with Exploration: An Entropy Perspectivecited by 281
  16. 2024RLHF Workflow: From Reward Modeling to Online RLHFcited by 267
  17. 2022The Parallelism Tradeoff: Limitations of Log-Precision Transformerscited by 193
  18. 2023Does Writing with Language Models Reduce Content Diversity?cited by 192
  19. 2024Aya Dataset: An Open-Access Collection for Multilingual Instruction Tuningcited by 177
  20. 2023Art or Artifice? Large Language Models and the False Promise of Creativitycited by 172
  21. 2024The Effect of Sampling Temperature on Problem Solving in Large Language Modelscited by 170
  22. 2024From Decoding to Meta-Generation: Inference-time Algorithms for Large Language Modelscited by 137
  23. 2020Evaluating the Evaluation of Diversity in Natural Language Generationcited by 129
  24. 2024Benchmarking LLMs via Uncertainty Quantificationcited by 126
  25. 2022Truncation Sampling as Language Model Desmoothingcited by 123
  26. 2023Future Lens: Anticipating Subsequent Tokens from a Single Hidden Statecited by 121
  27. 2024Language Model Cascades: Token-level uncertainty and beyondcited by 121
  28. 2024The Good, The Bad, and The Greedy: Evaluation of LLMs Should Not Ignore Non-Determinismcited by 116
  29. 2024A Thorough Examination of Decoding Methods in the Era of LLMscited by 114
  30. 2019Evaluating the State-of-the-Art of End-to-End Natural Language Generation: The E2E NLG Challengecited by 86
  31. 2024Are Large Language Models Capable of Generating Human-Level Narratives?cited by 79
  32. 2021Revisiting the Uniform Information Density Hypothesiscited by 68
  33. 2024LLM Discussion: Enhancing the Creativity of Large Language Models via Discussion Framework and Role-Playcited by 66
  34. 2023LatestEval: Addressing Data Contamination in Language Model Evaluation through Dynamic and Time-Sensitive Test Constructioncited by 58
  35. 2024One fish, two fish, but not the whole sea: Alignment reduces language models' conceptual diversitycited by 58
  36. 2024Do language models plan ahead for future tokens?cited by 49
  37. 2024AI as Humanity's Salieri: Quantifying Linguistic Creativity of Language Models via Systematic Attribution of Machine Text against Web Textcited by 42
  38. 2025Modifying Large Language Model Post-Training for Diverse Creative Writingcited by 40

50 most cited of 63 All 63 (JSON)

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2025: 7; 2026: 19.

26 papers.

Sort:

By area:

  1. 2025KL-Regularized Reinforcement Learning is Designed to Mode Collapsecited by 31
  2. 2025Entropy After </Think> for reasoning model early exitingcited by 24
  3. 2025Random Policy Valuation is Enough for LLM Reasoning with Verifiable Rewardscited by 10
  4. 2025Let it Calm: Exploratory Annealed Decoding for Verifiable Reinforcement Learningcited by 10
  5. 2025Don't Throw Away Your Pretrained Modelcited by 10
  6. 2025Optimizing Diversity and Quality through Base-Aligned Model Collaborationcited by 8
  7. 2026The Chameleon's Limit: Investigating Persona Collapse and Homogenization in Large Language Modelscited by 8
  8. 2025Mixture-of-Minds: Multi-Agent Reinforcement Learning for Table Understandingcited by 7
  9. 2026Reaching Beyond the Mode: RL for Distributional Reasoning in Language Modelscited by 6
  10. 2026EfficientRollout: System-Aware Self-Speculative Decoding for RL Rolloutscited by 6
  11. 2026One Frozen Simulator Is Not Enough: Simulator Collapse in Multi-Agent RLcited by 6
  12. 2026The One-Word Census: Answer-Choice Conformity Across 105 Language Modelscited by 3
14 more
  1. 2026The Homogenization Problem in LLMs: Towards Meaningful Diversity in AI Safetycited by 2
  2. 2026PowerFlow: Unlocking the Dual Nature of LLMs via Principled Distribution Matchingcited by 2
  3. 2026Process Supervision of Confidence Margin for Calibrated LLM Reasoningcited by 2
  4. 2026Sampling More, Getting Less: Calibration is the Diversity Bottleneck in LLMscited by 2
  5. 2026Using Reward Uncertainty to Induce Diverse Behaviour in Reinforcement Learningcited by 2
  6. 2026Communication Enhances LLMs' Stability in Strategic Thinkingcited by 1
  7. 2026DMAP: A Distribution Map for Textcited by 1
  8. 2026Not All Denoising Steps Are Equal: Model Scheduling for Faster Masked Diffusion Language Modelscited by 1
  9. 2026Can You Keep a Secret? Involuntary Information Leakage in Language Model Writingcited by 1
  10. 2026How Language Models Fail: Token-Level Signatures of Committed and Persistent Reasoning Failurescited by 1
  11. 2026Towards Physical Intuitions for Alignment Dynamics: A Case Study With Randomness Crystallizationcited by 1
  12. 2026Iterative Finetuning is Mostly Idempotentcited by 0
  13. 2026Fine-Tuning Improves Information Conveyance in Language Modelscited by 0
  14. 2026Rethinking Probability-Based Reinforcement Learning From Posterior Concentrationcited by 0

Similar, not cited

Close in wording, no citation either way

  1. 2026Beyond Surface Alignment: Grounding the Dynamics of Situational Understanding and Generative Control in LLMs
  2. 2026Controlling Distributional Bias in Multi-Round LLM Generation via KL-Optimized Fine-Tuning
  3. 2026LLMs as High-Dimensional Nonlinear Autoregressive Models with Attention: Training, Alignment and Inference
  4. 2024Your Weak LLM is Secretly a Strong Teacher for Alignment
  5. 2026Recipes for Steering and Scaling LLMs via Sampling
  6. 2024A Probabilistic Perspective on Unlearning and Alignment for Large Language Models
  7. 2026How Sampling Shapes LLM Alignment: From One-Shot Optima to Iterative Dynamics
  8. 2024Creativity Has Left the Chat: The Price of Debiasing Language Models