arXiv:2412.08628cs.CV2024-12中稿 · AAAI

提出高效开源词汇全景分割框架,推理速度超当前方法19倍。

EOV-Seg: Efficient Open-Vocabulary Panoptic Segmentation

  • 单阶段共享结构,用词汇感知选择模块减少特征交互负担。
  • 引入双向动态嵌入专家,利用ViT-CLIP空间感知能力提升效率。
  • 仅用COCO训练,在ADE20K上达24.5 PQ,推理快至23.8 FPS。

开放词汇全景分割旨在对多样场景中无界词汇内容进行分割与分类。现有方法多采用两阶段或单阶段框架:前者多次裁剪图像并提取特征,后者依赖重型掩码解码器通过多层Transformer自注意力和交叉注意力弥补空间信息缺失,均带来显著计算开销。为解决效率瓶颈,本文提出EOV-Seg,首个面向高效的单阶段、共享、空间感知开放词汇全景分割框架。创新点包括:1)词汇感知选择(VAS)模块,增强视觉特征语义理解并减轻掩码解码器负担;2)双向动态嵌入专家(TDEE),高效利用基于ViT的CLIP主干的空间感知能力。在仅使用COCO训练数据的情况下,EOV-Seg在ADE20K上实现24.5 PQ、32.1 mIoU,推理速度达11.6 FPS,比当前最优方法快4-19倍。尤其使用ResNet50主干时,仅71M参数即可在单块RTX 3090上达到23.8 FPS。

原文摘要 · Abstract (English)

Open-vocabulary panoptic segmentation aims to segment and classify everything in diverse scenes across an unbounded vocabulary. Existing methods typically employ two-stage or single-stage framework. The two-stage framework involves cropping the image multiple times using masks generated by a mask generator, followed by feature extraction, while the single-stage framework relies on a heavyweight mask decoder to make up for the lack of spatial position information through self-attention and cross-attention in multiple stacked Transformer blocks. Both methods incur substantial computational overhead, thereby hindering the efficiency of model inference. To fill the gap in efficiency, we propose EOV-Seg, a novel single-stage, shared, efficient, and spatialaware framework designed for open-vocabulary panoptic segmentation. Specifically, EOV-Seg innovates in two aspects. First, a Vocabulary-Aware Selection (VAS) module is proposed to improve the semantic comprehension of visual aggregated features and alleviate the feature interaction burden on the mask decoder. Second, we introduce a Two-way Dynamic Embedding Experts (TDEE), which efficiently utilizes the spatial awareness capabilities of ViT-based CLIP backbone. To the best of our knowledge, EOV-Seg is the first open-vocabulary panoptic segmentation framework towards efficiency, which runs faster and achieves competitive performance compared with state-of-the-art methods. Specifically, with COCO training only, EOV-Seg achieves 24.5 PQ, 32.1 mIoU, and 11.6 FPS on the ADE20K dataset and the inference time of EOV-Seg is 4-19 times faster than state-of-theart methods. Especially, equipped with ResNet50 backbone, EOV-Seg runs 23.8 FPS with only 71M parameters on a single RTX 3090 GPU. Code is available at https://github.com/nhw649/EOV-Seg.

全景分割开放词汇高效模型ViT-CLIP

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