Single-Head Attention in High Dimensions: A Theory of Generalization, Weights Spectra, and Scaling Laws
高维中的单头注意力:一般化、权重谱和扩展定律的理论
机构 * Statistical Physics of Computation Laboratory, École polytechnique fédérale de Lausanne (EPFL)(计算物理实验室,瑞士联邦理工学院(EPFL)) ; Information, Learning and Physics Laboratory, École polytechnique fédérale de Lausanne (EPFL)(信息、学习与物理实验室,瑞士联邦理工学院(EPFL))
AI总结 本文研究了高维序列任务中训练的注意力层权重谱结构,通过随机矩阵理论等工具,揭示了训练误差、插值阈值及键查询矩阵谱的高维特性,并预测了功率谱定律的出现。
Journal ref ICML 2026