首页 时政热点 科技头条 智能AI 安全攻防 数码硬件 开发者生态 汽车 游戏 社会热点 开源推荐 医疗健康 归档 标签 关于
智能AI morning

扩大注意力:具有固定数量的循环状态的线性时间变压器的泰勒级数路线

摘要

arXiv:2610.20637v1 Announce Type: new Abstract: We expand the exponential kernel of attention into a truncated Taylor series of $n$ terms and factorize each term into a tensor-power inner product, whi...

the and times into states per for kernel truncated terms
2026-10-16 1 阅读 约1分钟阅读 Zicong Huang
分享:
字号:
arXiv:2610.20637v1 公告类型:新摘要:我们将注意力的指数内核扩展为 $n$ 项的截断泰勒级数,并将每个项分解为张量幂内积,从而将因果历史折叠为恰好 $n$ 的循环状态,每个状态由每个标记的一个一级外积更新,并且与上下文长度 $N$ 无关。递归相对于截断的内核是无损的(float64 中为 $2.2 \times 10^{-16}$),并且状态在块之间是相加的($5.7 \times 10^{-14}$),因此预填充允许并行扫描。 Softmax 的测量指数为 $t \propto N^{2.28}$,相对于状态的 $N^{0.91}​​$,交叉点为 $N \approx 728$ 和 $20.8\times$,$N = 8192$;对于 softmax,每个令牌的解码延迟相对于 $N^{1.04}$ 恒定为 $7.3\,\mu$s,在 $N = 65536$ 时为 $76.0\times$。精度以 $n$ 呈指数级提高,而成本以组合方式增长,因为第 $m$ 个状态是 $\binom{d+m-1}{m}$ 宽(由 SVD 确认的界限),因此最优值是二阶,$n = 3$,误差为 $6.5 \times 10^{-4}$,每个头有 $9{,}945$ 个元素。查询和密钥规范化是先决条件:没有它,参数就会超出收敛半径,并且添加项将不再有帮助。
这篇文章对您有帮助吗?

订阅66必读

每日精选科技资讯,直达你的邮箱