arXiv:2609.02275cs.CLcs.AI2026-09

发现大模型生成内容多样性低于训练数据,提出可扩展的补救方法

Do Large Language Models Capture the Diversity in their Training Data?

论文配图:Do Large Language Models Capture the Diversity in their Training Data?
图 1 · 摘自论文原文
  • 用信息论测条件熵,无须多参考输出即可评估生成多样性
  • 跨模型、长度、解码策略,模型输出熵均低于训练数据
  • 提出矩阵熵重加权法,在不偏离原分布前提下提升多样性

大型语言模型虽以建模文本条件分布为目标,但其是否充分捕捉训练数据中的潜在多样性仍不明确。本文通过信息论视角,比较模型生成输出与对应训练数据的条件熵。基于配对输入-输出样本,利用条件熵及其基于冯诺依曼熵的矩阵形式,衡量在给定输入下输出的变异性,无需多个参考输出。在公开训练数据的OLMo、Pythia和GPT-Neo等模型族中,无论模型规模、序列长度或解码策略,模型生成输出的条件熵始终低于训练数据。该多样性差距也存在于语言建模之外的场景,如类条件图像生成器和文本条件的MS-COCO模型。为此,我们提出后处理校正机制:为每个输入生成多组输出,并通过矩阵熵投影进行重加权,提升条件多样性,同时保持接近原始模型分布。我们证明了矩阵条件熵泛函的凹性,使熵约束投影成为凸优化问题,并开发了可扩展的镜面下降算法实现。结果揭示现代生成模型与训练数据间存在系统性条件多样性缺口,并提供了一个可测量与缓解该缺口的信息理论框架。

原文摘要 · Abstract (English)

Large language models are trained to model conditional distributions over text, yet it remains inadequately understood whether they capture the full diversity of plausible outputs present in their training data. We study this question through an information-theoretic lens by comparing the conditional entropy of model-generated outputs with that of the corresponding training data. Given paired input-output samples, we use conditional entropy and its matrix-based analogue based on von Neumann entropy to measure output variability beyond what is explained by the conditioning input, without requiring multiple reference outputs for the same prompt. Across LLM families with publicly available training data, including OLMo, Pythia, and GPT-Neo, we consistently find that model-generated outputs exhibit lower conditional entropy than their training data, across different model scales, sequence lengths, and decoding strategies. We observe a similar conditional diversity gap beyond language modeling, including class-conditioned ImageNet generators and text-conditioned models trained on MS-COCO. To address this gap, we propose a post-hoc correction mechanism that generates multiple outputs for each input and reweights them through a matrix-entropy projection, increasing conditional diversity while remaining close to the original model distribution. We prove the concavity of the matrix-based conditional entropy functional, which makes the resulting entropy-constrained projection a convex optimization problem, and develop a scalable mirror-descent algorithm for its implementation. Our results reveal a systematic conditional diversity gap between modern generative models and their training data, and provide an information-theoretic framework for measuring and mitigating this gap.

大模型多样性信息论生成模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。