DeBiFormer:带有可变形代理双层路由注意力的视觉Transformer

https://arxiv.org/pdf/2410.08582v1

摘要

带有各种注意力模块的视觉Transformer在视觉任务上已表现出卓越的性能。虽然使用稀疏自适应注意力(如在DAT中)在图像分类任务中取得了显著成果,但在对语义分割任务进行微调时,由可变形点选择的关键值对缺乏语义相关性。BiFormer中的查询感知稀疏注意力旨在使每个查询关注前 k k k个路由区域。然而,在注意力计算过程中,所选的关键值对受到过多不相关查询的影响,从而降低了对更重要查询的关注度。为了解决这些问题,我们提出了可变形双级路由注意力(DBRA)模块,该模块使用代理查询优化关键值对的选择,并增强了注意力图中查询的可解释性。在此基础上,我们引入了带有DBRA模块的新型通用视觉Transformer——可变形双级路由注意力Transformer(DeBiFormer)。DeBiFormer已在各种计算机视觉任务上得到验证,包括图像分类、目标检测和语义分割,有力地证明了其有效性。代码可访问:https://github.com/maclong01/DeBiFormer
关键词:视觉Transformer,自注意力机制,图像识别

1 引言

视觉Transformer在计算机视觉领域近期展现出了巨大的潜力[15,29,44]。它能够捕获数据中的长距离依赖关系[29,41],并几乎引领了一种更灵活、更适合拟合大量数据的无卷积模型[44]。此外,它还具有高并行性,这有利于大型模型的训练和推理[11,41]。计算机视觉领域观察到,视觉Transformer的采用和发展呈现出爆炸式增长[1,14,15,29,44,45]。
在这里插入图片描述

为了提高注意力,大量研究精心设计了高效的注意力模式,其中每个查询都通过较小部分的关键值对进行选择性聚焦。如图1所示,在各种表示方法中,一些方法包括局部窗口[50]和空洞窗口[45, 40, 24]。此外,一些研究在方法论上通过数据稀疏性适应采取了不同的路径,如[5,47]中的工作所示。然而,尽管在合并或选择关键和值令牌时采用了不同的策略,但这些令牌对于查询来说并不具有语义性。采用这种方法时,当应用于预训练的ViT[41]和DETR[1]的其他下游任务时,查询并非源自语义区域的关键值对。因此,强制所有查询关注于不足的一组令牌可能无法产生最优结果。最近,随着动态查询感知稀疏注意力机制的出现,查询由最具动态语义性的关键值对进行聚焦,这被称为双级路由注意力[56]。然而,在这种方法中,查询由语义关键值对处理而非源自详细区域,这可能并非在所有情况下都能产生最优结果。此外,在计算注意力时,为所有查询选择的这些关键值和值受到太多不太相关查询的影响,导致对重要查询的注意力降低,这对执行分割任务时具有重大影响[13,25]。

为了使查询的注意力更加高效,我们提出了可变形双级路由注意力(DBRA),这是一种用于视觉识别的注意力中注意力架构。在DBRA的过程中,第一个问题是如何定位可变形点。我们使用了[47]中的观察结果,即注意力具有一个偏移网络,该网络以查询特征为输入,并为所有参考点生成相应的偏移量。因此,候选可变形点以高灵活性和效率向重要区域移动,以捕获更多信息性特征。第二个问题是如何从语义相关的关键值对中聚合信息,然后将信息回传给查询。因此,我们提出了一种注意力中注意力架构,其中如上文所示向可变形点移动的部分作为查询的代理。由于关键值对是为可变形点选择的,我们使用[56]中的观察结果来选择一小部分最具语义相关性的关键值对,即一个区域仅通过关注前 k k k个路由区域来所需的部分。然后,在选择了语义相关的关键值对后,我们首先使用带有可变形点查询的令牌到令牌注意力。接着,我们应用第二个令牌到令牌注意力将信息回传给查询,其中作为关键值对的可变形点被设计为表示语义区域中最重要的点。

综上所述,我们的贡献如下:

  1. 我们提出了可变形双级路由注意力(DBRA),这是一种用于视觉识别的注意力中注意力架构,能够灵活且语义化地获取数据依赖的注意力模式。
  2. 通过利用DBRA模块,我们提出了一种新的主干网络,称为DeBiFormer。根据注意力热图的可视化结果,该网络具有更强的识别能力。
  3. 在ImageNet[35]、ADE20K[55]和COCO[17]上的大量实验表明,我们的模型始终优于其他竞争基线。

2 相关工作

2.1 视觉Transformer

基于Transformer的主干网络结合了通道级MLP[38]块,通过通道混合嵌入每个位置的特征。此外,还使用注意力[41]块进行跨位置关系建模并促进空间混合。Transformer最初是为自然语言处理[41,11]而设计的,随后通过DETR[1]和ViT[41]等工作被引入计算机视觉领域。与卷积神经网络(CNN)相比,Transformer的主要区别在于它使用注意力替代卷积,从而促进了全局上下文建模。然而,传统的注意力机制计算所有空间位置之间的成对特征亲和力,这带来了巨大的计算负担和内存占用,特别是在处理高分辨率输入时。因此,一个关键的研究重点是设计更高效的注意力机制,这对于减轻计算需求至关重要,尤其是处理高分辨率输入时。

2.2 注意力机制

大量研究旨在减轻传统注意力机制带来的计算和内存复杂性。方法包括稀疏连接模式[6]、低秩近似[42]和循环操作[10]。在视觉Transformer的上下文中,稀疏注意力变得流行起来,特别是在Swin Transformer[29]取得显著成功后。在Swin Transformer框架中,注意力被限制在非重叠的局部窗口中,并引入了一种创新的移位窗口操作。该操作促进了相邻窗口之间的通信,为其处理注意力机制提供了独特的方法。为了在不超过计算限制的情况下实现更大或近似全局的感受野,最近的研究结合了多种手动设计的稀疏模式。这些模式包括空洞窗口[45,40,24]和十字形窗口[14]的集成。此外,一些研究致力于使稀疏模式适应数据,如DAT[47]、TCFormer[53]和DPT[5]等工作所示。尽管它们通过使用不同的合并或选择策略来减少关键值令牌的数量,但重要的是要认识到这些令牌缺乏语义特异性。相反,我们加强了查询感知的关键值令牌选择。

我们的工作受到一个观察结果的启发:对于重要查询,语义上关注的区域可能表现出显著差异,如ViT[41]和DETR[1]等预训练模型的可视化所示。在实现通过粗细粒度方法实现的查询自适应稀疏性时,我们提出了一种注意力中注意力架构,该架构结合了可变形注意力[47]和双级路由注意力[56]。与可变形注意力[47]和双级路由注意力[56]不同,我们的可变形双级路由注意力旨在加强最具语义性和灵活性的关键值对。相比之下,双级路由注意力仅关注定位少数高度相关的关键值对,而可变形注意力则优先识别少数最具灵活性的关键值对。

3 我们的方法:DeBiFormer

3.1 预备知识

首先,我们回顾了最近视觉Transformer中使用的注意力机制。以扁平化的特征图 x ∈ R N × C x \in \mathrm{R}^{N \times C} xRN×C 作为输入,具有 M M M个头的多头自注意力(MHSA)块表示为

\begin{array}{c}
q=xW_{q}, k=xW_{k}, v=xW_{v} \
z{(m)}=\sigma\left(q{(m)}k^{(m) \top} / \sqrt{d}\right)v^{(m)}, m=1, \ldots, M \
z=\operatorname{Concat}\left(z^{(1)}, \ldots, z^{(M)}\right)W_{o}
\end{array}

其中, σ ( ⋅ ) \sigma(\cdot) σ() 表示softmax函数, d = C / M d=C / M d=C/M 是每个头的维度。 z ( m ) z^{(m)} z(m) 表示第 m m m个注意力头的嵌入输出,而 q ( m ) , k ( m ) , v ( m ) ∈ R N × d q^{(m)}, k^{(m)}, v^{(m)} \in \mathrm{R}^{N \times d} q(m),k(m),v(m)RN×d 分别表示查询、键和值嵌入。 W q , W k , W v , W o ∈ R C × C W_{q}, W_{k}, W_{v}, W_{o} \in \mathrm{R}^{C \times C} Wq,Wk,Wv,WoRC×C 是投影矩阵。带有归一化层和恒等捷径的第 l l l个Transformer块(其中LN表示层归一化)表示为
z l ′ = MHSA ⁡ ( L N ( z l − 1 ) ) + z l − 1 z l = MLP ⁡ ( L N ( z l ′ ) ) + z l ′ \begin{array}{c} z_{l}^{\prime}=\operatorname{MHSA}\left(L N\left(z_{l-1}\right)\right)+z_{l-1} \\ z_{l}=\operatorname{MLP}\left(L N\left(z_{l}^{\prime}\right)\right)+z_{l}^{\prime} \end{array} zl=MHSA(LN(zl1))+zl1zl=MLP(LN(zl))+zl

3.2 可变形双层路由注意力(DBRA)

所提出的可变形双层路由注意力(DBRA)的架构如图2所示。我们首先采用一个可变形注意力模块,该模块包含一个偏移网络,该网络基于查询特征为参考点生成偏移量,从而创建可变形点。然而,这些点往往会在重要区域聚集,导致某些区域过度集中。
在这里插入图片描述

为解决此问题,我们引入了可变形点感知区域划分,确保每个可变形点仅与键值对的一个小子集进行交互。然而,仅依赖区域划分可能会导致重要区域和不太重要区域之间的不平衡。为解决此问题,DBRA模块被设计为更有效地分配注意力。在DBRA中,每个可变形点充当代理查询,与语义区域键值对计算注意力。这种方法确保每个重要区域仅分配少数可变形点,从而使注意力分散到图像的所有关键区域,而不是聚集在一个点上。

通过使用DBRA模块,不太重要区域的注意力减少,更重要区域的注意力增加,确保整个图像中注意力的平衡分布。

可变形注意力模块和输入投影。如图2所示,给定输入特征图 x ∈ R H × W × C x \in \mathrm{R}^{H \times W \times C} xRH×W×C,通过以因子 r r r 对输入特征图进行下采样,生成一个均匀的点网格 p ∈ R H G × W G × 2 p \in \mathrm{R}^{H_{G} \times W_{G} \times 2} pRHG×WG×2,其中 H G = H / r , W G = W / r H_{G}=H / r, W_{G}=W / r HG=H/r,WG=W/r,作为参考。为了获得每个参考点的偏移量,将特征进行线性投影以生成查询令牌 q = x W q q=x W_{q} q=xWq,然后将其输入到 θ offset  ( ⋅ ) \theta_{\text {offset }}(\cdot) θoffset () 子网络中,以产生偏移量 Δ p = θ offset  ( q ) \Delta p = \theta_{\text {offset }}(q) Δp=θoffset (q)。随后,在变形点的位置对特征进行采样作为键和值,并通过投影矩阵进行进一步处理:

q = x W q , Δ p = θ offset  ( q ) , x ˉ = φ ( x ; p + Δ p ) q=x W_{q}, \Delta p=\theta_{\text {offset }}(q), \bar{x}=\varphi(x ; p+\Delta p) q=xWq,Δp=θoffset (q),xˉ=φ(x;p+Δp)

其中, x ˉ \bar{x} xˉ 分别表示变形后的键 k ˉ \bar{k} kˉ 和值 v ˉ \bar{v} vˉ 嵌入。具体来说,我们将采样函数 φ ( ⋅ ; ⋅ ) \varphi(\cdot ; \cdot) φ(;) 设置为双线性插值,使其可微:

φ ( z ; ( p x , p y ) ) = ∑ r x , r y g ( p x , r x ) g ( p y , r y ) z [ r y , r x , : ] \varphi\left(z ;\left(p_{x}, p_{y}\right)\right)=\sum_{r_{x}, r_{y}} g\left(p_{x}, r_{x}\right) g\left(p_{y}, r_{y}\right) z\left[r_{y}, r_{x},:\right] φ(z;(px,py))=rx,ryg(px,rx)g(py,ry)z[ry,rx,:]

其中,函数 g ( a , b ) = max ⁡ ( 0 , 1 − ∣ a − b ∣ ) g(a, b)=\max (0,1-|a-b|) g(a,b)=max(0,1ab),且 ( r x , r y ) \left(r_{x}, r_{y}\right) (rx,ry) 表示 z ∈ R H × W × C z \in \mathrm{R}^{H \times W \times C} zRH×W×C 上所有位置的索引。在类似于可变形注意力的设置中,当 g g g 在最接近 ( p x , p y ) (p_x, p_y) (px,py) 的四个整数点上不为零时,方程7简化为这四个位置上的加权平均。

区域划分与区域间路由。给定可变形注意力特征图输入 x ˉ ∈ R H G × W G × C \bar{x} \in \mathrm{R}^{H_{G} \times W_{G} \times C} xˉRHG×WG×C 和特征图 x ∈ R H × W × C x \in \mathrm{R}^{H \times W \times C} xRH×W×C,过程首先将其划分为大小为 S × S S \times S S×S 的非重叠区域,使得每个区域包含 H G W G S 2 \frac{H_{G} W_{G}}{S^{2}} S2HGWG 个特征向量,并将重塑后的 x ˉ \bar{x} xˉ 记为 x r ‾ ∈ R S 2 × H G W G S 2 × C \overline{x^{r}} \in \mathrm{R}^{S^{2} \times \frac{H_{G} W_{G}}{S^{2}}} \times C xrRS2×S2HGWG×C,将 x x x 记为 x r ∈ R S 2 × H W S 2 × C x^{r} \in \mathrm{R}^{S^{2} \times \frac{H W}{S^{2}}} \times C xrRS2×S2HW×C。然后,我们通过线性投影得到查询、键和值:

q ^ = x r ‾ W q , k ^ = x r W k , v ^ = x r W v \hat{q}=\overline{x^{r}} W_{q}, \hat{k}=x^{r} W_{k}, \hat{v}=x^{r} W_{v} q^=xrWq,k^=xrWk,v^=xrWv

接下来,我们使用BiFormer[56]中介绍的区域间方法,通过构建有向图来建立注意关系。首先,通过每个区域的平均值得到区域查询和键 q ^ r , k ^ r ∈ S S 2 × C \hat{q}^{r}, \hat{k}^{r} \in \mathrm{~S}^{S^{2} \times C} q^r,k^r SS2×C。然后,通过 Q r Q^{r} Qr K r ⊤ K^{r^{\top}} Kr 矩阵乘法得到区域间亲和图的邻接矩阵 A r ∈ S 2 × S 2 A^{r} \in \mathrm{~S}^{2} \times \mathrm{S}^{2} Ar S2×S2

A r = q ^ r ( k ^ r ) ⊤ A^{r}=\hat{q}^{r}\left(\hat{k}^{r}\right)^{\top} Ar=q^r(k^r)

其中,邻接矩阵 A r A^{r} Ar 量化了两个区域之间的语义关系。该方法的关键步骤是通过使用topk操作符和路由索引矩阵 I r ∈ N S 2 × k I^{r} \in \mathrm{~N}^{S^{2} \times k} Ir NS2×k 保留每个区域的topk连接来修剪亲和图:

I r = topk ⁡ ( A r ) I^{r}=\operatorname{topk}\left(A^{r}\right) Ir=topk(Ar)

双层标记到可变形层标记注意力。利用区域路由矩阵 I r I^{r} Ir,我们可以应用标记注意力。对于区域 i i i 内的每个可变形查询标记,其注意力跨越位于topk路由区域中的所有键值对,即由 I i , 1 r , I i , 2 r , … , I i , k r I_{i, 1}^{r}, I_{i, 2}^{r}, \ldots, I_{i, k}^{r} Ii,1r,Ii,2r,,Ii,kr 索引的那些。因此,我们继续收集键和值的过程:

k ^ g = gather ⁡ ( k ^ , I r ) , v ^ g = gather ⁡ ( v ^ , I r ) \hat{k}^{g}=\operatorname{gather}\left(\hat{k}, I^{r}\right), \hat{v}^{g}=\operatorname{gather}\left(\hat{v}, I^{r}\right) k^g=gather(k^,Ir),v^g=gather(v^,Ir)

其中, k ^ g , v ^ g ∈ R S 2 × k H W S 2 × C \hat{k}^{g}, \hat{v}^{g} \in \mathrm{R}^{S^{2} \times \frac{k H W}{S^{2}} \times C} k^g,v^gRS2×S2kHW×C 是收集的键和值。然后,我们对 k ^ g , v ^ g \hat{k}^{g}, \hat{v}^{g} k^g,v^g 应用注意力:

O ^ = x ^ + W o ′ ( Attention  ( q ^ , k ^ g , v ^ g ) + L C E ( v ^ ) ) O = MLP ⁡ ( L N ( O ^ ) ) + O ^ \begin{array}{c}\hat{O}=\hat{x}+W_{o^{\prime}}\left(\text { Attention }\left(\hat{q}, \hat{k}^{g}, \hat{v}^{g}\right)+L C E(\hat{v})\right) \\O=\operatorname{MLP}(L N(\hat{O}))+\hat{O}\end{array} O^=x^+Wo( Attention (q^,k^g,v^g)+LCE(v^))O=MLP(LN(O^))+O^

其中, W o ′ W_{o^{\prime}} Wo 是输出特征的投影权重, L C E ( ⋅ ) L C E(\cdot) LCE() 使用核大小为5的深度卷积。

可变形层标记到标记注意力。之后,通过[56]语义关注的可变形特征被重塑为 O r ∈ R H G × W G × C O^{r} \in \mathbb{R}^{H_{G} \times W_{G} \times C} OrRHG×WG×C,并在键和值的位置进行参数化:

k = O r W k , v = O r W v k=O^{r} W_{k}, v=O^{r} W_{v} k=OrWk,v=OrWv

k k k v v v 分别表示语义变形键和值的嵌入。使用现有方法,我们对 q , k , v q, k, v q,k,v 和相对位置偏移 R R R 执行自注意力。注意力的输出公式如下:

z m = W o ˉ ( σ ( q m k ( m ) ⊤ / d + ϕ ( B ^ ; R ) ) v m ) z^{m}=W_{\bar{o}}\left(\sigma\left(q^{m} k^{(m) \top} / \sqrt{d}+\phi(\hat{B} ; R)\right) v^{m}\right) zm=Woˉ(σ(qmk(m)/d +ϕ(B^;R))vm)

这里, ϕ ( B ^ ; R ) ∈ R H W × H G W G \phi(\hat{B} ; R) \in \mathbb{R}^{H W \times H_{G} W_{G}} ϕ(B^;R)RHW×HGWG 对应位置嵌入,遵循先前工作[29]的方法。然后,将 z m z^{m} zm 通过 W o W_{o} Wo 投影得到最终输出 z z z,如方程3所示。

3.3 模型架构

利用DBRA作为基本构建块,我们引入了一种新的视觉转换器,称为DeBiFormer。如图3所示,我们遵循最新的最先进的视觉转换器[14,29,56,47],使用四阶段金字塔结构。在第 i i i 阶段,我们在第一阶段使用重叠补丁嵌入,在第二到第四阶段使用补丁合并模块[26,34]。这是为了降低输入空间分辨率,同时增加通道数。随后,使用 N i N_{i} Ni 个连续的DeBiFormer块来转换特征。在每个DeBiFormer块内,我们遵循最近的方法论[26,40,56],在开始时使用 3 × 3 3 \times 3 3×3 深度卷积。这是为了隐式编码相对位置信息。之后,我们依次使用一个DBRA模块和一个具有扩展比 e e e 的2-ConvFFN模块,分别用于跨位置关系建模和每个位置的嵌入。DeBiFormer以三种不同的模型尺寸实例化,通过按表1中概述的网络宽度和深度进行缩放来实现。每个注意力头包含32个通道,我们使用具有MLP扩展比 e = 3 e=3 e=3 的双层ConvFFN和可变形层ConvFFN。对于BRA,我们在四个阶段使用topk =1,4,16, S 2 S^{2} S2,对于DBRA,我们使用topk =4,8,16, S 2 S^{2} S2。此外,我们将区域划分因子 S S S 设置为特定值:分类任务中 S = 7 S=7 S=7,语义分割任务中 S = 8 S=8 S=8,目标检测任务中 S = 20 S=20 S=20
在这里插入图片描述

4 实验

我们通过实验评估了所提出的DeBiFormer在各种主流计算机视觉任务上的有效性,包括图像分类(第4.1节)、语义分割(第4.2节)和目标检测以及实例分割(第4.3节)。在我们的方法中,我们从ImageNet-1K [35]数据集开始从头训练图像分类模型。随后,我们在ADE20K [55]数据集上对预训练的主干网络进行微调,以进行语义分割,并在COCO [17]数据集上进行微调,以进行目标检测和实例分割。此外,我们进行了消融研究,以验证所提出的可变形双级路由注意力(Deformable Bi-level Routing Attention)和DeBiFormer的top-k选择的有效性(第4.4节)。最后,为了验证我们DeBiFormer的识别能力和可解释性,我们对注意力图进行了可视化(第5节)。
在这里插入图片描述

4.1 在ImageNet-1K上的图像分类

设置。我们在ImageNet-1K [35]数据集上进行了图像分类实验,遵循DeiT [39]的实验设置以进行公平比较。具体来说,每个模型在8个V100 GPU上以224×224的输入大小训练300个epoch。我们使用AdamW作为优化器,权重衰减为0.05,并采用余弦衰减学习率调度策略,初始学习率为0.001,同时前五个epoch用于线性预热。批量大小设置为1024。为避免过拟合,我们使用了正则化技术,包括RandAugment [9](rand-m9-mstd0.5-inc1)、MixUp [54](prob=0.8)、CutMix [52](prob=1.0)、随机擦除(prob=0.25)以及增加随机深度[23](对于DeBiFormer-T/S/B,prob分别为0.1/0.2/0.4)。结果。我们在表2中报告了结果,展示了具有相似计算复杂度的top-1准确率。我们的DeBiFormer在所有三个尺度上都优于Swin Transformer [29]、PVT [44]、DeiT [39]、DAT[47]和Biformer [56]。在不将卷积插入Transformer块或使用重叠卷积进行块嵌入的情况下,DeBiFormer相对于BiFormer [56]对应版本分别实现了0.5pt、0.1pt和0.1pt的增益。
在这里插入图片描述

4.2 在ADE20K上的语义分割

设置。与现有工作相同,我们在SemanticFPN [46]和UperNet [48]上使用了我们的DeBiFormer。在这两种情况下,主干网络都使用ImageNet-1K预训练权重进行初始化。优化器是AdamW [31],批量大小为32。为进行公平比较,我们遵循PVT [44]的相同设置,用80k步训练模型,并遵循Swin Transformer [29]的相同设置,用160k步训练模型。
在这里插入图片描述

结果。表8展示了两个不同框架的结果。结果表明,在使用Semantic FPN框架的情况下,我们的DeBiFormer-S/B分别实现了49.2/50.6 mIoU,比BiFormer提高了0.3pt/0.7pt。对于UperNet框架,也观察到了类似的性能增益。通过使用DBRA模块,我们的DeBiFormer能够捕获最多的语义键值对,这使得注意力选择更加合理,并在下游语义任务上实现了更高的性能。

4.3 目标检测和实例分割

设置。我们使用DeBiFormer作为Mask RCNN [19]和RetinaNet [16]框架中的主干网络,以评估模型在COCO 2017 [17]数据集上对于目标检测和实例分割的有效性。实验使用MMDetection [3]工具箱进行。在COCO上进行训练之前,我们使用ImageNet-1K预训练权重对主干网络进行初始化,并遵循与BiFormer [56]相同的训练策略以公平比较我们的方法。请注意,由于设备限制,我们在这些实验中设置小批量大小为4,而在BiFormer中此值为16。有关实验具体设置的详细信息,请参阅补充论文。
在这里插入图片描述

结果。我们在表4.2中列出了结果。对于使用RetinaNet进行的目标检测,我们报告了不同IoU阈值(50%,75%)下三个目标尺寸(即小、中、大(S/M/L))的平均精度(mAP)和平均精度(AP)。从结果中可以看出,尽管DeBiFormer的整体性能仅与一些最具竞争力的现有方法相当,但在大目标(AP_L)上的性能却优于这些方法,尽管我们使用的资源有限。这可能是因为DBRA更合理地分配了可变形点。这些点不仅关注小事物,还关注图像中的重要事物。因此,注意力不仅局限于小区域,从而提高了大目标的检测准确性。对于使用Mask R-CNN进行的实例分割,我们报告了不同IoU阈值(50%,75%)下的边界框和掩码的平均精度(AP_b和AP_m)。请注意,尽管受到设备限制(小批量大小),我们的DeBiFormer仍然取得了出色的性能。我们认为,如果小批量大小可以与其他方法相同,我们将能够取得更好的结果,这在语义分割任务中已经得到了证明。

4.4 消融研究

DBRA的有效性。我们将DBRA与几种现有的稀疏注意力机制进行了比较。遵循CSWIN [14],我们为公平比较将宏观架构设计与Swin-T [29]对齐。具体来说,我们在四个阶段分别使用了2、2、6、2个块和非重叠的补丁嵌入,并将初始补丁嵌入维度设置为 C = 96 C=96 C=96,MLP扩展比率设置为 e = 4 e=4 e=4。结果如表5所示。在图像分类和语义分割方面,我们的可变形双级路由注意力(Deformable Bi-level Routing Attention)性能明显优于现有的稀疏注意力机制。

分区因子 S S S。与BiFormer类似,我们选择使用 S S S作为训练尺寸的除数,以避免填充。我们使用分辨率为 224 = 7 × 32 224=7 \times 32 224=7×32的图像分类,并设置 S = 7 S=7 S=7,以确保每个阶段的特征图尺寸都能被整除。这一选择与Swin Transformer [29]中使用的策略一致,其中窗口大小为7。
在这里插入图片描述

Top-k选择。我们系统地调整了 k k k,以确保在后续阶段区域尺寸减小时,有合理数量的令牌被关注到可变形查询上。探索 k k k的各种组合是一个可行的选择。在表9中,我们按照DeBiFormer-STL(“STL”表示Swin-T布局)报告了在IN-1K上的消融结果。从这些实验中得出的一个关键观察结果是,增加关注到可变形查询的令牌数量对准确性和延迟有不利影响,而在第1和第2阶段增加关注到的令牌数量对准确性有影响。

不同阶段的可变形双级路由多头注意力(DBRMHA)。为了评估设计选择的影响,我们系统地用DBRMHA块替换了不同阶段中的双级路由注意力块,如表7所示。最初,所有阶段都使用双级路由注意力,类似于BiFormer-T [56],在图像分类中实现了 81.3 % 81.3 \% 81.3%的准确率。仅将第4阶段的一个块替换为DBRMHA,准确率立即提高了 + 0.21 +0.21 +0.21。将第4阶段的所有块都替换为DBRMHA,又增加了 + 0.05 +0.05 +0.05。在第3阶段进一步替换DBRMHA块继续提高了各项任务的性能。尽管早期阶段的替换带来的增益逐渐减少,但我们最终确定了一个版本——DeBiFormer,其中所有阶段都使用可变形双级路由注意力,以保持简洁性。
在这里插入图片描述

5 Grad-CAM可视化

为了进一步说明所提出的DeBiFormer识别重要区域注意力的能力,我们使用Grad-CAM [36]可视化了BiFormer-Base和DeBiFormer-Base最关注的区域。如图4所示,通过使用DBRA模块,我们的DeBiFormer-Base模型在定位目标对象方面表现更好,其中更多的区域被关注到。此外,我们的模型降低了在不必要区域的注意力,并更加关注必要区域。根据对更多必要区域的注意力,我们的DeBiFormer模型更加连续和完整地关注语义区域,这表明我们的模型具有更强的识别能力。这种能力相比BiFormer-Base带来了更好的分类和语义分割性能。
在这里插入图片描述

6 结论

本文介绍了可变形双级路由注意力Transformer(Deformable Bi-level Routing Attention Transformer),这是一种专为图像分类和密集预测任务设计的新型分层视觉Transformer。通过可变形双级路由注意力,我们的模型优化了查询-键-值交互,同时自适应地选择语义相关区域。这实现了更高效和有意义的注意力。大量实验表明,与强大的基线相比,我们的模型具有有效性。我们希望这项工作能为设计灵活且语义感知的注意力机制提供见解。

7 补充材料

7.1 偏移组

与[47]类似,为了促进变形点之间的多样性,我们遵循与MHSA中相似的范式,其中通道被分成多个头来计算各种注意力。因此,我们将通道分成 G G G组以生成不同的偏移量。偏移生成网络对来自不同组的特征共享权重。

7.2 可变形相对位置偏置

当然,将位置信息融入注意力机制已被证明对模型性能有益。诸如APE[15]、RPE[29]、CPE[8]、LogCPB[28]等方法以及其他方法已证明能够改善结果。Swin Transformer中引入的相对位置嵌入(RPE)特别编码了每对查询和键之间的相对位置,从而通过空间归纳偏置增强了普通注意力[29]。相对位置的显式建模特别适合可变形级别的注意力头。在这种情况下,变形键可以假设任意连续位置,而不是局限于固定的离散网格。

根据[47],相对坐标位移在空间维度上被限制在 [ − H , + H ] [-H,+H] [H,+H] [ − W , + W ] [-W,+W] [W,+W]范围内,并带有一个相对位置偏置(RPB),表示为 B ^ \hat{B} B^,其维度为 ( 2 H − 1 ) × ( 2 W − 1 ) (2 H-1) \times(2 W-1) (2H1)×(2W1)

然后,使用带参数偏置的双线性插值 φ ( B ^ ; R ) \varphi(\hat{B} ; R) φ(B^;R),在 [ − 1 , + 1 ] [-1,+1] [1,+1]范围内对相对位置进行采样。这是通过考虑连续相对位移来完成的,以确保覆盖所有可能的偏移值。

7.3 计算复杂度

可变形双层路由注意力(DBRA)的计算成本与Swin Transformer中的对应机制相当。DBRA的计算包括两部分:令牌到令牌的注意力和偏移量&采样。因此,这部分的计算是:

F L O P s def  = F L O P s attn  + F L O P s offset&sampling  = 2 H W N s C + 2 H W C 2 + 2 N s C 2 + ( k 2 + 6 ) N s C \begin{array}{l}FLOPs_{\text {def }}=FLOPs_{\text {attn }}+FLOPs_{\text {offset\&sampling }} \\=2 H W N_{s} C+2 H W C^{2}+2 N_{s} C^{2}+\left(k^{2}+6\right) N_{s} C\end{array} FLOPsdef =FLOPsattn +FLOPsoffset&sampling =2HWNsC+2HWC2+2NsC2+(k2+6)NsC

其中, N s = H W / r 2 N_{s}=H W / r^{2} Ns=HW/r2是采样点的数量, C C C是令牌嵌入维度。双层路由多头注意力的计算包括三部分:线性投影、区域到区域的路由和令牌到令牌的注意力。因此,这部分的计算是:

F L O P s b i = F L O P s proj  + F L O P s routing  + F L O P s attn  = 2 H W C 2 + 2 N s C 2 + 2 ( S 2 ) 2 C + 2 H W k N s S 2 C = 2 H W C 2 + 2 N s C 2 + C { 2 S 4 + 2 H W k N s S 2 } = 2 H W C 2 + 2 N s C 2 + C { 2 S 4 + k H W N s S 2 + k H W N s S 2 } ≥ 2 H W C 2 + 2 N s C 2 + 3 C { 2 S 4 ⋅ k H W N s S 2 ⋅ k H W N s S 2 } 1 3 = 2 H W C 2 + 2 N s C 2 + 3 C k 2 3 { 2 H W N s } 2 3 \begin{array}{l}FLOPs_{b i}=FLOPs_{\text {proj }}+FLOPs_{\text {routing }}+FLOPs_{\text {attn }} \\=2 H W C^{2}+2 N_{s} C^{2}+2\left(S^{2}\right)^{2} C+2 H W k \frac{N_{s}}{S^{2}} C \\=2 H W C^{2}+2 N_{s} C^{2}+C\left\{2 S^{4}+2 H W k \frac{N_{s}}{S^{2}}\right\} \\=2 H W C^{2}+2 N_{s} C^{2}+C\left\{2 S^{4}+\frac{k H W N_{s}}{S^{2}}+\frac{k H W N_{s}}{S^{2}}\right\} \\ \geq 2 H W C^{2}+2 N_{s} C^{2}+3 C\left\{2 S^{4} \cdot \frac{k H W N_{s}}{S^{2}} \cdot \frac{k H W N_{s}}{S^{2}}\right\}^{\frac{1}{3}} \\=2 H W C^{2}+2 N_{s} C^{2}+3 C k^{\frac{2}{3}}\left\{2 H W N_{s}\right\}^{\frac{2}{3}}\end{array} FLOPsbi=FLOPsproj +FLOPsrouting +FLOPsattn =2HWC2+2NsC2+2(S2)2C+2HWkS2NsC=2HWC2+2NsC2+C{2S4+2HWkS2Ns}=2HWC2+2NsC2+C{2S4+S2kHWNs+S2kHWNs}2HWC2+2NsC2+3C{2S4S2kHWNsS2kHWNs}31=2HWC2+2NsC2+3Ck32{2HWNs}32

其中, k k k是要注意的区域数量, S S S是区域划分因子。最后,DBRA的总计算包括两部分:可变形级别的多头注意力和双层路由多头注意力。因此,总计算量是:

F L O P s = F L O P s b i + F L O P s d e f = 2 H W N s C + 2 H W C 2 + 2 N s C 2 + ( k 2 + 6 ) N s C + 2 H W C 2 + 2 N s C 2 + 3 C k 2 3 { 2 H W N s } 2 3 = 2 H W N s C + 4 H W C 2 + 4 N s C 2 + ( k 2 + 6 ) N s C + 3 C k 2 3 { 2 H W N s } 2 3 \begin{array}{l}FLOPs=FLOPs_{b i}+FLOPs_{d e f} \\=2 H W N_{s} C+2 H W C^{2}+2 N_{s} C^{2}+\left(k^{2}+6\right) N_{s} C \\+2 H W C^{2}+2 N_{s} C^{2}+3 C k^{\frac{2}{3}}\left\{2 H W N_{s}\right\}^{\frac{2}{3}} \\=2 H W N_{s} C+4 H W C^{2}+4 N_{s} C^{2} \\+\left(k^{2}+6\right) N_{s} C+3 C k^{\frac{2}{3}}\left\{2 H W N_{s}\right\}^{\frac{2}{3}}\end{array} FLOPs=FLOPsbi+FLOPsdef=2HWNsC+2HWC2+2NsC2+(k2+6)NsC+2HWC2+2NsC2+3Ck32{2HWNs}32=2HWNsC+4HWC2+4NsC2+(k2+6)NsC+3Ck32{2HWNs}32

换句话说,DBRA实现了 O ( ( H W N s ) 2 3 ) O\left(\left(H W N_{s}\right)^{\frac{2}{3}}\right) O((HWNs)32)的复杂度。例如,对于图像分类的层次模型,其第三阶段具有 224 × 224 224 \times 224 224×224输入,通常具有 H = W = 14 H=W=14 H=W=14 S 2 = 49 S^{2}=49 S2=49 N s = 1 N_{s}=1 Ns=1 C = 384 C=384 C=384的计算规模,因此具有多头自注意力的计算复杂度。此外,通过增大下采样因子 r r r并根据区域划分因子 S S S进行缩放,可以进一步降低复杂度,使其适用于具有更高分辨率输入的任务,如目标检测和实例分割。

7.4 要关注的令牌

在表9中,我们展示了要关注查询的令牌和要关注可变形点的令牌。与其他方法相比,DeBiFormer每个查询要关注的令牌最少,但在Imagenet1K、ADE20K(S-FPN头)和COCO(Retina头)上表现出高性能。
在这里插入图片描述

7.5 更多可视化结果

有效感受野分析 为了评估不同模型中,输入尺寸为224x224时中心像素的有效感受野(ERF)[32],我们在图5中展示了比较分析。为了证明我们DeBiFormer的强大表示能力,我们还比较了具有相似计算成本的几种SOTA(state-of-the-art,当前最优)方法的有效感受野。如图5所示,我们的DeBiFormer在这些方法中拥有最大且最一致的有效感受野,同时保持了强大的局部敏感性,这是很难实现的。
在这里插入图片描述

Grad-CAM分析 为了进一步展示DBRA(动态双分支注意力,Dynamic Bi-branch Attention)的工作原理,我们在图6中展示了更多的可视化结果。得益于灵活的键值对选择,在大多数情况下,我们的DeBiFormer在早期阶段就关注于重要对象。同时,由于变形点的合理分配,它在多对象场景中也能更早地关注于不同的重要区域。凭借强大的DBRA模块,我们的DeBiFormer在最后两个阶段具有更大的热图区域,这代表了更强的识别能力。
在这里插入图片描述

7.6 详细实验设置

ImageNet-1K上的图像分类 如主文所述,每个模型在8个V100 GPU上以224x224的输入尺寸训练300个epoch。实验设置严格遵循DeiT[39]以进行公平比较。更多详细信息,请参阅提供的表10。

目标检测和实例分割 当将我们的DeBiFormer微调至COCO[17]上的目标检测和实例分割时,我们考虑了两种常见框架:Mask R-CNN[19]和RetinaNet[16]。对于优化,我们采用AdamW优化器,初始学习率为0.0002,由于设备限制,小批量大小为4。当训练不同大小的模型时,我们根据图像分类中使用的设置调整训练设置。训练模型时使用的详细超参数见表11。
在这里插入图片描述

语义分割 对于ADE20K,我们为所有训练了160K迭代的模型使用AdamW优化器,初始学习率为0.00006,权重衰减为0.01,小批量大小为16。在测试方面,我们在主要比较中报告了使用单尺度(SS)和多尺度(MS)测试的结果。对于多尺度测试,我们尝试了从0.5倍到1.75倍训练分辨率的分辨率范围。为了设置不同模型中的路径丢弃率,我们使用了与目标检测和实例分割相同的超参数。表8显示了Upernet框架在单尺度和多尺度IoU下的结果。

7.7 限制和未来工作

与具有简单静态模式的稀疏注意力相比,我们提出了一种新的注意力方法,该方法由两个组件组成。首先,我们修剪区域级图,并为重要区域收集键值对,这些区域由高度灵活的键值对所关注。然后,我们应用令牌到令牌的注意力。虽然这种方法由于在顶级k路由的语义相关区域级别和可变形的重要区域上操作而不会引起太多计算,但它不可避免地在线性投影期间涉及额外的参数容量交易。在未来的工作中,我们计划研究高效的稀疏注意力机制,并增强具有参数容量意识的视觉Transformer。

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/web/55464.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

软件测试面试题600多条及答案

这些问题都是软件测试领域常见的面试问题,以下是一些可能的答案: 什么是软件测试? 软件测试是一系列活动,旨在评估软件产品的质量和性能,以确保它符合规定的需求和标准。它包括执行程序或系统以验证其满足规定需求的过…

“探索Adobe Photoshop 2024:订阅方案、成本效益分析及在线替代品“

设计师们对Adobe Photoshop这款业界领先的图像编辑软件肯定不会陌生。如果你正考虑加入Photoshop的用户行列,可能会对其价格感到好奇。Photoshop的价值在于其强大的功能,而它的价格也反映了这一点。下面,我们就来详细了解一下Adobe Photoshop…

数据结构(8.2_1)——插入排序

插入排序 算法思想&#xff1a;每次将一个待排序的记录按其关键字大小插入到前面已排序好的子序列中&#xff0c;直到全部记录插入完成。 代码实现 #include <stdio.h>void InsertSort(int A[], int n) {int i, j.temp;for (i 1; i < n; i) {//将各元素插入已排好…

Axure重要元件二——内联框架

亲爱的小伙伴&#xff0c;在您浏览之前&#xff0c;烦请关注一下&#xff0c;在此深表感谢&#xff01; 课程主题&#xff1a;内联框架 课程内容&#xff1a;认识内联框架、基本嵌入 应用场景&#xff1a;表单、图片、文字嵌入式场景、交互应用 一、认识内联框架 内联框架的…

如何安全擦除 iPhone 上的所有数据,避免隐私泄露?

在当今的数字时代&#xff0c;隐私安全尤为重要。特别是在转让或出售 iPhone 之前&#xff0c;擦除设备上的所有内容是每位用户都应注意的关键步骤。尽管苹果自带了删除数据的功能&#xff0c;但有时这并不足以保证数据完全无法恢复。本文将结合 iPhone 自带的"抹掉所有内…

软考(中级-软件设计师)计算机系统篇(1018)

十、存储系统 10.1 层次结构主存–辅存&#xff1a;实现虚拟存储系统&#xff0c;解决了主存容量不够的问题。 Cache–主存&#xff1a;解决了主存与CPU速度不匹配的问题。 10.2 分类 1、按位置分类&#xff1a;可分为内存和外存。 内存&#xff08;主存&#xff09;&#…

【从零开发Mybatis】引入XNode和XPathParser

引言 在上文&#xff0c;我们发现直接使用 DOM库去解析XML 配置文件&#xff0c;非常复杂&#xff0c;也很不方便&#xff0c;需要编写大量的重复代码来处理 XML 文件的读取和解析&#xff0c;代码可读性以及可维护性相当差&#xff0c;使用起来非常不灵活。 因此&#xff0c…

o1快慢思考的风又吹到了Agent!

智能体&#xff08;Agent&#xff09;通过自然对话与用户互动有两个任务&#xff1a;交谈和规划/推理。对话回应必须基于所有可用信息&#xff0c;行动必须有助于实现目标。与用户交谈和进行多步推理和规划之间的二分法&#xff0c;类似卡尼曼引入的人类快速思考和慢速思考系统…

库卡ForceTorqueControl(二)

1. 基准坐标系RCS 基准坐标系 RCS 是力 / 力矩控制的参考系。基准坐标系的原点始终是当前的TCP。 1.1 BASE 的 RCS 姿态 基准坐标系的姿态与当前基础坐标系&#xff08;基座坐标系&#xff09;的姿态一致。它不取决于刀具的姿态。基准坐标系的原点是当前的 TCP。 示例&#xff…

【数据库设计】概念结构设计

引入——整体解释 上次我们讲完了关系模型&#xff0c;这次我们来讲新的章节&#xff1a;数据库设计 该怎样有效地管理和存储现实中的数据&#xff1f;答案是设计一个优秀的数据库。现实中的数据转化成关系表中的数据需要经过四个主要的设计步骤。 现实世界需求分析——>…

java常用工具包

Java标准库&#xff08;Java Standard Library&#xff09; 比喻&#xff1a;就像你厨房里的基础调料&#xff0c;没有它们&#xff0c;你很难做出美味的菜肴。Java标准库包含了进行基本编程所需的所有核心类和方法&#xff0c;如字符串处理、集合框架、输入输出操作等。 关键…

C++ 内存布局 - Part6: 虚继承

1. 关于虚继承 虚继承可以在菱形继承体系中&#xff0c;防止派生类中有多份重复祖基类内容。如下图所示&#xff0c;如果是常规继承&#xff0c;Class Final中会有两份Class Base的内容。通过虚继承&#xff0c;即Derived1 虚继承自Base, Derived2 也虚继承自Base, 那么Final中…

003_ipc概述及信号

【背景】 程序运行起来后&#xff0c;每个模块都有自己的进程&#xff0c;那么不同的模块如何进行通讯或者数据交换呢&#xff1f; 上面这张图说明了linux的ipc是继承最初的Unix 的IPC逻辑的&#xff0c;那么具体关系和概述讲解&#xff0c;请参考此链接的原文&#xff1a;htt…

mac 桌面版docker no space left on device

报错信息 docker pull镜像时报&#xff1a; failed to register layer: Error processing tar file(exit status 1): write /home/admin/oceanbase_bak/bin/observer: no space left on device 解决 增加 docker 虚拟磁盘大小。 调整完点击重启即可。

助力语音技术发展,景联文科技提供语音数据采集服务

语音数据采集是语音识别技术、语音合成技术以及其他语音相关应用的重要基础。采集高质量的语音数据有助于提高语音识别的准确性&#xff0c;同时也能够促进语音技术的发展。 景联文科技作为专业的数据采集标注公司&#xff0c;支持语音数据采集。可通过手机、专业麦克风阵列、专…

两个案例全面阐述全链路测试怎么做

首先我们先针对全链路功能测试部分进行一下讲解。去年的时候&#xff0c;有一家电商公司可能知道我一直在帮银行做相关的测试&#xff0c;就请我帮他们去做一些规划。这个平台有虚拟订单&#xff0c;也有实体订单&#xff0c;方式不太一样。 还涉及到分账分佣以及跟银行的对接…

大数据-174 Elasticsearch Query DSL - 全文检索 full-text query 匹配、短语、多字段 详细操作

点一下关注吧&#xff01;&#xff01;&#xff01;非常感谢&#xff01;&#xff01;持续更新&#xff01;&#xff01;&#xff01; 目前已经更新到了&#xff1a; Hadoop&#xff08;已更完&#xff09;HDFS&#xff08;已更完&#xff09;MapReduce&#xff08;已更完&am…

计算机网络基础(1)

个人主页&#xff1a;C忠实粉丝 欢迎 点赞&#x1f44d; 收藏✨ 留言✉ 加关注&#x1f493;本文由 C忠实粉丝 原创 计算机网络基础 收录于专栏【计算机网络】 本专栏旨在分享学习计算机网络的一点学习笔记&#xff0c;欢迎大家在评论区交流讨论&#x1f48c; 目录 1. 计算机网…

简易CPU设计入门:验证取指令模块【未完成】

项目代码下载 还是请大家首先准备好本项目所用的源代码。如果已经下载了&#xff0c;那就不用重复下载了。如果还没有下载&#xff0c;那么&#xff0c;请大家点击下方链接&#xff0c;来了解下载本项目的CPU源代码的方法。 下载本项目代码 准备好了项目源代码以后&#xff…

MySQL详细学习攻略 MySQL基础非常全面教程 MySQL安装教程

MySQL安装教程 章节目录 一、MySQL简介与安装前准备 二、Windows系统下MySQL的安装 三、Linux系统下MySQL的安装 四、MySQL安装后的基本配置 五、MySQL服务的启动与停止 六、MySQL客户端工具的使用 七、MySQL安装常见问题与解决方案 一、MySQL简介与安装前准备 重点内容知识…