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

使用去噪扩散隐式模型实现连贯的图像修复

Towards Coherent Image Inpainting Using Denoising Diffusion Implicit Models

UC Santa Barbara(加州大学圣巴巴拉分校);MIT-IBM Watson AI Lab(麻省理工学院-IBM沃森人工智能实验室);IBM Research(IBM研究院);MIT CSAIL(麻省理工学院计算机科学与人工智能实验室)

Guanhua Zhang, Jiabao Ji, Yang Zhang, Mo Yu, Tommi Jaakkola, Shiyu Chang

AI总结:

本文提出COPAINT,利用贝叶斯框架联合修改图像的可见与未可见区域,通过使后验近似误差在去噪步骤中逐渐降至零,实现无失配的连贯图像修复,在主客观指标上均优于现有扩散方法。

摘要:

图像修复是指基于部分可见的参考图像生成完整自然图像的任务。近期,许多研究兴趣集中在使用固定的扩散模型解决这一问题。这些方法通常直接将中间或最终生成图像的可见区域替换为参考图像或其变体的相应区域。然而,由于未可见区域未被直接修改以匹配上下文,这导致可见与未可见区域之间出现不连贯。为解决该不连贯问题,少数方法引入了严格的贝叶斯框架,但由于计算后验分布时的近似误差,它们往往会在生成图像与参考图像之间引入不匹配。本文提出COPAINT,能够连贯地修复整幅图像而不引入不匹配。COPAINT同样使用贝叶斯框架联合修改可见与未可见区域,但以一种使误差在去噪步骤中逐渐降至零的方式来近似后验分布,从而强烈惩罚与参考图像的任何不匹配。实验验证COPAINT在客观与主观指标上均优于现有的基于扩散的方法。代码可在https://github.com/UCSB-NLP-Chang/CoPaint/获取。

英文摘要:

Image inpainting refers to the task of generating a complete, natural image based on a partially revealed reference image. Recently, many research interests have been focused on addressing this problem using fixed diffusion models. These approaches typically directly replace the revealed region of the intermediate or final generated images with that of the reference image or its variants. However, since the unrevealed regions are not directly modified to match the context, it results in incoherence between revealed and unrevealed regions. To address the incoherence problem, a small number of methods introduce a rigorous Bayesian framework, but they tend to introduce mismatches between the generated and the reference images due to the approximation errors in computing the posterior distributions. In this paper, we propose COPAINT, which can coherently inpaint the whole image without introducing mismatches. COPAINT also uses the Bayesian framework to jointly modify both revealed and unrevealed regions, but approximates the posterior distribution in a way that allows the errors to gradually drop to zero throughout the denoising steps, thus strongly penalizing any mismatches with the reference image. Our experiments verify that COPAINT can outperform the existing diffusion-based methods under both objective and subjective metrics. The codes are available at https://github.com/UCSB-NLP-Chang/CoPaint/.

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

↑