arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2412.20439cs.CV

用于弱监督语义分割的图像增强智能体

Image Augmentation Agent for Weakly Supervised Semantic Segmentation

Xi'an Jiaotong-Liverpool University(西安交通大学利物浦大学);University of Liverpool(利物浦大学);Microsoft(微软公司)

Wangyu Wu, Xianglin Qiu, Siqi Song, Zhenhong Chen, Xiaowei Huang, Fei Ma, Jimin Xiao

AI总结:

针对弱监督语义分割(WSSS)受固定数据集局限的问题,提出图像增强智能体(IAA),利用大型语言模型(LLMs)和扩散模型生成额外图像,通过提示自优化机制和在线过滤器提升生成质量,在PASCAL VOC 2012和MS COCO 2014数据集上超越现有方法。

摘要:

弱监督语义分割(WSSS)仅使用图像级标签就取得了显著进展。然而,大多数现有WSSS方法专注于设计新的网络结构和损失函数以生成更准确的密集标签,却忽视了固定数据集带来的局限性,这会制约性能提升。我们认为,更多样化的可训练图像能为WSSS提供更丰富的信息,帮助模型理解更全面的语义模式。因此,本文提出一种名为图像增强智能体(IAA)的新方法,表明从数据生成角度增强WSSS是可行的。IAA主要设计了一个增强智能体,该智能体利用大型语言模型(LLMs)和扩散模型自动为WSSS生成额外图像。在实践中,为解决LLMs生成提示时的不稳定性,我们开发了一种提示自优化机制,使LLMs能重新评估生成提示的合理性,从而产生更连贯的提示。此外,我们在扩散生成过程中插入了一个在线过滤器,动态确保生成图像的质量和平衡性。实验结果表明,我们的方法在PASCAL VOC 2012和MS COCO 2014数据集上显著超越了最先进的WSSS方法。

英文摘要:

Weakly-supervised semantic segmentation (WSSS) has achieved remarkable progress using only image-level labels. However, most existing WSSS methods focus on designing new network structures and loss functions to generate more accurate dense labels, overlooking the limitations imposed by fixed datasets, which can constrain performance improvements. We argue that more diverse trainable images provides WSSS richer information and help model understand more comprehensive semantic pattern. Therefore in this paper, we introduce a novel approach called Image Augmentation Agent (IAA) which shows that it is possible to enhance WSSS from data generation perspective. IAA mainly design an augmentation agent that leverages large language models (LLMs) and diffusion models to automatically generate additional images for WSSS. In practice, to address the instability in prompt generation by LLMs, we develop a prompt self-refinement mechanism. It allow LLMs to re-evaluate the rationality of generated prompts to produce more coherent prompts. Additionally, we insert an online filter into diffusion generation process to dynamically ensure the quality and balance of generated images. Experimental results show that our method significantly surpasses state-of-the-art WSSS approaches on the PASCAL VOC 2012 and MS COCO 2014 datasets.

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

↑