无需训练,通过迭代优化注意力图实现高质量图像分割
iSeg: An Iterative Refinement-based Framework for Training-free Segmentation

- 用梯度下降降低自注意力熵,抑制无关信息干扰
- 在Cityscapes上比现有方法提升3.8%的语义分割准确率
- 支持多种图像和交互方式,适合零样本分割场景
Stable Diffusion展现出强大的文本到图像生成能力,暗示其蕴含丰富的语义线索用于对象分组。研究人员已尝试利用Stable Diffusion实现无需训练的分割。现有方法多仅一次使用自注意力图来优化交叉注意力图,表明自注意力图包含有用语义信息。为充分挖掘自注意力图潜力,本文开展深入实验,研究以自注意力图迭代优化交叉注意力图的有效性,提出名为iSeg的训练自由分割框架。iSeg引入熵减少的自注意力模块,采用梯度下降策略降低自注意力图熵值,从而抑制与无关全局信息对应的弱响应。结合该模块,iSeg通过迭代优化稳定提升细化后的交叉注意力图。此外,设计类别增强的交叉注意力模块,生成更精确的初始交叉注意力图,促进迭代优化效果。大量实验在多个数据集和多样化分割任务上验证了所提方法优势,在Cityscapes上的无监督语义分割任务中,相比文献中最佳训练自由方法,mIoU绝对提升3.8%。同时,iSeg可支持不同图像类型与交互形式,项目地址:https://linsun449.github.io/iSeg。
原文摘要 · Abstract (English)
Stable diffusion has demonstrated strong image synthesis ability to given text descriptions, suggesting it to contain strong semantic clue for grouping objects. The researchers have explored employing stable diffusion for training-free segmentation. Most existing approaches refine cross-attention map by self-attention map once, demonstrating that self-attention map contains useful semantic information to improve segmentation. To fully utilize self-attention map, we present a deep experimental analysis on iteratively refining cross-attention map with self-attention map, and propose an effective iterative refinement framework for training-free segmentation, named iSeg. The proposed iSeg introduces an entropy-reduced self-attention module that utilizes a gradient descent scheme to reduce the entropy of self-attention map, thereby suppressing the weak responses corresponding to irrelevant global information. Leveraging the entropy-reduced self-attention module, our iSeg stably improves refined cross-attention map with iterative refinement. Further, we design a category-enhanced cross-attention module to generate accurate cross-attention map, providing a better initial input for iterative refinement. Extensive experiments across different datasets and diverse segmentation tasks reveal the merits of proposed contributions, leading to promising performance on diverse segmentation tasks. For unsupervised semantic segmentation on Cityscapes, our iSeg achieves an absolute gain of 3.8% in terms of mIoU compared to the best existing training-free approach in literature. Moreover, our proposed iSeg can support segmentation with different kinds of images and interactions. The project is available at https://linsun449.github.io/iSeg.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。