We present in this paper a new architecture, the Pattern Attention Transformer (PAT), that is composed of the new doughnut kernel. Compared with tokens in the NLP field, Transformer in computer vision has the problem of handling the high resolution of pixels in images. In ViT, an image is cut into square-shaped patches. As the follow-up of ViT, Swin Transformer proposes an additional step of shifting to decrease the existence of fixed boundaries, which also incurs 'two connected Swin Transformer blocks' as the minimum unit of the model. Inheriting the patch/window idea, our doughnut kernel enhances the design of patches further. It replaces the line-cut boundaries with two types of areas: sensor and updating, which is based on the comprehension of self-attention (named QKVA grid). The doughnut kernel also brings a new topic about the shape of kernels beyond square. To verify its performance on image classification, PAT is designed with Transformer blocks of regular octagon shape doughnut kernels. Its architecture is lighter: the minimum pattern attention layer is only one for each stage. Under similar complexity of computation, its performances on ImageNet 1K reach higher throughput (+10%) and surpass Swin Transformer (+0.1 acc1).
翻译:本文提出了一种新的网络结构,即Pattern Attention Transformer(PAT),它是由新的多福糖球体卷积核构成的。与NLP领域中的标记相比,计算机视觉中的Transformer处理高分辨率像素图像的问题。在ViT中,将图像切成正方形的块。作为ViT的后续,Swin Transformer提出了一个附加步骤来减少固定边界的存在,这也产生了“两个连接的Swin变换器块”作为模型的最小单元。继承了补丁/窗口思想,我们的多福糖球体卷积核进一步增强了补丁的设计。它用传感器和更新两种区域替换了线切边界,这是基于自我关注(名为QKVA网格)的理解。多福糖球体卷积核还带来了关于卷积核形状的新话题,超越正方形形状的设计。为了验证其在图像分类方面的性能,PAT是由规则八边形形状的多福糖球体卷积核的Transformer块组成的。该架构更轻巧:每个阶段的最小pattern attention层仅为1。在类似的计算复杂性下,ImageNet 1K上的性能与Swin Transformer相比具有更高的吞吐量(+10%),并且超过了Swin Transformer(+0.1 acc1)。