arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2409.08251cs.CV

面向全景叙事定位的冻结文本到图像扩散模型动态提示方法

Dynamic Prompting of Frozen Text-to-Image Diffusion Models for Panoptic Narrative Grounding

School of Artificial Intelligence, Beihang University(北京航空航天大学人工智能学院);School of Computer Science and Information Engineering, Hefei University of Technology(合肥工业大学计算机与信息工程学院);School of Software, Beihang University(北京航空航天大学软件学院);Meituan(美团);Institute of Information Engineering, Chinese Academy of Sciences(中国科学院信息工程学院)

Hongyu Li, Tianrui Hui, Zihan Ding, Jing Zhang, Bin Ma, Xiaoming Wei, Jizhong Han, Si Liu

AI总结:

针对全景叙事定位任务中冻结扩散模型静态提示存在任务差距、视觉-语言交互不足的问题,提出抽取-注入短语适配器动态更新提示并回注多模态线索,结合多级互聚合模块优化分割,在基准上取得新SOTA。

摘要:

全景叙事定位(Panoptic Narrative Grounding,PNG)的核心目标是实现细粒度图文对齐,该任务要求根据叙述性字幕对提及的目标进行全景分割。以往的判别式方法通过全景分割预训练或CLIP模型适配,仅能实现弱对齐或粗粒度对齐。随着文本到图像扩散模型的最新进展,已有多项研究证明这类模型可通过交叉注意力图实现细粒度图文对齐,并能提升通用分割性能。然而,直接将短语特征作为静态提示,将冻结扩散模型应用于PNG任务仍存在较大的任务差距,且视觉-语言交互不充分,导致性能不佳。为此,我们提出在扩散模型UNet中加入抽取-注入短语适配器(Extractive-Injective Phrase Adapter,EIPA)旁路,利用图像特征动态更新短语提示,并将多模态线索回注,从而更充分地发挥扩散模型的细粒度图文对齐能力。此外,我们还设计了多级互聚合(Multi-Level Mutual Aggregation,MLMA)模块,对多级图像特征与短语特征进行双向融合,以优化分割结果。在PNG基准上的大量实验表明,本文方法取得了新的最优性能。

英文摘要:

Panoptic narrative grounding (PNG), whose core target is fine-grained image-text alignment, requires a panoptic segmentation of referred objects given a narrative caption. Previous discriminative methods achieve only weak or coarse-grained alignment by panoptic segmentation pretraining or CLIP model adaptation. Given the recent progress of text-to-image Diffusion models, several works have shown their capability to achieve fine-grained image-text alignment through cross-attention maps and improved general segmentation performance. However, the direct use of phrase features as static prompts to apply frozen Diffusion models to the PNG task still suffers from a large task gap and insufficient vision-language interaction, yielding inferior performance. Therefore, we propose an Extractive-Injective Phrase Adapter (EIPA) bypass within the Diffusion UNet to dynamically update phrase prompts with image features and inject the multimodal cues back, which leverages the fine-grained image-text alignment capability of Diffusion models more sufficiently. In addition, we also design a Multi-Level Mutual Aggregation (MLMA) module to reciprocally fuse multi-level image and phrase features for segmentation refinement. Extensive experiments on the PNG benchmark show that our method achieves new state-of-the-art performance.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑