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

用于语义分割的金字塔融合Transformer

Pyramid Fusion Transformer for Semantic Segmentation

The Chinese University of Hong Kong(香港中文大学);SenseTime Research(商汤科技研究院)

Zipeng Qin, Jianbo Liu, Xiaolin Zhang, Maoqing Tian, Aojun Zhou, Shuai Yi, Hongsheng Li

AI总结:

本文提出金字塔融合Transformer(PFT),通过并行交叉注意力和跨尺度查询间注意力融合多尺度特征,提升掩码级语义分割性能,在ADE20K等数据集上取得领先结果。

摘要:

最近提出的MaskFormer为语义分割任务提供了一个全新的视角:它从流行的像素级分类范式转向了掩码级分类方法。本质上,它生成与类别片段对应的成对概率和掩码,并在推理过程中将它们组合起来得到分割图。在我们的研究中,我们发现基于单尺度特征的逐掩码分类解码器不足以提取可靠的概率或掩码。为了挖掘特征金字塔中丰富的语义信息,我们提出了一种基于Transformer的金字塔融合Transformer(Pyramid Fusion Transformer,PFT),用于具有多尺度特征的逐掩码方法语义分割。所提出的Transformer解码器在可学习查询与特征金字塔中的每个空间特征之间并行执行交叉注意力,并使用跨尺度查询间注意力来交换互补信息。我们在三个广泛使用的语义分割数据集上取得了具有竞争力的性能。特别是在ADE20K验证集上,我们使用Swin-B骨干网络的结果在单尺度和多尺度推理中均超过了使用更大Swin-L骨干网络的MaskFormer,分别达到了54.1 mIoU和55.7 mIoU。使用Swin-L骨干网络,我们实现了单尺度56.1 mIoU和多尺度57.4 mIoU,在该数据集上获得了最先进的性能。在三个广泛使用的语义分割数据集上的大量实验验证了我们所提出方法的有效性。

英文摘要:

The recently proposed MaskFormer gives a refreshed perspective on the task of semantic segmentation: it shifts from the popular pixel-level classification paradigm to a mask-level classification method. In essence, it generates paired probabilities and masks corresponding to category segments and combines them during inference for the segmentation maps. In our study, we find that per-mask classification decoder on top of a single-scale feature is not effective enough to extract reliable probability or mask. To mine for rich semantic information across the feature pyramid, we propose a transformer-based Pyramid Fusion Transformer (PFT) for per-mask approach semantic segmentation with multi-scale features. The proposed transformer decoder performs cross-attention between the learnable queries and each spatial feature from the feature pyramid in parallel and uses cross-scale inter-query attention to exchange complimentary information. We achieve competitive performance on three widely used semantic segmentation datasets. In particular, on ADE20K validation set, our result with Swin-B backbone surpasses that of MaskFormer's with a much larger Swin-L backbone in both single-scale and multi-scale inference, achieving 54.1 mIoU and 55.7 mIoU respectively. Using a Swin-L backbone, we achieve single-scale 56.1 mIoU and multi-scale 57.4 mIoU, obtaining state-of-the-art performance on the dataset. Extensive experiments on three widely used semantic segmentation datasets verify the effectiveness of our proposed method.

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

↑