不微调、不改结构,用扩散模型注意力直接做零样本指代分割
RefAM: Attention Magnets for Zero-Shot Referral Segmentation
- 利用扩散模型的注意力分数做零样本指代分割,无需额外训练
- 发现停用词聚集注意力,可过滤降噪;深层存在全局注意力陷阱,可抑制或重定向
- 通过添加停用词重新分配注意力,生成更精准的定位热图,适合无标注场景
现有指代分割方法多依赖微调或多模型组合,常需额外训练与架构改动。本文提出一种新方法,直接利用生成式扩散模型中扩散变压器的特征与注意力分数,无需任何训练或结构修改。为系统评估,我们扩展了涵盖图像与视频的视觉-语言定位基准。关键发现:停用词会聚集多余注意力,可被过滤以降低噪声;深层网络中出现全局注意力陷阱(GAS),可安全抑制或重定向至辅助标记,提升定位精度。进一步提出注意力重分配策略,通过添加停用词将背景激活划分为更小簇,生成更锐利、局部化的热图。基于此构建RefAM框架,结合交叉注意力图、GAS处理与重分配策略,在多个零样本指代图像与视频分割基准上表现优异,超越多数已有方法,实现无需微调、无额外组件、无复杂推理的新最佳性能。
原文摘要 · Abstract (English)
Most existing approaches to referring segmentation achieve strong performance only through fine-tuning or by composing multiple pre-trained models, often at the cost of additional training and architectural modifications. Meanwhile, large-scale generative diffusion models encode rich semantic information, making them attractive as general-purpose feature extractors. In this work, we introduce a new method that directly exploits features, attention scores, from diffusion transformers for downstream tasks, requiring neither architectural modifications nor additional training. To systematically evaluate these features, we extend benchmarks with vision-language grounding tasks spanning both images and videos. Our key insight is that stop words act as attention magnets: they accumulate surplus attention and can be filtered to reduce noise. Moreover, we identify global attention sinks (GAS) emerging in deeper layers and show that they can be safely suppressed or redirected onto auxiliary tokens, leading to sharper and more accurate grounding maps. We further propose an attention redistribution strategy, where appended stop words partition background activations into smaller clusters, yielding sharper and more localized heatmaps. Building on these findings, we develop RefAM, a simple training-free grounding framework that combines cross-attention maps, GAS handling, and redistribution. Across zero-shot referring image and video segmentation benchmarks, our approach achieves strong performance and surpasses prior methods on most datasets, establishing a new state of the art without fine-tuning, additional components and complex reasoning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。