arXiv:2505.19795cs.CVcs.AI2025-05被引 4

用点定位提升ViT图像分割精度,无需额外训练

The Missing Point in Vision Transformers for Universal Image Segmentation

  • 分两阶段:先生成无类别掩码,再用ViT聚焦中心点分类
  • 在ADE20K上达54.0 PQ,Cityscapes上87.4 mIoU,性能领先
  • 仅需粗标注或边界框,降低标注成本,适合通用分割场景

图像分割在计算机视觉中仍具挑战性,需兼顾鲁棒的掩码生成与精确分类。现有基于掩码的方法虽能捕捉全局上下文并生成高质量掩码,但在模糊边界和类别不平衡情况下准确分类仍存难题。本文提出ViT-P,一种两阶段分割框架,将掩码生成与分类解耦:第一阶段通过提案生成器产生无类别掩码提案;第二阶段基于视觉变换器(ViT)构建点分类模型,聚焦掩码中心点以优化预测。ViT-P作为免预训练适配器,可无缝集成各类预训练ViT,不改变其结构,适用于密集预测任务。此外,我们证明粗略标注或边界框即可有效提升分类性能,无需额外细粒度标注数据集训练,显著降低标注成本。在COCO、ADE20K和Cityscapes上的大量实验验证了其有效性,在ADE20K全景分割上达到54.0 PQ,Cityscapes语义分割达87.4 mIoU,ADE20K语义分割达63.6 mIoU,均达当前最优。代码与预训练模型已开源。

原文摘要 · Abstract (English)

Image segmentation remains a challenging task in computer vision, demanding robust mask generation and precise classification. Recent mask-based approaches yield high-quality masks by capturing global context. However, accurately classifying these masks, especially in the presence of ambiguous boundaries and imbalanced class distributions, remains an open challenge. In this work, we introduce ViT-P, a novel two-stage segmentation framework that decouples mask generation from classification. The first stage employs a proposal generator to produce class-agnostic mask proposals, while the second stage utilizes a point-based classification model built on the Vision Transformer (ViT) to refine predictions by focusing on mask central points. ViT-P serves as a pre-training-free adapter, allowing the integration of various pre-trained vision transformers without modifying their architecture, ensuring adaptability to dense prediction tasks. Furthermore, we demonstrate that coarse and bounding box annotations can effectively enhance classification without requiring additional training on fine annotation datasets, reducing annotation costs while maintaining strong performance. Extensive experiments across COCO, ADE20K, and Cityscapes datasets validate the effectiveness of ViT-P, achieving state-of-the-art results with 54.0 PQ on ADE20K panoptic segmentation, 87.4 mIoU on Cityscapes semantic segmentation, and 63.6 mIoU on ADE20K semantic segmentation. The code and pretrained models are available at: https://github.com/sajjad-sh33/ViT-P}{https://github.com/sajjad-sh33/ViT-P.

图像分割ViT点定位低标注成本

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。