发现语言模型层间更新有主导方向与独立残差,功能关键计算藏在几何上分离的残差中。
On the Geometric Structure of Layer Updates in Deep Language Models
- 将层更新分解为词元主导成分与残差成分,揭示其几何结构。
- 残差成分与主导成分对齐度低,投影弱,且大模型中相关性高达0.95。
- 适合研究模型内部机制、深层架构设计与可解释性的人参考。
我们研究深度语言模型中层间更新的几何结构。不同于分析中间表示编码了什么信息,我们关注表示如何从一层传递到下一层。结果表明,层间更新可分解为主导的词元级成分和一个无法被受限词元级函数类捕捉的残差成分。在多种架构(包括Transformer和状态空间模型)中,完整更新几乎完全对齐于词元级成分,而残差成分表现出显著更弱的对齐性、更大的角度偏差,以及在主导词元级子空间上的投影极低。这说明残差并非简单修正,而是具有独立几何特征的变换组成部分。该几何分离具有功能性后果:受限词元级模型的近似误差与输出扰动强相关,斯皮尔曼相关系数常超过0.7,大型模型中可达0.95。综合来看,大多数层间更新表现为沿主导方向的结构化重参数化,而功能关键计算集中于几何上独立的残差部分。本框架提供了一种简单、架构无关的方法,用于探查现代语言模型中层更新的几何与功能结构。
原文摘要 · Abstract (English)
We study the geometric structure of layer updates in deep language models. Rather than analyzing what information is encoded in intermediate representations, we ask how representations change from one layer to the next. We show that layerwise updates admit a decomposition into a dominant tokenwise component and a residual that is not captured by restricted tokenwise function classes. Across multiple architectures, including Transformers and state-space models, we find that the full layer update is almost perfectly aligned with the tokenwise component, while the residual exhibits substantially weaker alignment, larger angular deviation, and significantly lower projection onto the dominant tokenwise subspace. This indicates that the residual is not merely a small correction, but a geometrically distinct component of the transformation. This geometric separation has functional consequences: approximation error under the restricted tokenwise model is strongly associated with output perturbation, with Spearman correlations often exceeding 0.7 and reaching up to 0.95 in larger models. Together, these results suggest that most layerwise updates behave like structured reparameterizations along a dominant direction, while functionally significant computation is concentrated in a geometrically distinct residual component. Our framework provides a simple, architecture-agnostic method for probing the geometric and functional structure of layer updates in modern language models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。