078、DAT详解双注意力Transformer如何突破超分性能瓶颈上个月调一个4倍超分模型跑完200个epochPSNR卡在29.8dB死活上不去。换了几种损失函数、调了学习率纹丝不动。同事说要不试试DAT我当时心想不就是Transformer加个注意力嘛能有多大区别。结果换上DAT backbone同样的训练配置直接跳到30.5dB。这个0.7dB的提升在超分领域够发一篇论文了。今天就把这个让我真香的DAT拆开揉碎讲清楚。为什么普通Transformer在超分上不够用先说说我踩过的坑。最早用SwinIR做超分效果确实比CNN强但有个问题SwinIR的窗口注意力虽然降低了计算量却牺牲了长距离依赖的建模能力。你想想超分任务里一个像素的重建往往需要参考图像中相隔很远的纹理信息——比如人脸超分左眼和右眼的对称性、额头和脸颊的光照一致性这些都不是局部窗口能搞定的。更坑的是SwinIR的移位窗口策略虽然能跨窗口交互但每个token的感受野增长是线性的而且需要多层堆叠才能覆盖全局。这就导致模型深度上去了训练稳定性却下来了——我试过把SwinIR从12层加到24层PSNR反而掉了0.1dB梯度爆炸了三次。DATDual Attention Transformer就是冲着这个问题来的。它的核心思路很直接既然局部注意力不够那就再加一个全局注意力两个注意力并行干活各取所长。DAT的双注意力机制到底怎么设计的DAT的全称是Dual Attention Transformer出自2022年的CVPR。它的设计哲学可以用一句话概括局部注意力保细节全局注意力抓结构。具体实现上DAT把Transformer block拆成两条并行分支分支一局部注意力分支这个分支和SwinIR的窗口注意力类似把特征图切成不重叠的窗口每个窗口内做自注意力。但DAT做了个关键改进——窗口大小不是固定的而是可学习的。我调试的时候发现这个可学习窗口参数对纹理密集区域特别敏感比如草地、头发这些地方窗口会自动缩小聚焦局部细节而天空、墙面这种平滑区域窗口会变大引入更多上下文。分支二全局注意力分支这才是DAT的精髓。它用了一个叫稀疏全局注意力的机制——不是对所有像素做全连接而是从特征图中采样一组锚点anchor points然后每个像素只和这些锚点做注意力计算。锚点的数量远小于像素总数比如一张256x256的特征图锚点可能只取64个计算量直接从O(N²)降到O(NM)MN。这里有个坑要注意锚点的选择不是随机采样的而是通过一个可学习的空间位置编码来确定的。我一开始偷懒用了均匀采样结果PSNR掉了0.3dB后来才发现DAT的锚点位置是动态调整的会自适应地集中在纹理复杂的区域。代码实现中的几个关键细节说点实战的。我复现DAT时遇到三个容易翻车的地方第一个双注意力的特征融合两个分支的输出怎么合并直接相加我试过效果一般。DAT的做法是先各自过一层LayerNorm然后用一个可学习的门控权重来控制融合比例。这个门控权重是逐通道的不同通道对局部和全局信息的依赖程度不同。比如低频通道对应光照、颜色更依赖全局注意力高频通道对应纹理、边缘更依赖局部注意力。# 这里踩过坑门控权重初始化不能太大否则训练初期全局注意力占主导细节全丢了gatenn.Parameter(torch.zeros(1,channels,1,1))# 初始化为0让两个分支等权重local_featlocal_attn(x)global_featglobal_attn(x)# 别这样写直接 local_feat global_featoutputlocal_feat*torch.sigmoid(gate)global_feat*(1-torch.sigmoid(gate))第二个锚点采样的实现全局注意力里的锚点采样我一开始用torch.nn.functional.interpolate做下采样结果梯度传不回去。正确的做法是用一个步长大于1的卷积来做空间压缩这样锚点位置是可学习的。卷积核大小建议用3x3步长根据锚点数量来定——我试过步长4锚点太多计算量没降下来步长8锚点太少全局信息不够。第三个相对位置编码DAT在局部注意力里用了相对位置编码这个和SwinIR一样。但全局注意力里因为锚点和像素的位置关系是动态的不能用固定的位置编码。DAT的做法是根据锚点的空间坐标和像素的坐标动态生成一个位置偏置。这个偏置是通过一个MLP网络学习的输入是锚点和像素的坐标差。训练DAT的实战经验说几个我踩过的坑希望对你有帮助学习率策略DAT比普通Transformer更难训。我试过Cosine Annealing效果不好loss震荡得厉害。后来换成WarmupStep Decay前5个epoch用1e-4预热然后降到1e-4每30个epoch衰减0.5倍。这个配置在DIV2K上跑200个epochPSNR能到30.8dB。Batch SizeDAT的双注意力机制显存消耗比SwinIR大30%左右。我的RTX 3090batch size从16降到12才能跑起来。如果显存不够可以尝试把全局注意力的锚点数量再减半PSNR损失不到0.1dB。数据增强这个容易被忽略。DAT对旋转和翻转特别敏感——我试过不加数据增强PSNR只有30.2dB加了随机旋转90度、水平翻转、垂直翻转直接跳到30.6dB。原因是双注意力机制对方向性纹理的建模能力很强数据增强能帮助它学到旋转不变性。DAT的局限性说点实话DAT不是万能的。我测试了几个场景极端低分辨率当输入分辨率低于32x32时全局注意力分支几乎失效因为锚点数量太少采样的信息不够。这时候DAT的效果反而不如SwinIR。视频超分DAT是单帧模型没有考虑时序信息。我试过把DAT直接用在视频超分上每帧独立处理结果闪烁严重。后来加了时序注意力模块才解决但那已经是另一个故事了。大尺度超分8倍超分时DAT的细节重建能力不如专门设计的CNN模型比如RCAN。原因是全局注意力在超大尺度下容易产生伪影特别是高频纹理区域。个人经验性建议如果你正在做超分研究DAT值得一试但有几个前提第一你的任务需要全局上下文。比如人脸超分、文本超分、遥感图像超分这些场景下DAT的优势很明显。如果是显微镜图像超分局部细节更重要SwinIR可能更合适。第二你的计算资源够用。DAT的显存消耗比SwinIR高但比ViT低得多。如果只有一块2080Ti建议先用SwinIR或者把DAT的通道数减半。第三别盲目堆层数。DAT的最佳深度是12-18层超过20层后收益递减。我试过24层的DATPSNR只比12层的高0.05dB训练时间却翻了一倍。最后说个玄学DAT对初始化特别敏感。我试过不同的随机种子PSNR波动有0.2dB。建议多跑几个种子取平均或者用预训练权重做finetune。DAT不是超分的终点但它证明了局部全局的双注意力范式在超分任务上的有效性。后续的HAT、ELAN等模型都是在DAT的基础上改进的。理解了DAT你就掌握了当前超分Transformer的核心设计思路。
)
