arXiv:2604.23681cs.LGcs.CL2026-04被引 1

揭示Transformer中四种表征坍塌机制,提出对称性破缺新视角

Rank, Head-Channel Non-Identifiability, and Symmetry Breaking: A Precise Analysis of Representational Collapse in Transformers

论文配图:Rank, Head-Channel Non-Identifiability, and Symmetry Breaking: A Precise Analysis of Representational Collapse in Transformers
图 1 · 摘自论文原文
  • 证明层归一化不改变表示的仿射秩,修正了传统认知
  • 发现残差连接可抑制秩坍塌,而MLP真正作用是生成新特征方向
  • 提出位置门控输出投影,解决头通道不可识别问题,参数开销低于1.6%

Dong等(2021)指出仅用自注意力构建的Transformer会迅速发生秩坍塌:所有标记表示收敛至单一方向。该研究提出以MLP作为补救。本文表明,尽管此结论在原研究范围内成立,但其图景并不完整,对架构理解有重要影响。首先,层归一化精确保持标记表示集的仿射秩,所谓“无作用”说法不准确;其次,残差连接在测度论意义上普遍阻碍秩坍塌,无需依赖MLP;MLP的核心功能在于生成原始嵌入线性空间之外的特征方向,这是多层注意力无法实现的。第三,识别出一种区别于秩坍塌的现象:头通道不可识别性。多头注意力各头输出经输出投影相加后,个体贡献无法唯一归属,每层存在n(H-1)d_k个自由度模糊。由于MLP作用于求和后的信号,无法修复此问题。本文提出构造性部分解法:位置门控输出投影(PG-OP),参数开销低于标准输出投影的1.6%。文献中已知的四种坍塌现象——深度、宽度上的秩坍塌,头通道不可识别性,熵坍塌——均被统一于对称性破缺框架下,各自对应前向传播的不同对称性。

原文摘要 · Abstract (English)

A widely cited result by Dong et al. (2021) showed that Transformers built from self-attention alone, without skip connections or feed-forward layers, suffer from rapid rank collapse: all token representations converge to a single direction. The proposed remedy was the MLP. We show that this picture, while correct in the regime studied by Dong, is incomplete in ways that matter for architectural understanding. Three results are established. First, layer normalisation is precisely affine-rank-neutral: it preserves the affine rank of the token representation set exactly. The widespread claim that LN "plays no role" is imprecise; the correct statement is sharper. Second, residual connections generically obstruct rank collapse in real Transformers such as BERT-base, in a measure-theoretic sense, without contribution from the MLP. The MLP's irreplaceable function is different: generating feature directions outside the linear span of the original token embeddings, which no stack of attention layers can produce. Third, a phenomenon distinct from rank collapse is identified: head-channel non-identifiability. After multi-head attention sums per-head outputs through the output projection, individual contributions cannot be canonically attributed to a specific head; n(H-1)d_k degrees of freedom per layer remain ambiguous when recovering a single head from the mixed signal. The MLP cannot remedy this because it acts on the post-summation signal. A constructive partial remedy is proposed: a position-gated output projection (PG-OP) at parameter overhead below 1.6% of the standard output projection. The four collapse phenomena identified in the literature -- rank collapse in depth, in width, head-channel non-identifiability, and entropy collapse -- are unified under a symmetry-breaking framework, each corresponding to a distinct symmetry of the Transformer's forward pass.

Transformer表征坍塌对称性破缺模型分析

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