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

通过生成图像基础模型的概念蒸馏实现前景-背景分离

Foreground-Background Separation through Concept Distillation from Generative Image Foundation Models

Friedrich–Alexander–Universität Erlangen–Nürnberg(弗里德里希-亚历山大-埃尔兰根-纽伦堡大学);Imperial College London(伦敦帝国学院)

Mischa Dombrowski, Hadrien Reynaud, Matthew Baugh, Bernhard Kainz

AI总结:

提出一种无需分割标签、利用预训练潜在扩散模型生成弱掩码并微调修复任务的方法,实现通用前景-背景分割,性能优于先前方法并接近全监督。代码已开源。

摘要:

为对象分割整理数据集是一项困难的任务。随着大规模预训练生成模型的出现,条件图像生成在结果质量和易用性方面得到了显著提升。在本文中,我们提出了一种新颖的方法,能够从简单的文本描述中生成通用的前景-背景分割模型,而无需分割标签。我们利用并探索预训练的潜在扩散模型,自动为概念和对象生成弱分割掩码。这些掩码随后用于在修复任务上微调扩散模型,从而能够细粒度地移除对象,同时提供合成的前景和背景数据集。我们证明,使用这种方法在判别性和生成性能上均优于先前的方法,并在无需像素级对象标签的情况下缩小了与全监督训练的差距。我们展示了在分割四种不同对象(人类、狗、汽车、鸟类)的任务以及医学图像分析中的用例场景的结果。代码可在 https://github.com/MischaD/fobadiffusion 获取。

英文摘要:

Curating datasets for object segmentation is a difficult task. With the advent of large-scale pre-trained generative models, conditional image generation has been given a significant boost in result quality and ease of use. In this paper, we present a novel method that enables the generation of general foreground-background segmentation models from simple textual descriptions, without requiring segmentation labels. We leverage and explore pre-trained latent diffusion models, to automatically generate weak segmentation masks for concepts and objects. The masks are then used to fine-tune the diffusion model on an inpainting task, which enables fine-grained removal of the object, while at the same time providing a synthetic foreground and background dataset. We demonstrate that using this method beats previous methods in both discriminative and generative performance and closes the gap with fully supervised training while requiring no pixel-wise object labels. We show results on the task of segmenting four different objects (humans, dogs, cars, birds) and a use case scenario in medical image analysis. The code is available at https://github.com/MischaD/fobadiffusion.

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

↑