Transformer通过正交子空间分解世界,实现高效表征。
Transformers learn factored representations
- 用正交子空间分解输入因素,降低表征维度
- 在条件独立时无损压缩,否则牺牲精度换效率
- 揭示模型内在偏好:先分块再建模,适合可分解数据
通过下一个词预测预训练的Transformer能将世界分解为若干部分,并在残差流的正交子空间中表示这些因子。我们提出两种表征假设:(1) 因子乘积空间的表示,维度随因子数量呈指数增长;(2) 正交子空间中的因子化表示,维度线性增长。当因子条件独立时,因子化表示无损,但会牺牲预测保真度,形成维度效率与准确性的权衡。我们推导出每种假设下激活几何结构的精确预测,包括子空间数量、维度及上下文嵌入分布。在具有已知潜在结构的合成任务上测试,当因子条件独立时,模型学习因子化表示,并在训练早期即使存在噪声或隐藏依赖仍倾向因子化,体现一种以保真度为代价追求因子化的归纳偏置。这为Transformer为何分解世界提供了原理性解释,并暗示复杂数据训练中也可能存在可解释的低维结构。
原文摘要 · Abstract (English)
Transformers pretrained via next token prediction learn to factor their world into parts, representing these factors in orthogonal subspaces of the residual stream. We formalize two representational hypotheses: (1) a representation in the product space of all factors, whose dimension grows exponentially with the number of parts, or (2) a factored representation in orthogonal subspaces, whose dimension grows linearly. The factored representation is lossless when factors are conditionally independent, but sacrifices predictive fidelity otherwise, creating a tradeoff between dimensional efficiency and accuracy. We derive precise predictions about the geometric structure of activations for each, including the number of subspaces, their dimensionality, and the arrangement of context embeddings within them. We test between these hypotheses on transformers trained on synthetic processes with known latent structure. Models learn factored representations when factors are conditionally independent, and continue to favor them early in training even when noise or hidden dependencies undermine conditional independence, reflecting an inductive bias toward factoring at the cost of fidelity. This provides a principled explanation for why transformers decompose the world into parts, and suggests that interpretable low dimensional structure may persist even in models trained on complex data.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。