arXiv:2607.23050cs.LG2026-07被引 2

揭示Transformer模型最小容量需求,提出可预测的注意力原生容量度量

The Entropic Bound for Transformers: Why Static Rank Fails and Attention-Native Rank Recovers

  • 用熵界定义任务内在秩,证明线性注意力下该秩为不可逾越的最低容量
  • 发现传统静态秩失效主因是注意力的输入依赖性,而非softmax或秩约束
  • 提出注意力原生秩概念,实现训练前数据即可预估模型容量,适合架构设计者

神经尺度定律描述了损失随模型、数据和计算量增长而下降的现象,但未回答一个前提问题:固定任务下,解决它所需的最小模型容量是多少?本文通过熵界——一种Transformer的任务内在容量谱表示——来研究此问题。首先证明,在线性注意力代理中,标记混合算子的内在秩 $r^*$ 是紧致下界:任何秩不足的模型都会产生不可避免的额外风险,且该界在 $r^*$ 处可达。进一步表明,在标准低秩隐式偏差假设下,梯度下降可恢复该秩,三个性质均经实证验证,并显示 $r^*$ 可在训练前从数据中恢复。随后探讨其是否适用于真实注意力:朴素迁移失败,通过受控插值阶梯定位原因——并非softmax或秩约束,而是注意力混合算子的输入条件性,静态权重核无法概括。为此,提出注意力原生内在秩:在注意力类中实现任务的最小查询-键核秩,并证明在此定义下,完整熵界结构(缺陷、可达性、可恢复性)在双线性与softmax注意力中均成立,能量有效秩作为鲁棒估计器抵抗softmax失真。最后,划定仅数据可预测的边界:$r^*$ 在线性QK注意力下完全可恢复,即使无价值映射;而由于非线性逆和核-值可识别性效应,softmax注意力仅支持部分预训练恢复。结果将熵界从事后描述重构为具有精确可预测边界的注意力原生容量度量。

原文摘要 · Abstract (English)

Neural scaling laws describe how loss decreases as models, data, and compute grow, but they do not answer a prior question: for a fixed task, what is the minimum model capacity required to solve it? We study this through the Entropic Bound, a spectral notion of task-intrinsic capacity for Transformers. We first prove that, in a linear attention surrogate, the intrinsic rank $r^*$ of the token-mixing operator is a tight lower bound: any rank-deficient model incurs unavoidable excess risk, and the bound is achievable at $r^*$. We further show that gradient descent recovers this rank under standard low-rank implicit-bias assumptions, confirm all three properties empirically, and show $r^*$ is recoverable from data before training. We then ask whether this transfers to real attention. A naive transfer fails, and a controlled interpolation ladder localizes the cause precisely: it is not softmax and not a rank constraint, but the input-conditioned nature of attention's mixing operator, which a static weight kernel cannot summarize. Motivated by this, we introduce an attention-native intrinsic rank -- the minimum query-key kernel rank realizing the task within the attention class -- and show that under this definition the full Entropic Bound structure (deficiency, achievability, recovery) is restored for both linear and softmax attention, with the energy effective rank as the estimator robust to softmax distortion. Finally, we map the boundary of data-only predictability: $r^*$ is exactly recoverable for linear QK attention, even without the value map at scale, while softmax attention admits only partial pre-training recovery due to nonlinear inversion and kernel-value identifiability effects. Our results reframe the Entropic Bound from a post-hoc descriptor into an attention-native capacity measure with a precisely characterized predictability frontier.

Transformer容量分析注意力机制模型效率

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