用框选提示生成多种语义解释,实现图像协同分割与描述
SGDiff: Scene Graph Guided Diffusion Model for Image Collaborative SegCaptioning

- 通过场景图引导扩散模型,关联预测文本描述和分割掩码
- 仅需一个边界框提示,就能生成多组对齐的图文结果
- 适合需要灵活交互的视觉理解任务,如内容创作与辅助标注
可控图像语义理解任务(如图像描述或分割)通常需要用户输入复杂提示(如文本或边界框),带来高成本且输出信息有限。本文提出新任务“图像协同分割与描述”(SegCaptioning),旨在将简单的提示(如对象边界框)转化为多组语义解释(描述+掩码对),供用户灵活选择。该任务挑战在于:从极简提示中准确捕捉用户意图,并同时生成多个语义一致的描述与掩码。为此,提出一种场景图引导的扩散模型(SGDiff)。首先设计提示中心的场景图适配器,将用户提示映射为结构化场景图以捕获意图;随后采用融合场景图引导的双模态变压器的扩散过程,揭示图文之间的复杂关联;最后引入多实体对比学习损失,显式对齐视觉与文本实体,确保结果对齐。在两个数据集上的大量实验表明,SGDiff在该任务上表现优异,仅用少量提示即可获得高质量的分割与描述结果。
原文摘要 · Abstract (English)
Controllable image semantic understanding tasks, such as captioning or segmentation, necessitate users to input a prompt (e.g., text or bounding boxes) to predict a unique outcome, presenting challenges such as high-cost prompt input or limited information output. This paper introduces a new task ``Image Collaborative Segmentation and Captioning'' (SegCaptioning), which aims to translate a straightforward prompt, like a bounding box around an object, into diverse semantic interpretations represented by (caption, masks) pairs, allowing flexible result selection by users. This task poses significant challenges, including accurately capturing a user's intention from a minimal prompt while simultaneously predicting multiple semantically aligned caption words and masks. Technically, we propose a novel Scene Graph Guided Diffusion Model that leverages structured scene graph features for correlated mask-caption prediction. Initially, we introduce a Prompt-Centric Scene Graph Adaptor to map a user's prompt to a scene graph, effectively capturing his intention. Subsequently, we employ a diffusion process incorporating a Scene Graph Guided Bimodal Transformer to predict correlated caption-mask pairs by uncovering intricate correlations between them. To ensure accurate alignment, we design a Multi-Entities Contrastive Learning loss to explicitly align visual and textual entities by considering inter-modal similarity, resulting in well-aligned caption-mask pairs. Extensive experiments conducted on two datasets demonstrate that SGDiff achieves superior performance in SegCaptioning, yielding promising results for both captioning and segmentation tasks with minimal prompt input.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。