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

IncepFormer:一种高效的基于金字塔池的语义分割初始Transformer

IncepFormer: Efficient Inception Transformer with Pyramid Pooling for Semantic Segmentation

Nanjing University of Aeronautics and Astronautics(南京航空航天大学);Shenzhen Technology University(深圳技术大学)

Lihua Fu, Haoyue Tian, Xiangping Bryce Zhai, Pan Gao, Xiaojiang Peng

摘要:

语义分割通常受益于全球范围内,精细的本地化信息,多尺度功能等,以推进基于转换器的分割器与这些方面,我们提出了一个简单而强大的语义分割架构,称为IncepFormer。IncepFormer有以下两个关键贡献。首先,它介绍了一种新的金字塔结构的Transformer编码器,收获全球的背景和罚款本地化功能同时进行。这些特征被连接并馈送到卷积层中以进行最终的每像素预测。其次,IncepFormer集成了一个类似Inception-like的架构,具有深度卷积,并且在每个自注意层中集成了一个轻量级的前馈模块,有效地获得了丰富的局部多尺度对象特征。在五个基准上进行的大量实验表明,我们的IncepFormer在准确性和速度方面都优于最先进的方法,例如,1)我们的IncepFormer-S在ADE 20 K上实现了47.7%的mIoU,比现有的最佳方法高出1%,而仅花费一半的参数和更少的FLOP。2)我们的IncepFormer-B最终在具有39.6M参数的Cityscapes数据集上实现了82.0%的mIoU。代码可用:github.com/shendu0321/IncepFormer。

英文摘要:

Semantic segmentation usually benefits from global contexts, fine localisation information, multi-scale features, etc. To advance Transformer-based segmenters with these aspects, we present a simple yet powerful semantic segmentation architecture, termed as IncepFormer. IncepFormer has two critical contributions as following. First, it introduces a novel pyramid structured Transformer encoder which harvests global context and fine localisation features simultaneously. These features are concatenated and fed into a convolution layer for final per-pixel prediction. Second, IncepFormer integrates an Inception-like architecture with depth-wise convolutions, and a light-weight feed-forward module in each self-attention layer, efficiently obtaining rich local multi-scale object features. Extensive experiments on five benchmarks show that our IncepFormer is superior to state-of-the-art methods in both accuracy and speed, e.g., 1) our IncepFormer-S achieves 47.7% mIoU on ADE20K which outperforms the existing best method by 1% while only costs half parameters and fewer FLOPs. 2) Our IncepFormer-B finally achieves 82.0% mIoU on Cityscapes dataset with 39.6M parameters. Code is available:github.com/shendu0321/IncepFormer.

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

↑