arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2201.01615cs.CV

Lawin Transformer:通过大窗口注意力利用多尺度表示改进语义分割Transformer

Lawin Transformer: Improving Semantic Segmentation Transformer with Multi-Scale Representations via Large Window Attention

Beijing University of Posts and Telecommunications(北京邮电大学)

Haotian Yan, Chuang Zhang, Ming Wu

AI总结:

本文提出Lawin Transformer,通过大窗口注意力将多尺度表示引入语义分割ViT,并结合LawinASPP解码器,在Cityscapes、ADE20K和COCO-Stuff上取得新SOTA并提升效率。

摘要:

多尺度表示对于语义分割至关重要。该领域已经见证了利用多尺度上下文信息的语义分割卷积神经网络(CNN)的蓬勃发展。受视觉Transformer(ViT)在图像分类中强大能力的启发,最近提出了一些语义分割ViT,其中大多数取得了令人印象深刻的结果,但以计算经济性为代价。在本文中,我们成功地将多尺度表示通过窗口注意力机制引入语义分割ViT,并进一步提高了性能和效率。为此,我们引入了大窗口注意力,它允许局部窗口以很小的计算开销查询更大范围的上下文窗口。通过调节上下文区域与查询区域的比率,我们使大窗口注意力能够捕获多个尺度的上下文信息。此外,采用空间金字塔池化框架与大窗口注意力协同工作,提出了一种新颖的解码器,名为大窗口注意力空间金字塔池化(LawinASPP),用于语义分割ViT。我们得到的ViT,即Lawin Transformer,由一个高效的分层视觉Transformer(HVT)作为编码器和一个LawinASPP作为解码器组成。实验结果表明,与现有方法相比,Lawin Transformer提供了更高的效率。Lawin Transformer进一步在Cityscapes(84.4% mIoU)、ADE20K(56.2% mIoU)和COCO-Stuff数据集上取得了新的最先进性能。代码将在https://github.com/yan-hao-tian/lawin发布。

英文摘要:

Multi-scale representations are crucial for semantic segmentation. The community has witnessed the flourish of semantic segmentation convolutional neural networks (CNN) exploiting multi-scale contextual information. Motivated by that the vision transformer (ViT) is powerful in image classification, some semantic segmentation ViTs are recently proposed, most of them attaining impressive results but at a cost of computational economy. In this paper, we succeed in introducing multi-scale representations into semantic segmentation ViT via window attention mechanism and further improves the performance and efficiency. To this end, we introduce large window attention which allows the local window to query a larger area of context window at only a little computation overhead. By regulating the ratio of the context area to the query area, we enable the $\textit{large window attention}$ to capture the contextual information at multiple scales. Moreover, the framework of spatial pyramid pooling is adopted to collaborate with $\textit{the large window attention}$, which presents a novel decoder named $\textbf{la}$rge $\textbf{win}$dow attention spatial pyramid pooling (LawinASPP) for semantic segmentation ViT. Our resulting ViT, Lawin Transformer, is composed of an efficient hierachical vision transformer (HVT) as encoder and a LawinASPP as decoder. The empirical results demonstrate that Lawin Transformer offers an improved efficiency compared to the existing method. Lawin Transformer further sets new state-of-the-art performance on Cityscapes (84.4% mIoU), ADE20K (56.2% mIoU) and COCO-Stuff datasets. The code will be released at https://github.com/yan-hao-tian/lawin

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑