Transformer数学推导——Q24 文档级跨注意力(Longformer)的滑动窗口复杂度推导

该问题归类到Transformer架构问题集——注意力机制——跨模态与多模态。请参考LLM数学推导——Transformer架构问题集

1. 问题背景:长序列建模的复杂度困境

传统 Transformer 的自注意力机制时间复杂度为 O(n^2)(n 为序列长度),当处理文档级长文本(如 n=4096 甚至更长)时,计算量会爆炸式增长(如 4096^2 \approx 16 百万次操作)。Longformer 提出的 ** 滑动窗口注意力(Sliding Window Attention)** 通过限制每个位置的注意力范围,将复杂度降至 O(nk)(k 为窗口大小),使其能高效处理长序列。

2. 滑动窗口注意力的核心机制

核心思想:每个位置仅关注其左右各 k 个邻居,形成大小为 w=2k+1 的局部窗口(边界位置窗口大小递减)。

  • 非重叠窗口:早期实现中窗口不重叠,但会导致上下文断裂;
  • 滑动窗口(重叠窗口):窗口每次滑动 s 步(通常 s=1 或 k),确保上下文连续性。 图示:对于序列 [1,2,3,4,5,6],若 k=1,窗口依次为 [1,2,3][2,3,4][3,4,5][4,5,6],每个位置被 
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

墨顿

唵嘛呢叭咪吽

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值