探究大模型后训练中输出多样性下降的根本原因
Where does output diversity collapse in post-training?
- 通过三类后训练路径对比,发现多样性损失由训练数据决定
- 思维链训练导致语义多样性在微调阶段大幅下降
- 模型权重已固化多样性损失,推理时无法修复
后训练语言模型的输出多样性低于基础模型,削弱了依赖多样样本的推理扩展方法,并可能导致创造性与价值导向任务的输出同质化。以往研究将此归因于特定后训练方法,但未区分训练数据构成与方法、生成格式与模型权重之间的关系。本文沿Olmo 3的三条并行后训练路径(Think、Instruct、RL-Zero)在15项任务上分析四类文本多样性指标,发现多样性崩溃位置与数据构成密切相关:Think路径在监督微调阶段丧失最多语义多样性,且DPO对Instruct的影响大于Think。在Think模型中抑制推理时的思维链会降低难题准确率,但答案级多样性不变,表明崩溃源于训练数据对模型权重的固化,而非生成格式。将六项可验证任务的多样性损失分解为质量控制(剔除错误输出)与残余部分(正确输出间的真正收敛),结果表明该比例因任务而异;尽管Think整体多样性更低,但其正确答案多样性仍高于Instruct。结论:多样性崩溃由训练期间的数据构成决定,无法仅通过推理阶段调整解决。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。