Where does output diversity collapse in post-training?
Not on the map. Why?
Abstract (arXiv)
Post-trained language models produce less varied outputs than their base counterparts. This output diversity collapse undermines inference-time scaling methods that rely on varied samples, and risks homogenizing model outputs on creative and value-laden tasks. Prior work attributes collapse to specific post-training methods, without separating the role of training data composition from the method, or the generation format from the model weights. We trace output diversity through three parallel post-training lineages of Olmo 3, Think (chain-of-thought distillation), Instruct (broad multi-source data), and RL-Zero, across 15 tasks and four text diversity metrics. We find that the location of collapse co-varies with data composition: the Think lineage loses most semantic diversity at supervised fine-tuning, and the effect of DPO is larger in Instruct than in Think. Suppressing chain-of-thought reasoning at inference in Think models drops accuracy on hard tasks, yet leaves answer-level diversity unchanged, showing that the collapse is embedded in the model weights by training data, not imposed by the generation format. Decomposing diversity loss on six verifiable tasks into a quality-control component (removal of incorrect outputs) and a residual component (genuine narrowing among correct outputs) reveals that the split is task-dependent, and Think models retain more correct-answer diversity than Instruct despite collapsing more in aggregate. Our results indicate that diversity collapse is determined during training by data composition and cannot be addressed at inference time alone.
Key facts. Cited by 11 papers in the map (data through 14 Oct 2026). Builds on 61 of its 69 references that are in the map. First posted on arXiv 17 Apr 2026.
Builds on
61 of 69 references.
References · 61
- 2022Training language models to follow instructions with human feedbackcited by 15,971
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,372
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,459
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2019Sentence-BERT: Sentence Embeddings using Siamese BERT-Networkscited by 9,170
- 2023Judging LLM-as-a-Judge with MT-Bench and Chatbot Arenacited by 8,626
- 2021Evaluating Large Language Models Trained on Codecited by 8,320
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,944
- 2022Self-Consistency Improves Chain of Thought Reasoning in Language Modelscited by 5,708
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
38 more
- 2021Program Synthesis with Large Language Modelscited by 3,637
- 2021TruthfulQA: Measuring How Models Mimic Human Falsehoodscited by 2,975
- 2018Don't Give Me the Details, Just the Summary! Topic-Aware Convolutional Neural Networks for Extreme Summarizationcited by 1,317
- 2023Instruction-Following Evaluation for Large Language Modelscited by 1,228
- 2018Hierarchical Neural Story Generationcited by 1,193
- 2024Tulu 3: Pushing Frontiers in Open Language Model Post-Trainingcited by 1,053
- 2025Does Reinforcement Learning Really Incentivize Reasoning Capacity in LLMs Beyond the Base Model?cited by 982
- 2016Abstractive Text Summarization Using Sequence-to-Sequence RNNs and Beyondcited by 812
- 2024WildChat: 1M ChatGPT Interaction Logs in the Wildcited by 629
- 2024From Crowdsourced Data to High-Quality Benchmarks: Arena-Hard and BenchBuilder Pipelinecited by 500
- 2023Understanding the Effects of RLHF on LLM Generalisation and Diversitycited by 396
- 2022UniXcoder: Unified Cross-Modal Pre-training for Code Representationcited by 361
- 2022The Vendi Score: A Diversity Evaluation Metric for Machine Learningcited by 311
- 2024CRUXEval: A Benchmark for Code Reasoning, Understanding and Executioncited by 300
- 2024To CoT or not to CoT? Chain-of-thought helps mainly on math and symbolic reasoningcited by 263
- 2024WildBench: Benchmarking LLMs with Challenging Tasks from Real Users in the Wildcited by 218
- 2024OpenMathInstruct-2: Accelerating AI for Math with Massive Open-Source Instruction Datacited by 200
- 2023Does Writing with Language Models Reduce Content Diversity?cited by 192
- 2025Reasoning Models Can Be Effective Without Thinkingcited by 187
- 2023Beyond Reverse KL: Generalizing Direct Preference Optimization with Diverse Divergence Constraintscited by 174
- 2024The PRISM Alignment Dataset: What Participatory, Representative and Individualised Human Feedback Reveals About the Subjective and Multicultural Alignment of Large Language Modelscited by 144
- 2025Artificial Hivemind: The Open-Ended Homogeneity of Language Models (and Beyond)cited by 141
- 2020Evaluating the Evaluation of Diversity in Natural Language Generationcited by 129
- 2024Homogenization Effects of Large Language Models on Human Creative Ideationcited by 127
- 2024MaxMin-RLHF: Alignment with Diverse Human Preferencescited by 123
- 2025Nemotron 3 Nano: Open, Efficient Mixture-of-Experts Hybrid Mamba-Transformer Model for Agentic Reasoningcited by 104
- 2025Jointly Reinforcing Diversity and Quality in Language Model Generationscited by 80
- 2024On the Algorithmic Bias of Aligning Large Language Models with RLHF: Preference Collapse and Matching Regularizationcited by 72
- 2024Preserving Diversity in Supervised Fine-Tuning of Large Language Modelscited by 70
- 2024Unintentional Unalignment: Likelihood Displacement in Direct Preference Optimizationcited by 67
- 2025Evaluating the Diversity and Quality of LLM Generated Contentcited by 51
- 2025Base Models Beat Aligned Models at Randomness and Creativitycited by 51
- 2025The Price of Format: Diversity Collapse in LLMscited by 43
- 2024Benchmarking Linguistic Diversity of Large Language Modelscited by 38
- 2024From Distributional to Overton Pluralism: Investigating Large Language Model Alignmentcited by 33
- 2025KL-Regularized Reinforcement Learning is Designed to Mode Collapsecited by 31
- 2022Semantic Diversity in Dialogue with Natural Language Inferencecited by 29
- 2025The Delta Learning Hypothesis: Preference Tuning on Weak Data can Yield Strong Gainscited by 28
50 most cited of 61 All 61 (JSON)
Built on by
Citing papers per year: 2026: 11.
11 papers.
- 2026Vector Policy Optimization: Training for Diversity Improves Test-Time Searchcited by 8
- 2026The One-Word Census: Answer-Choice Conformity Across 105 Language Modelscited by 3
- 2026Where Rollouts Begin: Low-Load, High-Leverage First-Token Diversification for RLVRcited by 2
- 2026Sixteen models, fewer than two voices: measuring ensemble dispersion where no answer is uniquely correctcited by 0
- 2026Locked at the Entrance, Open Inside: Where RLVR Narrows the Solution Spacecited by 0
- 2026Fine-Tuning Fixes Mode Collapse and Over-Dispersion in LLMscited by 0
- 2026What Does Chain-of-Thought Entropy Measure? A Channel Audit of Scaffolding, Routing, and Contentcited by 0
- 2026Strategically Diverse Sampling for Self-Trainingcited by 0
- 2026MaD-RL: Matching Distributions for Calibrating LLMs with Reinforcement Learningcited by 0
- 2026The Judge Is Not Its Twin: Post-training makes a model's writing more predictable but barely moves its taste, as a judge, toward predictable writingcited by 0
- 2026Gacha Decoding: Eliciting Diverse Generations Through Instruction Followingcited by 0
- 2026Gacha Decoding: Eliciting Diverse Generations Through Instruction Following1 Oct 2026
- 2026The Judge Is Not Its Twin: Post-training makes a model's writing more predictable but barely moves its taste, as a judge, toward predictable writing26 Sep 2026
- 2026Strategically Diverse Sampling for Self-Training25 Sep 2026
- 2026Fine-Tuning Fixes Mode Collapse and Over-Dispersion in LLMs15 Sep 2026
- 2026MaD-RL: Matching Distributions for Calibrating LLMs with Reinforcement Learning12 Sep 2026
- 2026Locked at the Entrance, Open Inside: Where RLVR Narrows the Solution Space29 Aug 2026
- 2026What Does Chain-of-Thought Entropy Measure? A Channel Audit of Scaffolding, Routing, and Content26 Aug 2026
- 2026Sixteen models, fewer than two voices: measuring ensemble dispersion where no answer is uniquely correct31 Jul 2026
- 2026The One-Word Census: Answer-Choice Conformity Across 105 Language Models14 Jul 2026
- 2026Where Rollouts Begin: Low-Load, High-Leverage First-Token Diversification for RLVR27 May 2026
- 2026Vector Policy Optimization: Training for Diversity Improves Test-Time Search21 May 2026
Similar, not cited
Close in wording, no citation either way
- 2026Language of Thought Shapes Output Diversity in Large Language Models
- 2025On the Effect of Sampling Diversity in Scaling LLM Inference
- 2023Diversity of Thought Improves Reasoning Abilities of LLMs
- 2025How Instruction and Reasoning Data shape Post-Training: Data Quality through the Lens of Layer-wise Gradients
- 2025Diverse, not Short: A Length-Controlled Data Selection Strategy for Improving Response Diversity of Language Models
- 2025Epistemic diversity across language models mitigates knowledge collapse
- 2025Modifying Large Language Model Post-Training for Diverse Creative Writing
- 2025Prismatic Synthesis: Gradient-based Data Diversification Boosts Generalization in LLM Reasoning