arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2603.06178cs.CV
收藏

利用生成能力使免训练扩散分割器扩展

Making Training-Free Diffusion Segmentors Scale with the Generative Power

Institute of Information Engineering, CAS(中国科学院信息工程研究所);School of Cyber Security, University of Chinese Academy of Sciences(中国科学院大学网络空间安全学院);State Key Laboratory of AI Safety, Institute of Computing Technology, CAS(中国科学院计算技术研究所人工智能安全国家重点实验室);School of Cyber Science and Tech., Shenzhen Campus of Sun Yat-sen University(中山大学深圳校区网络空间安全学院);Alibaba Group(阿里巴巴集团);School of Computer Science and Tech., University of Chinese Academy of Sciences(中国科学院大学计算机科学与技术学院);Key Laboratory of Big Data Mining and Knowledge Management, CAS(中国科学院大数据挖掘与知识管理重点实验室);Beijing Academy of Artificial Intelligence(北京人工智能研究院)

Benyuan Meng, Qianqian Xu, Zitai Wang, Xiaochun Cao, Longtao Huang, Qingming Huang

AI总结:

本文研究了如何通过提升扩散模型的生成能力来改进免训练的图像分割方法,提出自动聚合和像素级重缩放技术以解决注意力图与全局表示不一致及文本标记间得分不平衡的问题。

摘要:

作为强大的生成模型,文本到图像的扩散模型最近被探索用于判别任务。一项研究专注于在不进一步训练的情况下适应预训练的扩散模型以进行语义分割,从而产生免训练的扩散分割器。这些方法通常依赖于模型注意力层的交叉注意力图,这些图被假设能捕捉图像像素与文本标记之间的语义关系。理想情况下,此类方法应受益于更强大的扩散模型,即更强的生成能力应导致更好的分割。然而,我们发现现有方法往往无法相应扩展。为了理解这一问题,我们识别出两个根本性缺口:(i) 交叉注意力是在多个头和层之间计算的,但这些个体注意力图与统一的全局表示之间存在差异。(ii) 即使有全局图可用,它也不直接转化为准确的语义相关性用于分割,因为不同文本标记之间的得分不平衡。为了弥合这些缺口,我们提出了两种技术:自动聚合和像素级重缩放,这些技术共同使免训练分割能够更好地利用生成能力。我们在标准语义分割基准上评估了我们的方法,并进一步将其整合到生成技术中,证明了改进的性能和广泛的应用性。代码在https://github.com/Darkbblue/goca。

英文摘要:

As powerful generative models, text-to-image diffusion models have recently been explored for discriminative tasks. A line of research focuses on adapting a pre-trained diffusion model to semantic segmentation without any further training, leading to training-free diffusion segmentors. These methods typically rely on cross-attention maps from the model's attention layers, which are assumed to capture semantic relationships between image pixels and text tokens. Ideally, such approaches should benefit from more powerful diffusion models, i.e., stronger generative capability should lead to better segmentation. However, we observe that existing methods often fail to scale accordingly. To understand this issue, we identify two underlying gaps: (i) cross-attention is computed across multiple heads and layers, but there exists a discrepancy between these individual attention maps and a unified global representation. (ii) Even when a global map is available, it does not directly translate to accurate semantic correlation for segmentation, due to score imbalances among different text tokens. To bridge these gaps, we propose two techniques: auto aggregation and per-pixel rescaling, which together enable training-free segmentation to better leverage generative capability. We evaluate our approach on standard semantic segmentation benchmarks and further integrate it into a generative technique, demonstrating both improved performance broad applicability. Codes are at https://github.com/Darkbblue/goca.

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

↑