通过分步视觉推理提升开放词汇分割的准确性和可解释性
OpenSeg-R: Improving Open-Vocabulary Segmentation via Step-by-Step Visual Reasoning
- 用大模型生成图像通用与特定的分步推理链
- 在五个数据集上显著优于现有方法
- 适合需要高精度与可解释性的视觉任务
开放词汇分割(OVS)因其能泛化到预定义类别之外而受到关注。然而,现有方法通常采用简单前向推理,缺乏显式推理过程和可解释性,导致在开放世界中难以区分相似类别。为此,我们提出基于分步视觉推理的开放词汇分割框架OpenSeg-R。该框架利用大型多模态模型,在分割前进行层次化视觉推理,为每张图像生成通用与图像特定的推理链,形成从粗到细的结构化三元组来解释物体的视觉依据。基于这些推理步骤,构建详细描述提示词并输入分割器,生成更精确的分割掩码。据我们所知,OpenSeg-R是首个将显式分步推理引入OVS的框架。实验表明,它在五个基准数据集上的开放词汇语义分割任务中显著超越当前最优方法;在开放词汇全景分割中,各项指标均实现一致提升。定性结果进一步验证了该推理引导框架在提升分割精度与可解释性方面的有效性。代码已公开于https://github.com/Hanzy1996/OpenSeg-R。
原文摘要 · Abstract (English)
Open-Vocabulary Segmentation (OVS) has drawn increasing attention for its capacity to generalize segmentation beyond predefined categories. However, existing methods typically predict segmentation masks with simple forward inference, lacking explicit reasoning and interpretability. This makes it challenging for OVS model to distinguish similar categories in open-world settings due to the lack of contextual understanding and discriminative visual cues. To address this limitation, we propose a step-by-step visual reasoning framework for open-vocabulary segmentation, named OpenSeg-R. The proposed OpenSeg-R leverages Large Multimodal Models (LMMs) to perform hierarchical visual reasoning before segmentation. Specifically, we generate both generic and image-specific reasoning for each image, forming structured triplets that explain the visual reason for objects in a coarse-to-fine manner. Based on these reasoning steps, we can compose detailed description prompts, and feed them to the segmentor to produce more accurate segmentation masks. To the best of our knowledge, OpenSeg-R is the first framework to introduce explicit step-by-step visual reasoning into OVS. Experimental results demonstrate that OpenSeg-R significantly outperforms state-of-the-art methods on open-vocabulary semantic segmentation across five benchmark datasets. Moreover, it achieves consistent gains across all metrics on open-vocabulary panoptic segmentation. Qualitative results further highlight the effectiveness of our reasoning-guided framework in improving both segmentation precision and interpretability. Our code is publicly available at https://github.com/Hanzy1996/OpenSeg-R.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。