语言模型生成多样性的瓶颈在于概率分配不准,导致输出变窄。
Sampling More, Getting Less: Calibration is the Diversity Bottleneck in LLMs

- 从概率分布角度分析推理时的错误分配,揭示多样性崩溃根源
- 14个模型实测显示,有效词排序和分布形状均存在校准偏差
- 适合研究生成质量、提示工程与模型评估的开发者参考
多样性对语言模型的应用(如创意生成、科学发现)至关重要,但现代大模型常陷入有限的输出子集。尽管已有方法衡量多样性不足,但对推理过程中逐步的概率分布如何导致问题仍不清楚。本文提出有效性-多样性框架,将多样性坍缩归因于模型在解码时对有效与无效延续的分布分配不当。该框架分解出两种互补的校准偏差:一是顺序校准问题,有效词未可靠排在无效词之上,导致基于排名的截断规则需在恢复有效输出与排除无效内容间权衡;二是形状校准问题,概率质量过度集中在少数有效延续上,同时存在大量混合有效与无效的尾部项,保持高有效性会限制多样性。我们形式化了这两种机制,并证明局部误差在解码步骤中累积,造成显著的序列级多样性损失。通过设计可控诊断任务(含已知有效集合和最优截断基线),我们在14个不同家族和规模的语言模型上验证,多样性坍缩并非特定采样策略的局限,而是由模型分布中的顺序与形状校准偏差所致。
原文摘要 · Abstract (English)
Diversity is essential for language-model applications ranging from creative generation to scientific discovery, yet modern LLMs often collapse into a narrow subset of plausible outputs. While prior work has developed benchmarks for measuring this lack of diversity, less is known about how the step-by-step probability distributions at inference time cause the problem. We introduce a validity--diversity framework that attributes diversity collapse to how an LLM allocates probability mass across valid and invalid continuations during decoding. This framework decomposes the bottleneck into two complementary forms of miscalibration. First, order calibration: valid tokens are not reliably ranked above invalid tokens, so rank-based cutoff rules must trade off between recovering valid continuations and admitting invalid ones. Second, shape calibration: probability mass is overly concentrated only on few valid continuations while having a heavy-tail of mixed valid and invalid tokens, so maintaining high validity limits diversity. We formalize both mechanisms and show that local failures compound across decoding steps, producing strong sequence-level losses in diversity. Empirically, we develop controlled diagnostics for probing these bottlenecks, including tasks with exactly known valid sets and oracle cutoff baselines. Across 14 language models spanning multiple families and scales, we find that diversity collapse is not merely a limitation of particular sampling heuristics, but a consequence of order and shape miscalibration in the LLM distribution.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。