arXiv:2605.06366cs.LG2026-05被引 1

发现扩散语言模型早期层存在激活坍缩现象,关键异常值对生成质量至关重要。

Layer Collapse in Diffusion Language Models

论文配图:Layer Collapse in Diffusion Language Models
图 1 · 摘自论文原文
  • 早期层激活模式高度相似且被单一大异常值主导,形成坍缩结构。
  • 移除该异常值导致输出陷入重复随机循环,但模型仍能承受3比特量化压缩。
  • 与自回归模型相反,越早的层冗余度越高,适合将稀疏化资源分配给早期层。

扩散语言模型(DLMs)作为自回归语言模型的竞争者近期涌现,但其激活动态差异仍不清晰。本文分析了LLaDA-8B的激活特性,发现早期几层存在显著的层坍缩现象:少数早期层的激活模式高度相似,由一个长期持续的单一超大异常值主导。尽管看似冗余,该异常值至关重要——剪枝后模型输出会退化为重复的随机词元循环。反直觉的是,DLM中整体冗余度更高,且早期层冗余最明显,这与自回归模型因欠训练导致深层冗余的现象相反。分析表明,这种坍缩并非由欠训练引起,而是过训练所致:主导异常值成为不可或缺的信息载体,其余表示则坍缩为冗余结构。这一发现具有重要实践意义,经受控预训练实验验证。在压缩方面,LLaDA在3比特GPTQ量化下仅在GSM8K上下降1.8%,而Llama-3.1-8B下降64.7%。最优稀疏分配策略也反转:在50%平均稀疏度下,将更多稀疏度分配给早期层可使LLaDA提升8.4%,而相同策略会使Llama下降8.4%。结果表明,DLM的训练目标从根本上重塑了层动态,对压缩与部署有直接影响。代码见github.com/Conzel/super-outlier-dlm。

原文摘要 · Abstract (English)

Diffusion language models (DLMs) have recently emerged as competitive alternatives to autoregressive (AR) language models, yet differences in their activation dynamics remain poorly understood. We characterize these dynamics in LLaDA-8B and identify a striking layer-collapse property: a few early layers exhibit highly similar, collapsed activation patterns dominated by a single large super-outlier persisting over a long token range. Despite its apparent redundancy, this outlier is critical: pruning it causes outputs to degrade into repetitive random token loops. Paradoxically, layers in LLaDA contain more redundant representations overall, with redundancy most pronounced in earlier layers -- the reverse of AR models, where deeper layers grow redundant due to undertraining. Our analysis indicates that layer collapse in DLMs is not driven by undertraining but by overtraining: a dominant outlier becomes an indispensable information carrier while remaining representations collapse into redundant structure. These findings have strong practical implications, verified through controlled pre-training experiments. DLMs are surprisingly robust to compression: LLaDA under 3-bit GPTQ quantization drops only -1.8% on GSM8K, whereas Llama-3.1-8B drops -64.7%. Optimal sparsity allocation also reverses between families: at 50% average sparsity, allocating more to early layers in LLaDA yields +8.4% over the reverse strategy, while the same allocation costs Llama -8.4%. Our findings reveal that the DLM training objective fundamentally reshapes layer dynamics relative to AR models, with direct consequences for compression and deployment. Code: github.com/Conzel/super-outlier-dlm.

扩散模型激活分析模型压缩层坍缩

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