arXiv:2410.08021cs.CV2024-10NeurIPS被引 41

用统一模型直接定位并分割图像中的目标,效果超越现有方法。

OneRef: Unified One-tower Expression Grounding and Segmentation with Mask Referring Modeling

  • 构建共享编码的单塔结构,统一视觉与语言特征空间。
  • 提出掩码指代建模,精准捕捉图文间的指代关系。
  • 无需复杂融合模块,适合追求简洁高效的开发者。

受视觉与语言独立编码的限制,现有定位与指代分割方法严重依赖大型Transformer融合编解码器及多种早期交互技术。同时,当前掩码视觉语言建模(MVLM)难以捕捉指代任务中图像-文本间的细微指代关系。本文提出OneRef,一种基于模态共享单塔Transformer的极简指代框架,统一视觉与语言特征空间。为建模指代关系,引入新型MVLM范式——掩码指代建模(MRefM),包含指代感知的掩码图像建模与指代感知的掩码语言建模,二者不仅能重建模态相关内容,还能跨模态建模指代内容。MRefM中提出指代感知的动态图像遮蔽策略,关注被指区域而非固定比例或通用随机遮蔽。借助统一的视觉语言特征空间与MRefM对指代关系的建模能力,该方法可直接回归指代结果,无需复杂技术。实验表明,该方法在定位与分割任务上持续优于现有方法,达到最先进性能,为未来研究提供重要启示。代码与模型已开源。

原文摘要 · Abstract (English)

Constrained by the separate encoding of vision and language, existing grounding and referring segmentation works heavily rely on bulky Transformer-based fusion en-/decoders and a variety of early-stage interaction technologies. Simultaneously, the current mask visual language modeling (MVLM) fails to capture the nuanced referential relationship between image-text in referring tasks. In this paper, we propose OneRef, a minimalist referring framework built on the modality-shared one-tower transformer that unifies the visual and linguistic feature spaces. To modeling the referential relationship, we introduce a novel MVLM paradigm called Mask Referring Modeling (MRefM), which encompasses both referring-aware mask image modeling and referring-aware mask language modeling. Both modules not only reconstruct modality-related content but also cross-modal referring content. Within MRefM, we propose a referring-aware dynamic image masking strategy that is aware of the referred region rather than relying on fixed ratios or generic random masking schemes. By leveraging the unified visual language feature space and incorporating MRefM's ability to model the referential relations, our approach enables direct regression of the referring results without resorting to various complex techniques. Our method consistently surpasses existing approaches and achieves SoTA performance on both grounding and segmentation tasks, providing valuable insights for future research. Our code and models are available at https://github.com/linhuixiao/OneRef.

指代分割单塔模型多模态建模

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