VPNeXt简化视觉Transformer解码,提升分割精度。
VPNeXt -- Rethinking Dense Decoding for Plain Vision Transformer
- 用视觉上下文重放机制替代复杂解码器,效率更高。
- 引入ViTUp模块,利用真实特征金字塔,优于传统模拟金字塔。
- 在VOC2012上突破长期存在的mIoU瓶颈,刷新记录。
我们提出VPNeXt,一种面向Plain Vision Transformer(ViT)的新模型。与以往研究采用同质范式不同,VPNeXt从密集表示角度重新思考了ViT的应用。针对两个核心问题:(1) 是否必须使用复杂的Transformer Mask Decoder才能获得良好表征?(2) Plain ViT是否真需依赖模拟金字塔进行上采样?对于前者,我们分析了原生Transformer Decoder有效性的潜在原因,并提出视觉上下文重放(VCR)机制,实现相似效果且更高效;对于后者,提出ViTUp模块,充分挖掘被忽视的ViT真实金字塔特征,实现优于早期模拟金字塔的上采样性能,这是首次在Plain ViT语义分割中实现此类功能。通过消融实验验证各模块有效性,并在对比实验与可视化中表明,VPNeXt以简单高效设计达到当前最优性能。尤其在VOC2012数据集上显著突破长期存在的mIoU瓶颈,创下最大改进纪录,超越自2015年以来的水平。
原文摘要 · Abstract (English)
We present VPNeXt, a new and simple model for the Plain Vision Transformer (ViT). Unlike the many related studies that share the same homogeneous paradigms, VPNeXt offers a fresh perspective on dense representation based on ViT. In more detail, the proposed VPNeXt addressed two concerns about the existing paradigm: (1) Is it necessary to use a complex Transformer Mask Decoder architecture to obtain good representations? (2) Does the Plain ViT really need to depend on the mock pyramid feature for upsampling? For (1), we investigated the potential underlying reasons that contributed to the effectiveness of the Transformer Decoder and introduced the Visual Context Replay (VCR) to achieve similar effects efficiently. For (2), we introduced the ViTUp module. This module fully utilizes the previously overlooked ViT real pyramid feature to achieve better upsampling results compared to the earlier mock pyramid feature. This represents the first instance of such functionality in the field of semantic segmentation for Plain ViT. We performed ablation studies on related modules to verify their effectiveness gradually. We conducted relevant comparative experiments and visualizations to show that VPNeXt achieved state-of-the-art performance with a simple and effective design. Moreover, the proposed VPNeXt significantly exceeded the long-established mIoU wall/barrier of the VOC2012 dataset, setting a new state-of-the-art by a large margin, which also stands as the largest improvement since 2015.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。