基于扩散模型的图像编辑领域的首篇综述(中科院深先院等单位)

轻松参会 · 公众号 · · 2024-03-01 10:31

正文

本文来自轻松参会某群友（也是本文某位作者）的投稿

AIGC大模型最火热的任务之一——基于Diffusion Model的图像编辑(editing)领域的首篇综述。长达26页，涵盖297篇文献！本文全面研究图像编辑前沿方法，并根据技术路线精炼地划分为3个大类、14个子类，通过表格列明每个方法的类型、条件、可执行任务等信息。此外，本文提出了一个全新benchmark以及LMM Score指标来对代表性方法进行实验评估，为研究者提供了便捷的学习参考工具。强烈推荐AIGC大模型研究者或爱好者阅读，紧跟热点。

作者：Yi Huang, Jiancheng Huang, Yifan Liu, Mingfu Yan, Jiaxi Lv, Jianzhuang Liu, Wei Xiong, He Zhang, Liangliang Cao, Shifeng Chen

单位: 中科院、Adobe公司、苹果公司(曹亮亮)等

论文链接:

Diffusion Model-Based Image Editing: A Survey arxiv.org/abs/2402.17525

Github网址：

https://github.com/SiatMMLab/Awesome-Diffusion-Model-Based-Image-Editing-Methods github.com/SiatMMLab/Awesome-Diffusion-Model-Based-Image-Editing-Methods

摘要

去噪扩散模型已成为各种图像生成和编辑任务的有力工具，有助于以无条件或输入条件的方式合成视觉内容。这些模型背后的核心理念是学习如何逆转逐渐向图像中添加噪声的过程，从而从复杂的分布中生成高质量的样本。在这份调查报告中，我们详尽概述了使用扩散模型进行图像编辑的现有方法，涵盖了该领域的理论和实践方面。我们从学习策略、用户输入条件和可完成的一系列具体编辑任务等多个角度对这些作品进行了深入分析和分类。此外，我们还特别关注图像的inpainting和outpainting，并探讨了早期的传统上下文驱动方法和当前的多模态条件方法，对其方法论进行了全面分析。为了进一步评估文本引导图像编辑算法的性能，我们提出了一个系统基准 EditEval，其特点是采用了创新指标 LMM Score。最后，我们讨论了当前的局限性，并展望了未来研究的一些潜在方向。附带的资源库发布在 GitHub - SiatMMLab/Awesome-Diffusion-Model-Based-Image-Editing-Methods: Diffusion Model-Based Image Editing: A Survey (arXiv).

介绍

在人工智能生成内容（AIGC）的领域中，利用人工智能来创建和修改数字内容，图像编辑被认为是创新和实际应用的一个重要领域。与从最小输入创建新图像的图像生成不同，图像编辑涉及改变图像的外观、结构或内容，包括从微妙的调整到重大变革的一系列变化。这项研究在包括数字媒体、广告和科学研究在内的各个领域都是基础性的，其中改变视觉内容是至关重要的。图像编辑的进化反映了数字技术的进步，从手工、劳动密集型过程发展到由基于学习的算法驱动的先进数字技术。这一进化中的一个关键进步是生成对抗网络（GANs）的引入，显著增强了创意图像操作的可能性。

扩散模型在图像编辑中的应用近年来引起了极大的兴趣，这一点从该领域大量研究出版物的显著增加中可以得到证明。这种日益增长的关注突显了与先前作品相比，扩散模型在提升图像编辑性能方面的潜力和多样性。鉴于这一重要进步，系统地回顾和总结这些贡献是必要的。然而，现有关于扩散模型的综述文献集中在其他特定的视觉任务上，如视频应用或图像恢复和增强。一些提到图像编辑的调查往往只提供了一个粗略的概述，缺少对方法的详细和集中探索。

为了弥补这一差距，我们进行了一项调查，提供了一份深入而全面的分析，专注于图像编辑。我们深入探讨了扩散模型在该领域实现的方法学、输入条件和广泛的编辑任务。该调查批判性地回顾了超过100篇研究论文，根据学习策略将它们组织成三个主要类别：基于训练的方法、测试时微调方法和无需训练和微调的方法。每个类别进一步基于其核心技术进行划分，分别在第4、5和6节中进行了详细讨论。我们还探索了这些方法中使用的10种不同类型的输入条件，包括文本、掩码、参考图像、类别、布局、姿势、草图、分割图、音频和拖动点，以展示扩散模型在多样化图像编辑场景中的适应性。此外，我们的调查提出了一个新的图像编辑任务分类，将其分为三个广泛的类别：语义编辑、风格编辑和结构编辑，涵盖了12个具体类型。图1直观地表示了研究在学习策略、输入条件和编辑任务类别上的统计分布。此外，我们特别关注了修复和外扩，它们共同构成了一种独特的编辑类型。我们探讨了早期的传统方法和当前的多模态条件方法，第7节提供了它们方法学的全面分析。我们还介绍了EditEval，一个旨在评估文本引导的图像编辑算法的基准，详细内容见第8节。特别是，通过利用大型多模态模型（LMMs）的先进视觉-语言理解能力，提出了一个有效的评估指标，LMM分数。最后，我们在第9节中提出了一些当前挑战和潜在的未来趋势作为展望。

总之，这项调查旨在系统地分类和批判性评估基于扩散模型的图像编辑研究的广泛文献。我们的目标是提供一个全面的资源，不仅综合了当前的发现，而且还指导了这一快速发展领域未来研究的方向。

图像编辑的分类

除了扩散模型在图像生成、恢复和增强方面取得的重大进步外，它们在图像编辑方面也实现了显著突破，相比之前占主导地位的生成对抗网络（GANs），提供了更强的可控性。不同于专注于从零开始创建新图像的图像生成，以及旨在修复和提高退化图像质量的图像恢复和增强，图像编辑涉及修改现有图像的外观、结构或内容，包括添加对象、替换背景和改变纹理等任务。

在这项调查中，我们根据学习策略将图像编辑论文分为三个主要组别：基于训练的方法、测试时微调方法和无需训练和微调的方法，分别在第4、5和6节中进行详细阐述。此外，我们探讨了这些方法使用的10种输入条件来控制编辑过程，包括文本、掩码、参考图像、类别、布局、姿势、草图、分割图、音频和拖动点。进一步地，我们调查了这些方法可以完成的12种最常见的编辑类型，它们被组织成以下三个广泛的类别。

- 语义编辑：此类别包括对图像内容和叙述的更改，影响所描绘场景的故事、背景或主题元素。这一类别内的任务包括对象添加、对象移除、对象替换、背景更改和情感表达修改。

- 风格编辑：此类别侧重于增强或转换图像的视觉风格和审美元素，而不改变其叙述内容。这一类别内的任务包括颜色更改、纹理更改和整体风格更改，涵盖艺术性和现实性风格。

- 结构编辑：此类别涉及图像内元素的空间布局、定位、视角和特征的变化，强调场景内对象的组织和展示。这一类别内的任务包括对象移动、对象大小和形状更改、对象动作和姿势更改以及视角/视点更改。

表1全面总结了调查论文的多角度分类，提供了快速搜索的可能性。

基于训练的方法

在基于扩散模型的图像编辑领域，基于训练的方法已经获得了显著的突出地位。这些方法不仅因其稳定的扩散模型训练和有效的数据分布建模而著名，而且在各种编辑任务中表现可靠。为了彻底检查这些方法，我们根据它们的应用范围、训练所需条件和监督类型将它们分类为四个主要组别，如图2所示。此外，在这些主要组别中，我们根据它们的核心编辑方法将方法分为不同的类型。这种分类展示了这些方法的范围，从针对特定领域的应用到更广泛的开放世界用途。

测试时微调方法

在图像生成和编辑中，测试时微调代表着精确性和控制性的重要一步。本节探讨各种微调策略（见图5）来增强图像编辑能力。这些方法，如图6所示，范围从微调整个去噪模型到专注于特定层或嵌入。我们研究微调整个模型、针对特定参数的方法以及优化基于文本的嵌入。此外，我们还讨论了超网络的集成和直接图像表示优化。这些方法共同展示了图像编辑中微调技术的不断复杂化和有效性，满足广泛的编辑需求和用户意图。

免训练和微调方法

在图像编辑领域，无需训练和微调的方法起点是它们快速且成本低，因为在整个编辑过程中不需要任何形式的训练（在数据集上）或微调（在源图像上）。本节根据它们修改的内容将它们分类为五个类别，如图7和8所示。它们巧妙地利用扩散模型内在的原则来实现编辑目标。

图像修复和外扩

图像修复和外扩通常被视为图像编辑的子任务，在具有不同目标和挑战的独特位置上占据一席之地。为了更好地解释，我们将它们分为两大类型（见图9），详细内容分别在第7.1节和第7.2节中介绍。

基准

在前面的章节中，我们深入探讨了基于扩散模型的图像编辑方法的方法学方面。除了这些分析之外，评估这些方法、检查它们在不同编辑任务中的能力至关重要。然而，现有的图像编辑基准测试有限，没有完全满足我们调查中确定的需求。例如，EditBench主要针对文本和掩码引导的修复，并忽略了涉及全局编辑如风格转换的更广泛任务。TedBench虽然扩展了任务范围，但缺乏详细指导，这对于评估依赖于文本指令而非描述的方法至关重要。此外，尽管EditVal基准试图提供更全面的任务和方法覆盖范围，但受到其从MS-COCO数据集中获取的图像质量限制，这些图像通常分辨率低且模糊。
为了解决这些问题，我们引入了EditEval，一个旨在评估通用扩散模型基础图像编辑方法的基准。EditEval包括一个精心策划的50张高质量图像的数据集，每张图像都附有文本提示。EditEval评估表1中选出的7个常见编辑任务的性能。此外，我们提出了LMM分数，这是一个定量评估指标，利用大型多模态模型（LMMs）的能力来评估不同任务上的编辑性能。除了LMM分数提供的客观评估外，我们还进行了用户研究以纳入主观评估。EditEval的构建和应用的详细信息如下所述。

挑战和未来方向

尽管在使用扩散模型进行图像编辑方面取得了成功，但仍有一些限制需要在未来的工作中加以解决。

更少步骤的模型推理 。大多数基于扩散的模型在推理过程中需要大量的步骤来获取最终图像，这既耗时又耗费计算资源，给模型部署和用户体验带来挑战。为了提高推理效率，已经研究了少步骤或一步生成的扩散模型。近期的方法通过从预训练的强扩散模型中提取知识来减少步骤数，以便少步骤模型能够模仿强模型的行为。一个更有趣但更具挑战性的方向是直接开发少步骤模型，而不依赖于预训练的模型，例如一致性模型。

高效模型 。训练一个能够生成逼真结果的扩散模型在计算上是密集的，需要大量的高质量数据。这种复杂性使得开发用于图像编辑的扩散模型非常具有挑战性。为了降低训练成本，近期的工作设计了更高效的网络架构作为扩散模型的骨干。此外，另一个重要方向是只训练部分参数，或者冻结原始参数并在预训练的扩散模型之上添加一些新层。

复杂对象结构编辑 。现有的工作可以在编辑图像时合成逼真的颜色、风格或纹理。然而，当处理复杂结构时，它们仍然会产生明显的人为痕迹，例如手指、标志和场景文字。已经尝试解决这些问题。以前的方法通常使用负面提示，如“六个手指、坏腿等”，以使模型避免生成此类图像，这在某些情况下是有效的，但不够稳健。近期的工作开始使用布局、边缘或密集标签作为指导，编辑图像的全局或局部结构。

复杂的光照和阴影编辑 。编辑对象的光照或阴影仍然是一个挑战，这需要准确估计场景中的光照条件。以前的工作如Total Relighting使用网络组合来估计前景对象的法线、反照率和阴影，以获得逼真的重新照明效果。最近，基于扩散的模型被提议用于编辑面部的光照（DiFaReli）。然而，利用预训练扩散模型强大的光照先验来编辑肖像或通用对象的光照仍然是一个开放领域。同样，ShadowDiffusion探索了基于扩散的阴影合成，可以生成视觉上令人愉悦的对象阴影。然而，使用扩散模型在不同背景条件下准确编辑对象的阴影仍然是一个未解决的问题。

总结

我们广泛回顾了基于扩散模型的图像编辑方法，从多个角度审视了这一领域。我们的分析首先根据学习策略将100多种方法分类为三个主要组别：基于训练、测试时微调和无需训练及微调的方法。然后，我们将图像编辑任务分为三个独特的类别：语义编辑、风格编辑和结构编辑，总共包含12种具体类型。我们探索了这些方法及其对提高编辑性能的贡献。在我们的图像编辑基准EditEval中，对7个任务以及最新的先进方法进行了评估。此外，引入了一种新的度量标准LMM分数，用于这些方法的比较分析。在我们的综述中，我们强调了图像编辑领域内广泛的潜力，并建议了未来研究的方向。